Profile cover
A

Senior AI Diffusion Model Distillation Engineer

Elvio Works

Remote · $200k+ / year · Full-time · Posted 4 weeks ago

This role is no longer accepting applications

Don't miss the next role like this

Create a free profile with your skills. We match you to new AI roles as they open, and companies hiring can contact you directly.

Create free profile, 2 minutes

Similar open roles you can apply to now:

See all open AI jobs

About the role

Every image our users generate costs us about four seconds of GPU time today. Cutting that to under one second would change our pricing, our margins and what we can offer on mobile. We build image generation tools for e-commerce teams producing product shots at volume. Our models are based on open diffusion architectures fine-tuned on product photography, and the quality is where we want it. Speed is not. You'll lead work on step distillation, consistency models and quantization to cut inference cost without hurting fidelity. We expect you to read recent papers, reproduce what's promising and be candid about what doesn't work.

Fully remote.

Responsibilities

  • Lead distillation and acceleration research for our production diffusion models
  • Reproduce and adapt recent methods such as consistency and adversarial distillation
  • Set up human and automated quality evaluations for faster models
  • Partner with infrastructure engineers to ship optimized models to production

Requirements

  • Deep expertise training or fine-tuning diffusion models in PyTorch
  • Experience with model compression, distillation or quantization
  • Track record of turning research ideas into shipped systems
  • Familiarity with Hugging Face Diffusers

Benefits

  • Fully remote, US-friendly hours
  • Top-of-market salary and equity
  • Dedicated GPU research budget
  • Health, dental and vision fully covered

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$200k+ / year

AI Expertise

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

pytorch python hugging face aws docker

Ready to apply for this role?

Create a free talent account.

  • Find AI roles in one place
  • Get AI-matched job recommendations
  • Reply to companies that message you
  • Build your public AI talent profile

This role is closed. Create your free profile to apply to similar open roles.

Don't miss the next one: get matched to new roles like this.

Create free profile