Machine Learning Engineer at Krea (San Francisco, CA)
Krea· San Francisco, CA·
Role details
Job description
ABOUT KREA
At Krea, we are building next-generation AI creative tools.
We're 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. We recently took this a step forward with the launch of Krea 2 krea.ai/krea-2-image-model, our first foundation model, built completely from scratch for aesthetic diversity and stylistic control.
We've raised over $83M and are backed by world-class investors such as a16z, Bain Capital, and Abstract. We work full-time and in-person at our waterfront office in San Francisco. We care about creativity: our team includes musicians, designers, visual artists, and engineers.
THIS ROLE
We're looking for a machine learning engineers who can work on large-scale image and video models training experiments.
WHAT YOU'LL 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
WHAT 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
WHAT WE OFFER
- Team: Work alongside a world-class team building the future of AI creative tooling
- Impact: Significant scope and company-wide impact
- Competitive compensation: generous salary & equity packages
- Health & wellness: 100% health & 99% dental/vision insurance premiums covered for employees, health FSA accounts, & long-term disability coverage
- Time off: Flexible PTO policy
- Financial planning: 401k with a 4% company-sponsored match
- Meals in the office: breakfast, lunch, dinner - you name it, we'll cover it
- Transit: Ubers covered to & from the office
- Sponsorship: We're open to sponsoring international visas where we can (e.g., STEM OPT, OPT, H-1B, O-1, E-3).
- And more!
Please note the above benefits & perks are for full-time employees
Why work at Krea
- Culture: Small, highly talented team (currently 17) that lives and breathes creativity – founders and many team members are musicians, poets, designers, videographers; “The team behind Krea is the work of a small, talented group of imaginative, incredibly dedicated people.”
- Engineering & research: Engineers from MIT, Princeton, UC Berkeley, University of Chicago; authors of state‑of‑the‑art models (Lumina, Anifusion). Freedom to do own research after building products for the community.
- Work environment: New office with Pacific Ocean views in San Francisco; majority of team shares a living space (noted early‑stage).
- Velocity: Weekly Thursday releases, exposure to cutting‑edge AI models, direct impact on a product used by millions.
- Open roles: Listed at jobs.ashbyhq.com/krea. Ideal candidate is “deeply technical, artistically opinionated.”