Apply Now

Senior / Staff AI Research Engineer, Data Infrastructure

Milpitas, CA, US • Posted 8 days ago • Updated 1 day ago

Full Time

On-site

Fitment

Dice Job Match Score™

📋 Comparing job requirements...

Job Details

Skills

FOCUS
Scalability
Artificial Intelligence
Demonstrations
Interfaces
Cloud Computing
Computer Science
Python
Extract
Transform
Load
Management
Data Deduplication
Storage
Amazon S3
GCS
Modeling
PPO
Deep Learning
PyTorch
JAX
Machine Learning (ML)
Training
Research
Collaboration
Robotics
Data Collection
Computer Hardware
Workflow
Evaluation
Insurance
Team Building

Summary

Why RoboForce

RoboForce is an AI robotics company developing Physical AI-powered Robo-Labor for dull, dirty, and dangerous work. The company's robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability.

We are looking for a Senior / Staff AI Research Engineer, Data Infrastructure to build the data and learning engine behind RoboForce's Physical AI stack. In this role, you will own the full pipeline - from raw teleoperation and UMI device data collection through curation, annotation, and storage, to post-training infrastructure that scores demonstrations, identifies failure patterns, and closes the loop back into 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.

Requirements

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.

Bonus Qualifications

Experience with robotics data collection hardware - teleoperation devices, UMI, GELLO, or similar - and the synchronization and preprocessing challenges they introduce.
Familiarity with robot learning pipelines: imitation learning, behavior cloning, or VLA/VLM fine-tuning workflows.
Experience building evaluation or experiment tracking infrastructure (e.g., Weights & Biases, MLflow, custom rollout loggers).
Proven ability to design annotation tooling or human-in-the-loop labeling systems for structured or multimodal data.

Benefits

Competitive stock options/equity programs.
Health, dental, and vision insurance, 401(k) plan.
Visa sponsorship and support for qualified candidates.
Lunches and dinners, a fully stocked kitchen, and regular team-building events.

Employers have access to artificial intelligence language tools (“AI”) that help generate and enhance job descriptions and AI may have been used to create this description. The position description has been reviewed for accuracy and Dice believes it to correctly reflect the job opportunity.

Dice Id: 80183818
Position Id: d3f803ce433375ca2a200e424cc35235
Posted 8 days ago

Create job alert

Never miss an opportunity! Create an alert based on the job you applied for.

Milpitas, California

•

Yesterday

Why RoboForce RoboForce is an AI robotics company developing Physical AI-powered Robo-Labor for dull, dirty, and dangerous work. The company's robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability. We are looking for a Senior / Staff AI Research Scientist, Foundation Models to advance robotic embodied intelligence. In this role, you will develop algorithms that enable robots to understand their environment, interpret and execute task

Full-time

Senior Research Engineer, Training Data Infrastructure in Foundation Models

Cupertino, California

•

Yesterday

We build frontier foundation models that power intelligent experiences at Apple. Our team works across the full training lifecycle: including pre-training foundation models, and developing mid-training approaches that bridge general capability and task-specific performance. What makes our work distinct is that we're engineering models specifically for Apple silicon and optimized for experiences that are private, personal, and deeply integrated into the OS. We're solving frontier problems in rewa

Full-time

ML Infrastructure Engineer

Redwood City, California

•

Yesterday

Join Us in Building the Future of Home Robotics At Sunday, we're developing personal robots to reclaim the hours lost to repetitive tasks. We're focused on an ambitious goal to make generalized robots broadly accessible, enabling households to take back quality time. We have spent the last 18 months building a talented team, securing capital, and validating our technology. We are now seeking passionate individuals to join us in the next phase of our growth. If you are ready to apply your skill

Full-time

Senior/Staff Software Engineer, ML Data Infrastructure

Mountain View, California

•

Yesterday

Who We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver , to support a wide range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology proven over years of self-driving deployments, Nuro gives the automakers and mobility platforms

Full-time

USD 193,930.00 - 352,290.00 per year

Search all similar jobs

More jobs at RoboForce in Milpitas, CA

Senior / Staff AI Research Engineer, Data Infrastructure

Dice Job Match Score™

Job Details

Skills

Summary

Similar Jobs