Machine Learning Engineer: Imitation and Reinforcement Learning for Robotics

The role

You will develop and scale learning architectures and build data systems for autonomous machines. This role involves designing, training, and deploying models for behavior cloning and reinforcement learning.

What you will do

• Design, train, validate, and launch models for behavior cloning and reinforcement learning
• Build and maintain data ingestion, labeling, and management pipelines
• Build metrics to evaluate model performance in open loop, simulation, and in the real world
• Collaborate with simulation, systems, and infrastructure teams to integrate ML models into real-world autonomous systems
• Deploy and debug models in real-world environments

Required qualifications

• 3+ years of practical experience applying Machine Learning with Deep Learning frameworks, such as PyTorch/Tensorflow/JAX to solve real-world problems
• 3+ years of professional experience building, deploying, and maintaining Machine Learning models in production environments
• Familiarity with recent literature and methods in learned behavior policies
• Practical experience in behavior cloning and/or reinforcement learning

Nice to have

• Experience with diffusion policies, Vision-Language-Action (VLA) models, or related technologies
• Published work in conferences such as ICRA, IROS, CoRL, CVPR, ECCV, ICCV, ICML, NeurIPS, …

About the company

Bedrock retrofits heavy equipment like excavators to enable autonomous work on live commercial job sites. The company has raised nearly $1B in total funding.