Machine Learning Engineer
At a glance
Mid-level Machine Learning Engineering role at Krea. San Francisco · full-time · no salary stated · we estimate $197,625–$279,000.
Krea has not published a salary for this role. The figure below is our estimate, not theirs.
aimlroles summary, based on the employer's posting.
What you'll do
- Train diffusion foundation models that generate images and video
- Build control modules such as IPAdapters or ControlNets for model outputs
- Turn new research approaches into techniques used in production
- Run large experiments on high-performance computing clusters and tune pipelines for massive image datasets
What you bring
- A track record applying image or video models at scale, shown through your work
- Strong command of deep learning frameworks and distributed training methods
- You iterate quickly and can put forward fresh research directions
Who this fits
This mid-level role suits an engineer focused on training image and video models and developing research methods. The job is full-time and in person at the San Francisco office.
From the employer
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, our first foundation model, built completely from scratch for aesthetic diversity and stylistic control, and with Krea 3 - our new image editing model.
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