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