Senior Research Engineer, ML Data Pipelines Senior Research Engineer, ML Data Pipelines

About the Role

Key Responsibilities

• Design, implement, and optimize scalable ML data pipelines for training multimodal foundation models.
• Collaborate closely with researchers to preprocess, transform, and manage large datasets for robot model training and evaluation.
• Develop tools for labeling and curating multiple streams of sensor data.
• Continuously monitor robot data collection processes and evaluate data quality.
• Implement and optimize PyTorch data loading modules for video processing and robot learning on large GPU clusters.

Requirements

• Bachelor's Degree in Computer Science, Robotics, Engineering, or a related field.
• 10+ years of full-time industry experience working with large-scale machine learning data pipelines.
• Proficiency in Python data processing libraries.
• Hands-on model training experience in PyTorch, JAX, or TensorFlow.
• Strong experience with large-scale GPU clusters, HPC environments, and job scheduling/orchestration tools (e.g., SLURM, Kubernetes).

Nice to Have

• Master's or PhD degree in Computer Science, Robotics, Engineering, or a related field.
• Strong experience with cloud infrastructure management (AWS, Azure, GCP) and data stores (Postgres, MySQL, ElasticSearch, Redis).
• Experience at autonomous driving or robotics companies training machine learning models on massive datasets.
• Demonstrated Tech Lead experience, coordinating a team of engineers and driving projects from conception to deployment.
• Contributions to popular open-source frameworks.

Qualifications

• Educational background in Computer Science, Robotics, Engineering, or related fields.
• Extensive industry experience (10+ years) in ML data pipelines.

Benefits & Perks

• Base salary range of $224,000 to $425,500, determined based on location, experience, and comparable roles.
• Eligibility for equity and other compensation benefits.