ML Engineer — Diffusion Image & Video Models

Krea; San Francisco (CA); On-site; USD 140,000 - 210,000; Full time

Job summary

Krea is seeking machine learning engineers to work on large-scale image and video model training experiments. You will train diffusion models and controllability modules, develop novel research techniques, and run large-scale experiments on HPC clusters to optimize data pipelines for massive image datasets.

Ideal candidates have a proven track record with image or video models, strong deep learning framework experience, and the ability to iterate rapidly while proposing new research directions.

Qualifications

• Proven track record in working with image or video models at scale (publications or open-source contributions a plus).
• Strong background in deep learning frameworks and distributed training paradigms.
• Ability to iterate rapidly, and propose creative research directions.

Responsibilities

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

Skills

Image/Video Models

Deep Learning

Distributed Training

Tools

PyTorch

TensorFlow

HPC Clusters

Job description

Krea is seeking machine learning engineers to work on large-scale image and video model training experiments. You will train diffusion models and controllability modules, develop novel research techniques, and run large-scale experiments on HPC clusters to optimize data pipelines for massive image datasets.

Ideal candidates have a proven track record with image or video models, strong deep learning framework experience, and the ability to iterate rapidly while proposing new research directions.