RoboForce
Senior / Staff AI Research Engineer, Data Infrastructure
$185K–$325K
Milpitas, CA +1 · Staff · Full-time · Posted Dec 11, 2025
- Location
- Milpitas, CA
Milpitas, California, United States - Experience
- 5+ years of experience.
- Education
- Bachelor's or Master's degree in Computer Science, Robotics, or related field.
- Visa sponsorship
- Visa sponsorship and green card support for qualified candidates.
About the job
Builds the data collection, curation, storage, evaluation, and post-training infrastructure that supports RoboForce’s Physical AI and robot policy learning.
What you'll do
- Develop end-to-end pipelines for multimodal robot demonstration data, including synchronization, versioning, and distributed storage.
- Build dataset curation, annotation, and post-training systems to score demonstrations, analyze failures, and feed data into retraining.
- Create rollout evaluation infrastructure and collaborate with researchers on data schemas and scalable robot-data systems.
What they ask for
- Bachelor's or Master's degree in Computer Science, Robotics, or a related field.
- 5+ years of experience.
- Strong proficiency in Python and experience building production-grade data pipelines and ETL systems.
- Experience with large-scale dataset management and post-training infrastructure; familiarity with PyTorch or JAX and ML training workflows.
Nice to have
- Experience with robotics data collection hardware such as teleoperation devices, UMI, or GELLO.
- Familiarity with robot learning pipelines, evaluation or experiment tracking, or annotation tooling.
Pay
USD 185,000–325,000 base salary
Tools
Python, S3, GCS, HDF5, WebDataset, Zarr, PPO, DPO
AI focus
Builds data and post-training infrastructure for Physical AI, robot policy learning, reinforcement learning, and model retraining.
Model-written summary · the original posting is authoritative