Machine Learning Engineer

Krea AI Krea AI · Multimodal · San Francisco, CA · Research

Machine Learning Engineer at Krea AI, focusing on large-scale training of foundation diffusion models for image and video generation, including controllability modules. The role involves developing novel research techniques and implementing them in production, optimizing data pipelines, and conducting experiments on HPC clusters.

What you'd actually do

  1. Train foundation diffusion models for image and video generation.
  2. Train controllability modules such as IPAdapters or ControlNets.
  3. Develop novel research techniques and put them into production.
  4. Conducting large-scale experiments on high-performance computing clusters, optimizing data pipelines for massive image datasets

Skills

Required

  • deep learning frameworks
  • distributed training paradigms
  • image models at scale
  • video models at scale

Nice to have

  • publications
  • open-source contributions
  • creative research directions

What the JD emphasized

  • large-scale image and video models training experiments
  • large-scale experiments
  • massive image datasets

Other signals

  • large-scale image and video models training experiments
  • Train foundation diffusion models for image and video generation
  • Develop novel research techniques and put them into production