# ML Researcher - Image / Video Diffusion at Krea

- Company: Krea
- What the company does: Backed by a16z.
- Company website: https://www.krea.ai
- Type: Startups (AI role)
- Level: Mid level
- Location: San Francisco
- Work setup: On-site
- Posted: 2026-09-01
- Apply by: 2026-10-16
- Apply: https://jobs.ashbyhq.com/krea/845dce21-e270-49a2-9229-5becd256b829
- Page: https://www.1752.vc/careers/jobs/krea-ml-researcher-image-video-diffusion/

## About the role

Train diffusion models for image and video generation on large GPU clusters. Fully optimize and profile large distributed training runs across model architectures, kernels, data loading, memory constraints, and communication.

## What they're looking for

- Proven track record in working with image or video models at scale (publications or open-source contributions a plus)
- Strong proficiency in PyTorch and understanding of its inner workings
- Strong background in distributed training paradigms such as FSDP, CP, SP, USP, TP, and EP. Knowing how different parallelism strategies work together and their tradeoffs
- Experience in profiling and debugging large distributed training. Being comfortable with analyzing traces to identify bottlenecks and look for improvements
- Good knowledge of low precision training / inference in FP8, NVFP4, and MXFP8
- Solid understanding of diffusion model training pipeline across pretraining, midtraining, preference optimization, and reinforcement learning

Tags: Research