2027 Internship Behavior ML Engineer, Learned Manipulation Policies

The role

You will train and iterate on diffusion flow matching and related behavior policies using real fleet demonstration data and simulated rollouts. You will also build evaluation pipelines and investigate failure modes for machine behavior.

What you will do

• Train and iterate on diffusion flow matching and related behavior policies using real fleet demonstration data and simulated rollouts
• Run architecture, conditioning, and hyperparameter explorations
• Build and improve evaluation pipelines for behavior models
• Investigate failure modes like distribution shift and mode collapse
• Work with simulation and eval teams to ensure offline metrics predict on-machine performance
• Contribute to the shared training codebase
• Share results regularly with behavior, controls, and autonomy teams

Required qualifications

• Currently pursuing a BS, MS, or PhD in computer science, robotics, machine learning, or a related field or bringing equivalent research or industry experience
• Strong Python and hands-on experience training models in PyTorch (or equivalent)
• Working understanding of generative modeling diffusion models, flow matching, VAEs, or similar and of imitation learning or behavior cloning
• Experience running and interpreting real training experiments
• Clear communication

Nice to have

• Published work or substantial project experience in diffusion, flow matching, robot learning, or imitation learning
• Experience training and building Vision Language Action (VLA) models
• Experience training Reinforcement Learning (RL) policies for robot manipulation
• Experience evaluating policies in simulation and reasoning about the sim-to-real gap
• Familiarity with large-scale training infrastructure and experiment tracking tooling
• Exposure to robotics, autonomous vehicles, or other physical-world control problems
• Interest in construction, earthwork, or heavy equipment

Before you apply

• Designed strictly for students (undergraduate or graduate)
• Opportunities for full-time conversion are highly limited

About the company

Bedrock retrofits heavy equipment like excavators to be autonomous. The team includes veterans from Waymo, Segment, and Uber Freight.