K

Senior Machine Learning Engineer - Diffusion Model Posttraining

krea • United States
Visa Sponsorship
This Job is No Longer Active This position is no longer accepting applications
AI Summary

Krea is seeking an expert to lead large-scale finetuning and posttraining of diffusion models for image generation. You will implement advanced techniques like RL, DPO, and distillation to enhance model aesthetics and quality. Candidates must have deep experience with PyTorch, distributed training, and VLM reward models.

Key Highlights
Build next-generation AI creative tools with a mission to empower human creativity
Raised over $83M with backing from a16z, Bain Capital, and Abstract
Work in-person at a waterfront office in San Francisco with a team of creatives and engineers

Searching for Development & Programming roles that provide visa sponsorship? Connect with international employers through Development & Programming Jobs with Visa Sponsorship opportunities actively seeking talented professionals.

Key Responsibilities
Finetune diffusion models at scale to improve image aesthetics and quality
Implement posttraining techniques including supervised finetuning, preference optimization, reinforcement learning, and distillation
Design comprehensive eval suites and reward designs for the reinforcement learning stage
Train custom VLM as reward models and custom LLMs for prompt expansion
Coordinate with data teams to manage collection of preference data and model evaluation results
Work on safety alignment of models for open source release
Collaborate with AI research and engineering teams to integrate advancements into products
Technical Skills Required
Diffusion Models PyTorch Reinforcement Learning

Explore our comprehensive directory of visa sponsorship jobs from employers worldwide who are ready to sponsor talented international professionals.

Benefits & Perks
100% health & 99% dental/vision insurance premiums covered
401k with a 4% company-sponsored match
Meals in the office covered
Ubers covered to & from the office
Visa sponsorship available

Interested in opportunities specifically in United State? Discover our dedicated Visa Sponsorship Jobs in United State page featuring roles from top employers in this location.

Nice to Have
Keeping up with developments in LLM, VLM, representation learning, and robotics research
Good research taste with a bias towards simplicity and methods that scale well

Job Description

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.
Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources
Job Overview
Posted Sep 03, 2026
Employment Full-time
Experience Not Applicable
Location United States
Category Programming
Company krea
Keep exploring

Similar Jobs

Explore other opportunities that match your interests

Krea

United States

New

Growth Engineer

  • 3D
  • AI
  • Android
  • AWS
Visa

Hidden company

Log in to see location

Premium

Senior Backend Engineer, AI Gateway

Log in free to see the details

Visas, Work Permits & Legal Insights

Stay informed about visa types, work permits, and immigration updates for international professionals.

Browse all articles

Subscribe to our newsletter

New visa sponsorship, relocation and remote tech jobs, delivered to your inbox.