Senior Software Engineer, Machine Learning, Infra & Eval jobs in United States
info-icon
This job has closed.
company-logo

Waymo · 7 months ago

Senior Software Engineer, Machine Learning, Infra & Eval

Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. The Senior Software Engineer in Machine Learning will collaborate with various stakeholders to deploy machine learning models and ensure system reliability while developing CI/CD pipelines and implementing best practices in model governance.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
check
H1B Sponsor Likelynote

Responsibilities

Collaborate with data scientists, engineers, and other stakeholders to deploy machine learning models into production
Develop and maintain CI/CD pipelines for automating data generation, model training, testing, and deployment processes
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, Beam etc)

Qualification

Machine LearningPythonMLOpsC++CI/CD PipelinesGCP/AWS/KubernetesTensorFlowJAXPyTorchTF ServingTorchserveKubeflowSagemaker PipelinesVertex AI PipelinesComputer VisionNatural Language ProcessingRecommendation Systems

Required

B.S. in Computer Science, Math, or equivalent real-world experience
5+ years building and maintaining high-scale distributed or ML inference systems
Coding and testing skills, specifically Python/C++
Familiarity with large-scale fleet management, testing, and deployment, e.g.: GCP/AWS/Kubernetes, Jenkins, GCS/S3, etc
Familiarity with large-scale MLOps tools, e.g.: tf.serving/Torchserve, Kubeflow/Sagemaker Pipelines/Vertex AI Pipelines, etc
Understanding of machine learning fundamentals and experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow

Preferred

Previous experience deploying and supporting machine learning models for computer vision, natural language processing, or recommendation systems

Benefits

Discretionary annual bonus program
Equity incentive plan
Generous Company benefits program

Company

Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. It is a sub-organization of Alphabet.

H1B Sponsorship

Waymo has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (231)
2024 (175)
2023 (268)
2022 (306)
2021 (298)
2020 (317)

Funding

Current Stage
Late Stage
Total Funding
$11.1B
Key Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M

Leadership Team

leader-logo
Tekedra Mawakana
Co-Chief Executive Officer
linkedin
leader-logo
Annabel Chang
Head of State Policy & Government Relations
linkedin
Company data provided by crunchbase