Machine Learning Engineer

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

About Krea

At Krea, we are building next-generation AI creative tools.

We are dedicated to making AI intuitive and controllable for creatives. Our mission is to build tools that empower human creativity, not replace it.

We believe AI is a new medium that allows us to express ourselves through various formats—text, images, video, sound, and even 3D. We're building better, smarter, and more controllable tools to harness this medium.

This job

We're looking for machine learning engineers who can work on large-scale image and video models training experiments.

Some stuff you can do:

• 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.

Example experience and skills we’re looking for

• 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.

A bit more about us

We’ve raised over $83M and are backed by world-class Silicon Valley investors such as Andreessen Horiwitz, and the cofounder of the Meta AI Research laboratory (FMK as Facebook AI Research) or founding members of OpenAI.