Senior/Staff AI Research Engineer, Data Infrastructure

Responsibilities

• Design and maintain end-to-end data collection pipelines ingesting multimodal demonstration data
• Build annotation tooling and data curation workflows for high-quality training datasets
• Develop post-SFT reinforcement learning infrastructure including reward scoring and failure pattern mining
• Build evaluation and test infrastructure to log policy rollouts on-robot
• Collaborate with ML researchers to define data schemas and pipeline interfaces
• Architect scalable storage and retrieval systems for heterogeneous robot data

Requirements

• Bachelor's or Master's degree in Computer Science, Robotics, or related field
• 5+ years of experience
• Strong proficiency in Python
• Experience building production-grade data pipelines and ETL systems
• Hands-on experience with large-scale dataset management (S3, GCS, HDF5, WebDataset, Zarr)
• Experience with post-training infrastructure (SFT, reward modeling, or RL loops)
• Familiarity with PyTorch or JAX
• Ability to work in-office 5 days/week in Milpitas, CA

Preferred Qualifications

• Experience with robotics data collection hardware (teleoperation, UMI, GELLO)
• Familiarity with imitation learning, behavior cloning, or VLA/VLM fine-tuning
• Experience with experiment tracking (Weights & Biases, MLflow)
• Ability to design human-in-the-loop labeling systems

Benefits

• Competitive stock options/equity programs
• Health, dental, and vision insurance
• 401(k) plan
• Visa sponsorship and green card support
• Lunches, dinners, and fully stocked kitchen

About the Company

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.