Machine Learning Engineer at Bedrock

Setup: Remote, Hybrid
Location: San Francisco, California
Type: Full-time
Posted: Jan 31, 2026

Description

Bedrock is seeking a Machine Learning Engineer specializing in behavior learning, focusing on data-driven behavior policies and robust data infrastructure. The role involves designing, training, validating, and launching models for behavior cloning and reinforcement learning, building data pipelines, evaluating model performance, integrating ML models into autonomous systems, and deploying/debugging models in real-world environments.

Required Skills

behavior learning
data-driven behavior policies
robust data infrastructure
learning architectures

Additional Information

Required Previous Experiences: 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
Preferred Qualifications: Familiarity with recent literature and methods in learned behavior policies
Practical experience in behavior cloning and/or reinforcement learning
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, …