2027 Internship Behavior ML Engineer, Learned Manipulation Policies
San Francisco, CAInternPosted 2w ago
About the role
Bedrock Robotics is seeking an internship in Behavior ML focused on learned manipulation policies for heavy equipment. The role centers on training diffusion-based policies for a 40-ton excavator, evaluating performance, and collaborating with simulation and evaluation teams to ensure offline metrics translate to on-machine success.
Responsibilities
- 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, and turn the results into clear findings the team can build on
- Build and improve evaluation pipelines that score behavior models on task success, smoothness, safety margins, and operator-likeness
- Investigate failure modes - distribution shift, mode collapse, out-of-distribution scenes and propose fixes
- Work with the simulation and eval teams to make sure offline metrics actually predict on-machine performance
- Contribute to the shared training codebase with clean, reviewed, reproducible work
- Share results regularly with the behavior, controls, and autonomy teams
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: you know how to tell a real improvement from noise
- Clear communication, you can explain what you tried, what happened, and what you'd do next
Skills
Required:PythonPyTorchGenerative modeling / diffusion modelsImitation learning / behavior cloningExperiment design and analysisCommunication
Benefits
- Fixed-term, project-based internship
- Opportunities to contribute to cutting-edge autonomous systems for construction equipment
- Hands-on skill development and resume-building experience
- Exposure to multiple teams (behavior, controls, autonomy) and real-world impact
Last updated Oct 15, 2026. Interested? Apply on the company site.
About Bedrock Robotics
Advanced Autonomy for the Built World
Advanced Autonomy for the Built World
AutonomyBedrock Operatorautonomyheavy machineryindustrial automationindustrial roboticsmachine intelligenceon-site autonomy
More open roles at Bedrock Robotics
- Hydraulics System Engineer4d agoHybrid (San Francisco, CA) · Engineering
- Electrical Systems Integration Engineer4d agoHybrid (San Francisco, CA), Hybrid (Lockhart, TX) · Engineering
- Bedrock Training & Certification Lead4d agoHybrid (San Francisco, CA), Hybrid (Lockhart, TX) · Operations
- Cloud - Staff Site Reliability Engineer4d agoHybrid (New York, NY), Hybrid (San Francisco, CA) · Engineering
- Head of Bedrock Operations Center1w agoHybrid (San Francisco, CA), Hybrid (Lockhart, TX) · Operations