remote
Diffusion Model Engineer - Collov Labs
AI Engineer
Diffusion Model Engineer to build and optimize generative models for high-quality image and video synthesis.
About the role
Key Responsibilities
- Develop and optimize diffusion models for image and video generation tasks
- Implement advanced sampling techniques to improve model quality and speed
- Collaborate with researchers to integrate new architectural advances
- Optimize models for deployment on edge devices and cloud platforms
- Design and implement evaluation metrics for generative model performance
- Ensure reproducibility and scalability of model training pipelines
Requirements
- 3+ years of experience in AI/ML with a focus on generative models
- Strong background in PyTorch and diffusion model architectures
- Experience with computer vision and image processing techniques
- Familiarity with model optimization and deployment strategies
- Ability to work with large-scale datasets and distributed training
Skills
diffusion modelspytorchcomputer visiongenerative aiimage synthesismodel optimization