2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning
Waymo, Mountain View, CA +1
What you will do
- Design, implement, and benchmark parameter-efficient fine-tuning (LoRA / QLoRA) modules in JAX/Flax for multi-task Vision Transformer backbones
- Develop dual-level distillation pipelines (intermediate feature matching and task-head logit distillation) to mitigate multi-task regressions during large-scale data scaling
- Collaborate with model optimization, quantization, and latency teams to validate static weight folding and low-precision quantization, ensuring zero latency overhead on onboard compute platforms
- Conduct extensive empirical ablations and evaluate perception metrics on large-scale autonomous driving datasets across diverse geographic domains