Job Description
📋 Description Architect and develop an efficient, high-performance ML runtime and serving system tailored for Lead integration and feature development for ML inference runtimes across domains, balancing Drive migration of ML workloads toward a JAX-native runtime architecture (OpenXLA/PjRT, TensorRT Collaborate with Waymo ML teams to analyze system-level ML workloads and apply hardware-aware Design robust tooling for profiling, benchmarking, and bottleneck identification across the ML 🎯 Requirements B.S. or M.S. in CS, EE, Deep Learning or related field 5+ years software engineering experience building, scaling, or maintaining ML systems 5+ years production programming in C++ 3+ years production experience in Python and major DL frameworks (e.g., PyTorch, JAX) Experience optimizing ML software for hardware accelerators (GPUs, TPUs, custom silicon) Experience building low-latency, highly concurrent distributed backend systems 🎁 Benefits The salary range is $213,000 - $263,000 USD, with bonus/equity benefits as described by Waymo Waymo employees are eligible for discretionary annual bonus, equity incentive plan, and benefits
