Machine Learning Engineer, Model Optimization
Core
Optimizing model training and inference for autonomous driving perception systems to improve efficiency on onboard and offboard hardware.
Role type
Machine Learning Engineer, Model Optimization
Builds
Optimized perception models for the Waymo Driver autonomous vehicle system
Domain
Autonomous driving / Computer Vision
Deliverable
production ML models
Required skills
Deep learning, Computer Vision, Python, PyTorch, JAX, Model architecture design, Hardware co-development, Sparse representation optimization, Real-time inference optimization
Preferred skills
C++, MS or PhD in ML/Robotics/CS, Publications at top-tier conferences (CVPR, NeurIPS, etc.)
Responsibilities
Optimize FLOPs utilization in model training and inference via architecture/hardware co-development; Optimize model inference for onboard and offboard platforms; Analyze and optimize real-time inference of complex model architectures
Seniority
Mid-level IC (3+ years experience)