Staff Software Engineer, Inference, ML Platform

March 20

Apply Now

Receive Emails with Similar Jobs

Waymo

WebsiteLinkedInAll Job Openings

Waymo is an autonomous driving technology company creating a new way forward in mobility.

1001 - 5000

💰 $2.5G Private Equity Round on 2021-06

Description

• Waymo is an autonomous driving technology company on a mission to improve mobility and save lives • The ML Platform team at Waymo develops tools for machine learning workflow automation and support • Focus on developing large-scale and efficient inference solutions for Waymo • Key to the development of Waymo's foundation models for driving • Collaborate with machine learning experts across the company and Alphabet • Develop scalable inference platforms for real-time and batch processing

Requirements

• B.S. in Computer Science, Math, or equivalent experience • 2+ years of experience as a technical lead • 5+ years working with high-scale distributed or ML inference systems • Understanding of machine learning fundamentals • Experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow • Experience in building scalable inference platforms • Lead software development using C++ or Python • Deploying on cloud platforms like AWS, Azure, or GCP • Preferred: MS in Computer Science or equivalent experience • Experience with multi-threaded and stream-based programming • Scaling model inference to multiple accelerators • Experience in ML compiler optimization • Deploying machine learning models for computer vision, NLP, or recommendation systems

Benefits

• Top-notch medical, dental, and vision insurance • Mental wellness support • Flexible Spending Account (FSA) and Health Saving Account (HSA) • Special wellness programs • Competitive compensation • Bonus opportunities • Equity • 401(k) plan • 1-on-1 financial coaching • 529 College Savings Plan • Employee discounts • Work flexibility • Remote working opportunities • Paid time off • Bereavement, sick, and parental leave • Paid parental leave • Subsidized backup childcare or adult/elder care • Access to fertility care or adoption support • Education reimbursement • Personal and professional development • Mentorship • Connection through Employee Resource Groups (ERGs) • Time off to volunteer • Access to Google offices, cafes, wellness centers, massages, and more • At-home fitness and cooking classes