← All roles

Senior / Staff AI Research Engineer, Data Infrastructure

RoboForce· AI
Apply on company site
Milpitas, CA$185k - $325kfull-timePosted 3mo ago

About the role

Senior / Staff AI Research Engineer responsible for building the data and learning engine behind RoboForce's Physical AI stack, owning end-to-end data pipelines from raw data collection to post-training infrastructure and feedback loops for model retraining.

Responsibilities

  • Design and maintain end-to-end data collection pipelines ingesting multimodal demonstration data from teleoperation devices and UMI hardware, including synchronization, versioning, and distributed storage at scale.
  • Build annotation tooling and data curation workflows — quality filtering, deduplication, episode scoring, and domain reweighting — to produce high-quality training datasets for robot policy learning.
  • Develop post-SFT reinforcement learning infrastructure: implement reward scoring on demonstrations, mine and categorize failure patterns, and feed curated failure data back into the retraining loop.
  • Build evaluation and test infrastructure to log policy rollouts on-robot, capture structured results, and surface actionable diagnostics for the research team.
  • Collaborate with ML researchers to define data schemas, episode formats, and pipeline interfaces that support rapid iteration on VLA and manipulation policy training.
  • Architect scalable storage and retrieval systems for heterogeneous robot data (vision, proprioception, action, language) across both cloud and on-prem environments.

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.

Skills

Required:PythonData pipelines and ETLLarge-scale dataset managementPost-training infrastructureDeep learning frameworksCollaboration with research

Benefits

  • Competitive stock options/equity programs.
  • Health, dental, and vision insurance, 401(k) plan.
  • Visa sponsorship and green card support for qualified candidates.
  • Lunches and dinners, a fully stocked kitchen, and regular team-building events.
Last updated Aug 13, 2026. Interested? Apply on the company site.

About RoboForce

Robotics for Humanity

RoboForce is building the world’s most advanced Robo-Labor system.

AI RoboticsIndustrial automationRobo-LaborRobotics for Humanityautomationoutsourcingphysical-ai-funding-landscaperobotics
Company profileWebsite

More open roles at RoboForce