Machine Learning Engineer — Distillation

About this role

ABOUT THE ROLE

We’re looking for a Machine Learning Engineer focused on model distillation to help us build smaller, faster, and more efficient models without sacrificing quality. You’ll work at the intersection of research and production—taking cutting-edge techniques and turning them into systems that scale.

This is a hands-on role with real ownership: you’ll design distillation pipelines, run large-scale experiments, and ship models used in production.

WHAT YOU’LL DO

- Design and implement knowledge distillation pipelines (teacher–student, self-distillation, multi-teacher, etc.)

- Distill large foundation models into smaller, faster, and cheaper models for inference

- Run and analyze large-scale training experiments to evaluate quality, latency, and cost tradeoffs

- Collaborate with research to translate new distillation ideas into production-ready code

- Optimize training and inference performance (memory, throughput, latency)

- Contribute to internal tooling, evaluation frameworks, and experiment tracking

- (Optional) Contribute back to open-source models, tooling, or research

WHAT WE’RE LOOKING FOR

- Strong background in machine learning or deep learning

- Hands-on experience with model distillation (LLMs or other neural networks)

- Solid understanding of training dynamics, loss functions, and optimization

- Experience with PyTorch (or JAX) and modern ML tooling

- Comfort running experiments on multi-GPU or distributed setups

- Ability to reason about model quality vs. performance tradeoffs

- Pragmatic mindset: you care about shipping, not just papers

NICE TO HAVE

- Experience distilling LLMs or large sequence models

- Experience with inference optimization (quantization, pruning, kernels, etc.)

- Familiarity with evaluation for language models

- Open-source contributions or research publications

- Experience in early-stage or fast-moving startups

WHY JOIN

- Work on core model quality and cost efficiency—not side projects

- High ownership and direct impact on product and roadmap

- Small, senior team with strong research + engineering culture

- Competitive compensation + meaningful equity

- Remote-friendly, async-first environment

Similar jobs

• Machine Learning Engineer — Adobe · San Francisco
• Machine Learning Engineer — Orchard · San Francisco
• Machine Learning Engineer — S&P Global · New York, NY
• Machine Learning Engineer — Samsung Electronics · TX
• Machine Learning Engineer — CACI · Lackland AFB, TX | 2AT SAN ANTONIO TX (TAX JURISDICTION - BEXAR COUNTY)
• Machine Learning Engineer — Extra Space Storage · Salt Lake City, UT
• Machine Learning Engineer — Claylabs · San Francisco
• Machine Learning Engineer — Reducto · San Francisco Office