Staff Machine Learning Engineer, Optimization

Staff Machine Learning Engineer, Optimization job at Waymo. Focus on optimizing model inference and training, ensuring these advancements generalize across multiple platforms. Work embedded in an ML Engineering and Modeling team to drive scale and multi-platform support of models.

Requirements

• Master’s degree or PhD in Computer Science, Engineering, or a related technical field
• 3+ years of experience in software development for neural model inference or neural model training, and 1+ years experience with neural model inference and training optimization on modern GPU/TPU architectures
• 5+ years experience in software development for real-time systems, ideally experience with real-time systems running on device (e.g., Waymo’s onboard system)
• Proficiency in C++, Python, and modern deep learning toolkits like PyTorch or JAX
• Passionate about low-level neural net optimization and willingness to learn new architectures and tools
• Deep understanding of latency and quality tradeoffs as it applies to neural network architectures and practical experience making said tradeoffs

Benefits

• discretionary annual bonus program
• equity incentive plan
• generous Company benefits program