Senior Machine Learning Engineer, Runtime and Serving

What this role involves

Waymo is hiring a Senior ML Engineer for its ML Optimization team to build the next-generation ML inference engine for autonomous driving. The role involves developing high-performance ML runtimes for onboard and offboard systems, integrating ML compilers, and collaborating with ML practitioners. Requires expertise in C++, Python, JAX, and hardware accelerators.

Skills this posting asks for

• c++
• python
• pytorch
• jax
• tensorrt
• cuda
• xla
• triton
• openxla
• pjrt
• gpu
• tpu
• distributed systems
• low-latency
• backend systems
• model compression
• ml inference
• runtime optimization
• hardware accelerators
• profiling
• benchmarking
• deep learning frameworks
• ml compilers
• onboard compute

Requirements

• 5 years of experience
• Level: senior
• Remote policy: hybrid

From the employer’s posting

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve acce…