Hiring: Staff Machine Learning Infrastructure Engineer Company: Atoms Location: San Francisco, CA Job Posted Time: 2026-09-10 10:59:36 Employment Type: Full-time Target Skills & Keywords : Data Pipeline, Integration Testing, Java, Kubernetes, MLOps, MLflow, Machine Learning, Node.js, Python, Rust About the job Experience: •8+ years of professional software engineering career experience Required Skills: •Design the high-performance training pipelines and validation environments that enable our world-class robotics and ML researchers to iterate rapidly. •Own the challenge of scaling distributed GPU workloads to support a high volume of concurrent training runs across an expanding vehicle fleet, building a platform that can flexibly run on whatever GPU capacity is available, regardless of provider or environment, directly accelerating innovation across the platform. Compensation: •$224,000 - $280,000 / year •You may also be eligible for equity awards Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!