Machine Learning Engineer, ML Resources

The ML Ops team, part of Waymo ML Platform team, builds tools and infrastructure to realize the ML flywheel at Waymo. This includes building automation and orchestration solutions to make complex ML workflows manageable and reliable. This team also partners closely with the modeling team to realize solutions to speed up developer velocity.

We're looking for a software engineer to join the team to build and maintain the critical data and ML pipelines that powers ML development at Waymo.

In this hybrid role, you will report to a Technical Lead Manager, Staff Software Engineer.

You will:

• Analyze and report on ML resource (compute, storage, accelerators) usage, costs, and efficiency trends across different teams and projects.
• Develop, maintain, and improve dashboards and tools for monitoring key resource metrics and providing actionable insights.
• Identify and investigate opportunities for resource optimization, cost reduction, and performance improvements in ML workflows.
• Support and guide ML engineers and researchers on best practices for resource utilization
• Contribute to the development, documentation, and enforcement of resource management policies and best practices.

You have:

• Bachelor's degree in Computer Science, Engineering, or related field, and 2+ years equivalent experience
• Experience with distributed systems principles and experience building distributed systems for production environments.
• Solid Python or C++ skills
• Experience monitoring, debugging, and troubleshooting complex distributed systems
• Experience communicating updates and resolutions to customers and other partners

We prefer:

• Experience with compute and storage management for medium to large organizations

#LI-Hybrid