Distillation Lead

waabi.ai

San Francisco, CA, United States
On-site
Model distillation expertise
Quantization-aware training implementation
Large-scale distributed training experience
Waabi is seeking a Distillation Lead to define the technical strategy for model compression across their Physical AI platform powering commercial autonomous trucks and robotaxis

Job Summary

  • Waabi is seeking a Distillation Lead to define the technical strategy for model compression across their Physical AI platform powering commercial autonomous trucks and robotaxis.
  • The role involves designing state-of-the-art distillation pipelines for generative models, quantization, and pruning to meet strict latency and memory targets for real-time deployment.
  • Candidates will mentor researchers and engineers while collaborating with infrastructure and autonomy teams to integrate compressed models into production environments.

Matching Summary

Waabi is seeking a Distillation Lead to define the technical strategy for model compression across their Physical AI platform powering commercial autonomous trucks and robotaxis.

Skills & Requirements

Must-have

  • Model distillation expertise
  • Quantization-aware training implementation
  • Large-scale distributed training experience
  • Python and PyTorch proficiency
  • Cross-functional collaboration skills

Nice-to-have

  • Hardware-aware optimization knowledge
  • Top-tier ML publication record
  • Experience with generative models
  • Background in autonomous vehicles
  • Open-source contribution history

Key Requirements

  • Bachelor's or Master's degree in Machine Learning or related field
  • Extensive hands-on experience with model compression techniques
  • Proven track record of driving projects from conception to production

Work Rights

Not specified

Tailored Resume

Cover Letter