Role overview
This team is looking for strong Machine Learning Engineers with
deep experience in GPU optimization,
systems-level performance, and large-scale model development. You’ll contribute to building advanced generative models and help bring research-driven breakthroughs into production environments used by real creators.
Responsibilities
- Drive research and development efforts in computer vision, with a focus on generative image and video models
- Build and optimize high-performance training and inference pipelines across GPUs
- Work closely with engineering and product teams to bring cutting-edge ML capabilities into production
- Contribute to systems and tooling that enable real-time or interactive creative workflows
Basic qualifications
- 5+ years of relevant engineering or ML research experience (computer vision, graphics, generative models, or similar)
- Strong experience with CUDA, C++, and low-level performance optimization
- Proficiency with modern ML frameworks such as PyTorch or TensorFlow
- Ability to write clean, maintainable research and production code
- Interest in human-in-the-loop creative systems and emerging generative media tools
- Comfort working in a fast-moving environment with tight iteration cycles
- Strong communication and collaboration skills
Tags & focus areas
Used for matching and alerts on DevFound Fulltime Remote Ai Machine Learning