Senior AI Data Infrastructure Engineer — Robotics & RL

RoboForce; Milpitas (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Stock options

Health, dental, vision insurance

401(k) plan

Visa sponsorship

Team lunches and events

Job summary

RoboForce is seeking a Senior / Staff AI Research Engineer, Data Infrastructure, to own the data and learning engine behind its Physical AI stack. You will build end-to-end data pipelines, annotation workflows, and post-training infrastructure to score demonstrations and guide retraining.

You will work with ML researchers to define data schemas, manage multimodal robot data, and ensure scalable storage across cloud and on-prem. This is a full-time, in-office role based in Milpitas, CA.

Qualifications

• Bachelor's or Master's degree in Computer Science, Robotics, or related field with 5+ years of experience.
• Strong proficiency in Python and experience building production-grade data pipelines and ETL systems.
• Hands-on experience with large-scale dataset management, including versioning, deduplication, quality filtering, and distributed storage (e.g., S3, GCS, HDF5, WebDataset, Zarr).
• Experience building or working with post-training infrastructure — SFT pipelines, reward modeling, or RL training loops (e.g., PPO, DPO, rejection sampling).
• Familiarity with deep learning frameworks (PyTorch, JAX) and ML training workflows sufficient to collaborate tightly with research teams.
• Requires 5 days/week in-office collaboration with the teams.

Responsibilities

• Design and maintain end-to-end data collection pipelines ingesting multimodal demonstration data from teleoperation devices and UMI hardware.
• Build annotation tooling and data curation workflows to produce high-quality training datasets for robot policy learning.
• Develop post-SFT reinforcement learning infrastructure and feed curated failure data back into retraining loops.
• Build evaluation and test infrastructure to log policy rollouts on-robot and surface diagnostics for the team.
• Collaborate with ML researchers to define data schemas, episode formats, and pipeline interfaces for rapid iteration on policy training.
• Architect scalable storage and retrieval systems for heterogeneous robot data across cloud and on-prem environments.

Skills

Python

ETL pipelines

Dataset management

PyTorch

JAX

SFT pipelines

Reinforcement learning

Collaboration with ML researchers

Education

Bachelor's or Master's in CS/Robotics or related field

Tools

Weights & Biases

MLflow

Job description

RoboForce is seeking a Senior / Staff AI Research Engineer, Data Infrastructure, to own the data and learning engine behind its Physical AI stack. You will build end-to-end data pipelines, annotation workflows, and post-training infrastructure to score demonstrations and guide retraining.

You will work with ML researchers to define data schemas, manage multimodal robot data, and ensure scalable storage across cloud and on-prem. This is a full-time, in-office role based in Milpitas, CA.