Related skills
python pytorch distributed training quantization pruning📋 Description
- Define and drive the technical strategy for distillation and compression across Waabi's AI stack.
- Design, implement, and scale distillation and efficiency pipelines.
- Collaborate with ML Platform, Infrastructure, Onboard Autonomy, and Simulation teams.
- Deliver compressed models for real-time onboard systems and high-throughput simulation.
- Mentor researchers and engineers; champion best practices in model compression.
- Set benchmarks and evaluation frameworks for efficiency vs. quality.
🎯 Requirements
- Deep distillation expertise with production impact.
- Bachelor's/Master's in ML, CV, Robotics or related field, or equivalent experience.
- Expert Python and PyTorch (or JAX) with large-scale distributed training.
- Technical leadership with a track record delivering projects to production.
- Cross-functional collaboration across infrastructure, platform, and autonomy teams.
- Bonus: hardware-aware optimization (TensorRT, ONNX) and CUDA.
🎁 Benefits
- Competitive compensation and equity awards.
- Health, Dental and Vision coverage (full-time).
- Unlimited Vacation.
- Flexible hours and Work from Home support.
- Daily drinks, snacks and catered meals when in office.
- Regular team building events on-site, off-site, and virtually.
Meet JobCopilot: Your Personal AI Job Hunter
Automatically Apply to Engineering Jobs. Just set your
preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.
Help us maintain the quality of jobs posted on Empllo!
Is this position not a remote job?
Let us know!