Robotics ML Training Engineer: End-to-End Scale

Weave Robotics

San Francisco (CA)

On-site

USD 180,000 - 250,000

Full time

14 days+

Application generator

Stand out for this role — generate a tailored resume and cover letter in about a minute.

Get past ATS filters

Job summary

Weave Robotics is seeking a senior ML systems engineer to own end-to-end training stack development, from raw fleet uploads to GPU-accelerated model training. You will optimize data pipelines, orchestrate experiments, and push production-grade infrastructure for terabyte-scale robot data in real homes and businesses.

The role emphasizes deep PyTorch/JAX expertise, multi-node distributed training, and performance tuning of GPUs, memory, and I/O.

Qualifications

Experience building ML training stacks for large-scale data
Ability to read loss curves and identify data bugs and instability
Hands-on experience with distributed training across multi-node clusters

Responsibilities

Build training stack end to end: distributed training, data loading, checkpointing, run orchestration, experiment tracking.
Large-scale data handling: ingests and process terabytes of multimodal robot data (video, proprioception, sensor streams).
Optimize research productivity by growing a scalable, reproducible codebase for experiments.
Sampling and curation decisions that affect model behavior.
Make runs fast: profile, fix bottlenecks, ensure reproducibility from data loading to GPU kernels.
Transition research prototypes into production infrastructure used by the team.