Senior AI Data Infrastructure Engineer — Onsite, Equity

RoboForce; Milpitas (CA); On-site; USD 185,000 - 325,000; Full time

Benefits offered by this job

Stock options/equity

Health, dental, vision insurance

401(k) plan

Job summary

RoboForce in Milpitas, CA, seeks a Senior / Staff AI Research Engineer, Data Infrastructure to own the full data pipeline—from raw teleoperation data collection to post-training infrastructure that scores demonstrations and guides retraining. You will design end-to-end pipelines, annotate data, and manage storage across cloud and on-prem environments.

The role emphasizes collaboration with ML researchers and building scalable systems for multimodal robot data, with a competitive compensation

Qualifications

• Bachelor's or Master's degree in Computer Science, Robotics, or related field.
• 5+ years of experience in AI data infrastructure or equivalent.
• Strong Python proficiency and production-grade data pipelines.

Responsibilities

• Design end-to-end data collection pipelines for multimodal demonstration data.
• Build annotation tooling and data curation workflows for high-quality training datasets.
• Develop post-SFT reinforcement learning infrastructure and reward scoring.
• Build evaluation and test infrastructure for on-robot policy rollout logging.
• Collaborate with ML researchers to define data schemas and interfaces.
• Architect scalable storage and retrieval across cloud and on-prem environments.

Skills

Python

ETL pipelines

Data pipelines

Versioning

Large-scale data management

Education

Bachelor's or Master's in CS/Robotics

Tools

PyTorch

JAX

MLflow

Weights & Biases

WebDataset

Zarr

S3

GCS

Job description

RoboForce in Milpitas, CA, seeks a Senior / Staff AI Research Engineer, Data Infrastructure to own the full data pipeline—from raw teleoperation data collection to post-training infrastructure that scores demonstrations and guides retraining. You will design end-to-end pipelines, annotate data, and manage storage across cloud and on-prem environments.

The role emphasizes collaboration with ML researchers and building scalable systems for multimodal robot data, with a competitive compensation