Skip to main content

ML Infrastructure Engineer

ziplineMillbrae, CALead L6
Overview: As ML Training & Inference Infrastructure Engineer on Zipline's Data Platform, you’ll build and scale data and ML systems that fuel our autonomous logistics. You’ll work at the intersection of autonomy and infrastructure, delivering reproducible, observable pipelines for model training, evaluation, and deployment. Your work enables faster iteration and safer, more capable learning for our large-scale fleet. This role offers high-impact work that directly supports life-saving deliveries and rapid growth. Compensation / Benefits equity compensation discretionary bonuses medical, dental, and vision insurance paid time off
Responsibilities: Build and operate infrastructure enabling learning algorithms from Zipline’s fleet data Design scalable data and ML pipelines for dataset creation, validation, training, evaluation, and deployment Own and improve data pipelines feeding the ML development loop Identify and mitigate bottlenecks in ML development, focusing on orchestration, performance, and reproducibility
Key requirements: 3+ years of professional software engineering experience; ML infrastructure, data infrastructure, robotics, autonomy, or safety-critical environments Strong Python production practices; API/service/schema design and operational workflows Experience building reproducible data and ML pipelines Experience monitoring data statistics, system performance, pipeline failures, and model signals Working knowledge of ML concepts (datasets, training, evaluation, optimization, statistics) and modern deep learning workflows Generalist mindset across cloud services, data platforms, tooling, and embedded/robotics constraints Experience with PyTorch or similar ML frameworks Strong ownership and communication; building secure systems for mission-critical workflows Experience with Kubernetes or similar container orchestration Experience with cloud and on-premise production infrastructure (prefer AWS) and IaC tools like Terraform or CloudFormation strong ownership clear communication cross-functional collaboration Python production software development ML infrastructure and data pipelines PyTorch or similar frameworks
APPLY NOW (opens in a new tab)

Matching similar jobs

Level

Lead L6

Location

Millbrae, CA

Occupation

Computer Systems Engineers/Architects

Industry

Computer Systems Design Services

Posted

today

APPLY NOW (opens in a new tab)