KA
Listed on Krea AI’s own careers site. You apply with them directly — we never stand between you and the employer.
What this role is
Krea is seeking a researcher to optimize and train large-scale diffusion models for image and video generation on distributed GPU clusters. This role suits experienced machine learning engineers comfortable with complex distributed training infrastructure, profiling large experiments, and collaborating closely with a creative-focused AI team.
Our summary, not Krea AI’s wording. The full posting is on their site.
What they ask for
Required
- Proven track record training image or video models at scale
- Strong proficiency in PyTorch and understanding of its internals
- Deep knowledge of distributed training strategies including FSDP, CP, SP, TP, and EP
- Experience profiling and debugging large distributed training runs
- Knowledge of low precision training in FP8, NVFP4, and MXFP8
- Solid understanding of diffusion model training across pretraining, midtraining, preference optimization, and RL
- Comfortable working with underspecified research goals and breaking them into concrete plans
- Ability to iterate rapidly and propose creative research directions
Nice to have
- Publications or open-source contributions working with image or video models at scale
- Familiarity with developments in LLM, VLM, representation learning, and robotics research