The role involves building and maintaining AI infrastructure, optimizing training performance, and developing tools for robot training and data workflows.
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 green card support for qualified candidates.
- Lunches and dinners, a fully stocked kitchen, and regular team-building events.
Compensation: Salary $185,000–$325,000 USD + Bonus + Equity
The base salary range above represents the expected compensation for this full-time U.S. position. Final compensation will be determined based on role scope, level, location, job-related skills, experience, and relevant education or training, and may fall outside the listed range in exceptional cases.
RoboForce Milpitas, California, USA Office
Milpitas, California , United States, 95035
Similar Jobs
Cloud • Information Technology • Machine Learning
Lead the marimo engineering team across its open-source Python notebook ecosystem and molab cloud service. Manage and mentor engineers, guide architecture and technical direction, oversee staffing and releases, improve testing and CI/CD, and establish reliable production operations including observability, incident response, and capacity planning. The role also involves evaluating community contributions, translating user feedback into product improvements, and strengthening engineering practices as the team grows.
Top Skills:
APIsAutomated TestingCi/CdGrafanaKubernetesObservability SystemsPrometheusPython
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Architects and operates enterprise cryptographic security services for a multi-tenant SaaS platform. Responsibilities include certificate lifecycle management, key management, secrets distribution, PKI, Kubernetes deployment, threat modeling, incident response, compliance with FIPS and regulated-environment requirements, and evaluating AI/ML for security applications. The role also mentors engineers, collaborates across product and security teams, and advances post-quantum cryptography readiness.
Top Skills:
Ci/CdFipsGoHelmHsmJavaKmsKubernetesPkiTls
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Own the Now Engage platform strategy, roadmap, and end-to-end delivery. Lead AI-native workspace, agent, integration, and customer success capabilities while aligning executives, business teams, engineering, architecture, and product stakeholders. Define requirements, prioritize initiatives, manage tradeoffs, drive adoption, and measure outcomes such as customer time to value. Apply technical expertise in system design, architecture, data flows, APIs, workflows, and AI/ML to deliver enterprise platform experiences.
Top Skills:
Ai/MlAPIsEnterprise WorkflowsNow EngageServicenowStrategic Portfolio Management (Spm)System Integrations
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
