4 days ago
Mountain View, CA, USAIntern
H1B sponsor
Responsibilities
- Design, implement, and benchmark LoRA and QLoRA modules in JAX/Flax for multi-task Vision Transformer backbones.
- Develop intermediate-feature and task-head-logit distillation pipelines to reduce multi-task regressions during data scaling.
- Validate static weight folding and low-precision quantization with model optimization, quantization, and latency teams.
- Conduct empirical ablations and evaluate perception metrics on large-scale autonomous driving datasets across geographic domains.
Requirements
- Currently pursuing a PhD or master’s degree in computer science, electrical engineering, machine learning, robotics, or a related technical field.
- Strong software engineering and deep learning development skills in Python and modern frameworks such as JAX, Flax, PyTorch, or TensorFlow.
- Theoretical understanding and hands-on experience with deep learning foundation models, Transformer architectures, and multi-task learning.
- Experience with model compression, parameter-efficient fine-tuning such as LoRA, QLoRA, or adapters, quantization, or knowledge distillation.
- Preferred: publications at top-tier computer vision or machine learning conferences.
- Preferred: experience with PTQ, QAT, INT8/INT4/MX4, low-precision numerics, or hardware-aware model optimization.
- Preferred: experience training and scaling large vision backbones or multimodal models on distributed TPU or GPU clusters.
- Preferred: familiarity with autonomous driving perception tasks including 3D object detection, semantics, tracking, or pedestrian intent prediction.
Benefits
- Competitive compensation with a housing or relocation bonus if applicable.
- Medical, dental, and vision insurance.
- Intern events and networking opportunities.
- Free breakfast, lunch, dinner, and snacks onsite.
- Free access to Google shuttles and an onsite gym.
- Hybrid onsite internship position.
Categories
About Waymo
Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.