Founding ML Infra Engineer for Robotics Training Platform

Reflection Robotics; Seattle (WA); On-site; USD 140,000 - 200,000; Full time

Job summary

Reflection Robotics seeks a founding infrastructure engineer to create the base platform for training robotics foundation models. You will design a self-serve interface to launch training jobs across clouds, and own the orchestration that decides run destinations for efficiency and performance.

You will optimize GPU utilization, data pipelines, and checkpointing, while ensuring reliability and cost-efficiency. In-person Seattle work is required.

Qualifications

• Strong systems engineering fundamentals used daily by engineers or robotics teams.
• Experience with distributed training and large-scale ML workloads.
• Proficiency with PyTorch.
• Experience across AWS, GCP, Azure and cost/performance tradeoffs.
• Comfort building and operating orchestration or scheduling systems.
• Track record shipping infrastructure that holds up under daily use.
• Ability to work in-person in Seattle on weekdays.

Responsibilities

• Build a streamlined, self-serve way for robotics engineers to launch training jobs, abstracting away the underlying cloud provider or cluster
• Design and build the orchestration layer that schedules and manages training jobs across multiple cloud providers
• Continuously monitor and optimize training jobs for cost, GPU utilization, and throughput
• Identify and eliminate bottlenecks in data loading, checkpointing, and distributed training that waste compute
• Build tooling to make it easy to compare performance and cost across providers and hardware types
• Set up monitoring and alerting so failed or underperforming jobs are caught quickly, not discovered days later
• Work closely with the robotics engineering team to understand their workflows and remove friction wherever it shows up

Skills

Systems engineering

Distributed ML

PyTorch

Multi-cloud

Orchestration

Reliability

On-site Seattle

GPU Utilization

Cost optimization

Tools

Kubernetes

Job description

Reflection Robotics seeks a founding infrastructure engineer to create the base platform for training robotics foundation models. You will design a self-serve interface to launch training jobs across clouds, and own the orchestration that decides run destinations for efficiency and performance.

You will optimize GPU utilization, data pipelines, and checkpointing, while ensuring reliability and cost-efficiency. In-person Seattle work is required.