Machine Learning Engineer

The role

Machine learning engineers will work on large-scale image and video models training experiments. This includes training foundation diffusion models and controllability modules.

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
• Conducting large-scale experiments on high-performance computing clusters
• Optimizing data pipelines for massive image datasets

Required qualifications

• 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

Nice to have

• publications or open-source contributions

Benefits

• Significant scope and company-wide impact
• generous salary & equity packages
• 100% health insurance premiums covered for employees
• 99% dental/vision insurance premiums covered for employees
• health FSA accounts
• long-term disability coverage
• Flexible PTO policy
• 401k with a 4% company-sponsored match
• breakfast, lunch, dinner
• Ubers covered to & from the office

Before you apply

• We're open to sponsoring international visas where we can (e.g., STEM OPT, OPT, H-1B, O-1, E-3)

About the company

Krea is building next-generation AI creative tools. They have raised over $83M and are backed by investors such as a16z, Bain Capital, and Abstract.