Machine Learning Engineer - Distillation

Responsibilities

• 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
• Contribute back to open-source models, tooling, or research

Languages