2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning
Core
Design, implement, and benchmark parameter-efficient fine-tuning modules and distillation pipelines for multi-task Vision Transformer backbones to optimize Waymo's autonomous driving perception models.
Role type
Summer Intern, Machine Learning Engineer (Model Optimization)
Builds
Optimized perception and foundation models for Level 4 autonomous driving
Domain
Autonomous driving, Machine Learning, Robotics
Deliverable
production ML models
Required skills
Deep learning development, Transformer architectures, Parameter-efficient fine-tuning (LoRA/QLoRA), Model quantization, Python, JAX/Flax
Preferred skills
Top-tier ML/CV publications, Hardware-aware optimization, Distributed training on TPU/GPU clusters, Autonomous driving perception tasks
Technologies
JAX, Flax, PyTorch, TensorFlow, Vision Transformers, LoRA, QLoRA
Responsibilities
Design and implement parameter-efficient fine-tuning modules; Develop dual-level distillation pipelines; Collaborate on static weight folding and low-precision quantization; Conduct empirical ablations on autonomous driving datasets
Seniority
Intern