Ai Distilation Expert
rb labsMillbrae, CA
Ai Distilation Expert
rb labsMillbrae, CA
yesterday
Occupations
Computer and Information Research ScientistsSoftware DevelopersComputer Systems Engineers/ArchitectsIndustries
Custom Computer Programming ServicesComputer Systems Design ServicesAll Other Professional, Scientific, and Technical ServicesAbout the role
About the Role:
We’re building a real-time generative AI system where latency is the core challenge. We’re looking for amodel distillation specialistto help us shrink large models into fast, deployment-ready versions without sacrificing quality.
This is a hands-on role: you’ll design and run distillation experiments, evaluate quality/latency tradeoffs, and ship the results to production GPU infrastructure.
What You’ll Do:
Design and run knowledge distillation pipelines– teacher/student setups for large neural models (LLMs and/or generative vision models)
Optimize for latency– reduce inference cost via distillation, quantization, pruning, and mixed precision
Evaluate rigorously– build empirical quality-vs-latency comparisons and A/B test variants
Own training loops– write and debug full PyTorch training code, not just run existing scripts
Ship to production– deploy distilled models on live GPU servers and measure real-world performance
Must Have 2 years hands-on PyTorch, including training/fine-tuning models from scratch – not just serving pretrained ones
Proven experience with model distillation– you’ve actually distilled a model and measured the results
Strong grasp ofmodel compression techniques : quantization, pruning, knowledge distillation, LoRA/parameter-efficient methods
Solid understanding ofloss designfor distillation (KL divergence, feature/logit matching, perceptual losses)
Comfortable reading and extendingdense, lightly-documented research code
Strong PlusExperience distillingautoregressive LLMs or speech models
Experience withGANs, diffusion, or flow-matchingmodels
GPU inference optimization: torch.compile, CUDA memory management, multi-GPU setups
Familiarity with streaming/real-time ML systems
Working StyleWe operate like a research lab: empirical, fast iteration, incomplete docs, self-directed testing. You’ll need to be comfortable designing your own experiments and pushing for more engineering structure as you go.
Matching similar jobs
JOB OVERVIEW
Experience level
Senior
Location
Millbrae, CA
Occupation
Computer and Information Research Scientists
Industry
Custom Computer Programming Services
Posted
yesterday
Tired of running searches?
Rank the roles you'd take once, and matches like these arrive on their own.
CREATE PROFILE (opens in a new tab)