Senior Software Engineer, Machine Learning, Reliability Infrastructure

In this role, you'll:

• Report to the Model Training and Inference Reliability TLM
• Design and implement scalable, reliable, and efficient ML infrastructure and pipelines.
• Collaborate with data scientists, engineers, and other partners to deploy machine learning models into production.
• Develop CI/CD pipelines for automating data generation, model training, testing, and deployment processes.
• Monitor and optimize model performance, scalability, and reliability in production environments.
• Implement best practices for model versioning, monitoring, and governance.
• Troubleshoot and resolve issues related to model deployment and performance.
• Stay current with emerging technologies and trends in ML Ops and DevOps domains.
• Be aware of best practices used in the Alphabet stack of ML technologies (e.g. TF, JAX, Flax, or Beam).

At a minimum, we’d like you to have:

• B.S. in Computer Science or Math, or equivalent real-world experience
• 5+ years building and maintaining high-scale distributed or ML inference systems
• Code and testing skills, specifically Python/C++
• Familiarity with large-scale fleet management, testing, and deployment (e.g. GCP/AWS/Kubernetes, Jenkins, GCS/S3)

It's preferred if you have:

• Previous experience deploying and supporting machine learning models for computer vision, natural language processing, or recommendation systems
• Understanding of machine learning fundamentals and experience with popular ML frameworks such as JAX, PyTorch, or Tensorflow

While at Waymo, you will enjoy benefits that cover…

Health and wellness: Our people are at the heart of everything we do. At Waymo, you can enjoy top-notch medical, dental and vision insurance, mental wellness support, a Flexible Spending Account (FSA), a Health Saving Account (HSA), and special wellness programs.

Financial wellness: Your financial peace of mind is important to us. At Waymo, we offer competitive compensation, bonus opportunities, equity, a generous 401(k) plan, 1-on-1 financial coaching, a 529 College Savings Plan and lots of other perks and employee discounts.

Flexibility and time off: Take the time you need to relax and recharge. Enjoy the flexibility to work from another location for four weeks per year. We support an on-site or hybrid work model and offer remote working opportunities, paid time off, bereavement, sick, and parental leave.

Supporting families: When it comes to growing your family or caring for your loved ones, you have our full support. Enhanced leave options include 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), and 20 days of subsidized backup childcare or adult/elder care.. Access to fertility care or adoption support as you grow your family.

Community and personal development: At Waymo, you'll find a range of opportunities to grow, connect, and give back. We offer education reimbursement, personal and professional development, mentorship, and other ways to connect through Employee Resource Groups (ERGs), other internal groups, and even time off to volunteer.

Cool perks: Access to Google offices, cafes, wellness centers, massages, and so much more. To support your wellbeing at home, you can enjoy at-home fitness and cooking classes, and more.

#LI-Hybrid