March 20
🏢 In-office - Bay Area
• The ML Platform team at Waymo provides a set of tools to support and automate the lifecycle of the machine learning workflow, including feature and experiment management, model development, debugging and evaluation, deployment, and monitoring • Looking for engineers with an ML infra or GPU background to help improve compute performance on the car • Collaborate with ML practitioners to understand their models and modify model architectures to run faster on the car • Build tools to benchmark and analyze workloads • Build tools to improve model latency and quality • Build tools to productize deep learning models for onboard and offboard deployment
• BS in Computer Science, Mathematics or a related field • 5+ years of industry experience • C++ programming skills • Passion for developing and optimizing ML software stacks for modern architectures (framework, runtime library, ML compiler, efficient deep learning) • Working knowledge of system performance, GPU optimization or ML compiler • MS in Computer Science, Mathematics or a related field • Python programming skills • Experience with efficient deep learning techniques such as quantization, NAS, distillation • Experience with any ML compiler and IRs (Triton, HLO, MLIR, or Relay)
• Top-notch medical, dental and vision insurance • Mental wellness support • Flexible Spending Account (FSA) • Health Saving Account (HSA) • Special wellness programs • Competitive compensation • Bonus opportunities • Equity • Generous 401(k) plan • 1-on-1 financial coaching • 529 College Savings Plan • Other perks and employee discounts • Flexibility to work from another location for four weeks per year • On-site or hybrid work model • Remote working opportunities • Paid time off • Bereavement, sick, and parental leave • Paid parental leave (birthing parent gets 24 weeks of paid leave with up to 4 weeks of additional leave before their due date, and non-birthing parent gets 18 weeks of paid leave) • 20 days of subsidized backup childcare or adult/elder care • Access to fertility care or adoption support • Education reimbursement • Personal and professional development • Mentorship • Employee Resource Groups (ERGs) • Time off to volunteer • Access to Google offices, cafes, wellness centers, massages, and more • At-home fitness and cooking classes
Apply Now