Responsibilities
* Design and develop foundational image/video diffusion ML models.
* Train large-scale distributed models.
* Build demos to showcase prototypes.
* Manage large datasets for training.
* Optimize and deploy models for production.
Key Qualifications
* PhD in Computer Science or related field with strong publications in ML, Computer Vision, Robotics, or Graphics.
* Expertise in generative models (image/video diffusion) and modern transformer architectures.
Preferred Qualifications
* Experience with GANs, diffusion models, LLMs, and large-scale deep learning in PyTorch or similar.
* Strong coding skills with fast ML experimentation.