Senior / Staff AI Research Engineer, Data Infrastructure

Required

• 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.