Machine Learning Engineer (AI)

TL;DR

Machine Learning Engineer (AI): Developing and optimizing large-scale image and video models with an accent on training foundation diffusion models and controllability modules. Focus on conducting large-scale experiments and optimizing data pipelines for massive datasets.

Location: On-site in San Francisco

Company

Krea is building next-generation AI creative tools to empower human creativity.

What you will 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.
• Optimize data pipelines for massive image datasets.

Requirements

• Proven track record in working with image or video models at scale.
• Strong background in deep learning frameworks and distributed training paradigms.
• Ability to iterate rapidly and propose creative research directions.