Machine Learning Engineer: Imitation and Reinforcement Learning for Robotics

The Role

We're seeking a Machine Learning Engineer with expertise in behavior learning to join our mission-critical team. Your responsibilities will include:

• Design, train, validate, and launch models for behavior cloning and reinforcement learning
• Build and maintain data ingestion, labeling, and management pipelines to ensure high-quality training datasets
• 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 these models in real-world environments, addressing practical issues such as latency, hardware constraints, and system integration

About the Role

At Bedrock, we're dedicated to advancing autonomy in the built world. Our team includes experts from leading tech companies like Waymo, Segment, and Uber Freight.

Responsibilities

• Design, train, validate, and launch models for behavior cloning and reinforcement learning
• Build and maintain data ingestion, labeling, and management pipelines to ensure high-quality training datasets
• 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 these models in real-world environments, addressing practical issues such as latency, hardware constraints, and system integration

Requirements

Requirements for this position include:

• 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

Qualifications

Qualifications for this role include:

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

Skills

Skills required for this position include:

• Deep understanding of machine learning and deep learning principles
• Experience with data engineering and data management
• Strong problem-solving skills and ability to work independently
• Excellent communication and collaboration skills

Benefits

Bedrock offers competitive benefits including:

• Flexible work arrangements
• Competitive compensation package
• Health, dental, and vision insurance
• 401(k) retirement plan
• Employee assistance program

Pay

Salary range for this position is $120,000 - $180,000 annually.

Schedule

Our ideal candidate is available full-time and works standard business hours.