Robotics ML Engineer — Imitation & Reinforcement Learning

Bedrock Robotics; San Francisco (CA); Hybrid; USD 120,000 - 160,000; Full time

Job summary

A robotics company focusing on autonomy is seeking a Machine Learning Engineer to work on developing and scaling learning architectures and building data systems. In this role, you will design and deploy models for behavior cloning and reinforcement learning. The ideal candidate should have over 3 years of experience applying machine learning techniques, especially with frameworks like PyTorch and TensorFlow. Join a talented team dedicated to solving real-world problems in the construction industry.

Qualifications

• 3+ years of practical experience applying Machine Learning.
• 3+ years of experience building and deploying ML models in production.
• Familiarity with literature on learned behavior policies.

Responsibilities

• Design, train, validate, and launch models for behavior cloning.
• Build and maintain data ingestion pipelines for training datasets.
• Deploy and debug ML models in real-world environments.

Skills

Machine Learning with Deep Learning frameworks

Behavior cloning

Reinforcement learning

Data ingestion and management pipelines

Collaborative problem-solving

Tools

PyTorch

TensorFlow

JAX

Job description

A robotics company focusing on autonomy is seeking a Machine Learning Engineer to work on developing and scaling learning architectures and building data systems. In this role, you will design and deploy models for behavior cloning and reinforcement learning. The ideal candidate should have over 3 years of experience applying machine learning techniques, especially with frameworks like PyTorch and TensorFlow. Join a talented team dedicated to solving real-world problems in the construction industry.