Distillation Lead

Job description

Job responsibilities

• Define and drive the technical strategy for model distillation and compression across Waabi's AI stack — spanning perception, world models, and planning — with an eye toward both onboard deployment and simulation…
• Design, implement, and scale state-of-the-art distillation and efficiency pipelines, which may include: Distillation for generative models (diffusion, autoregressive, flow-matching, video models)
• Quantization-aware training (QAT) and post-training quantization (PTQ)
• Knowledge distillation (feature-level, response-based, and relation-based)
• Structured and unstructured pruning and sparsification
• Low-rank factorization and efficient architecture design
• Speculative decoding and other inference-time efficiency techniques
• Collaborate closely with ML Platform, Infrastructure, Onboard, Autonomy, and Simulation teams to integrate compressed models into production pipelines and meet latency, memory, and throughput targets across deployment…