ML Researcher - Image / Video Diffusion

Highlights

• Train diffusion models for image and video generation on large GPU clusters.
• Build next-gen AI creative tools that empower human creativity, not replace it.
• Advance Krea 2 foundation model built for aesthetic diversity and control.

What they offer

• Generous salary and equity packages
• 100% health and 99% dental/vision premiums covered, FSA and disability
• Flexible PTO policy
• 401k with 4% company match
• Breakfast, lunch and dinner in the office
• Ubers covered to and from the office

Ideal candidate

Experienced researcher-engineer passionate about creative AI who thrives in fast, in-person teams. Action-biased and comfortable with ambiguity, with taste for simple methods that scale.

Requirements

• Proven track record working with image or video models at scale
• Strong proficiency in PyTorch and understanding of its inner workings
• Strong background in distributed training paradigms such as FSDP, CP, SP, TP and EP
• Experience profiling and debugging large distributed training runs
• Knowledge of low-precision training/inference in FP8, NVFP4 and MXFP8
• Understanding of diffusion training across pretraining, midtraining, preference optimization and RL
• Comfortable breaking ambiguous research goals into concrete plans and execution