Internship 2027 Behavior Machine Learning Engineer, World Models

Bedrock Robotics develops autonomous systems for heavy construction equipment and deploys them in real-world construction environments. The Behavior ML intern will train and evaluate world models using fleet data, explore model architectures, build evaluation metrics, and collaborate with behavior and controls teams to deploy models on excavators.

What you'll do:

• Train and evaluate world models on real fleet data collected from autonomous machines in the field
• Run architecture explorations and ablations, and make a defensible case for what's actually working
• Build evaluation harnesses and metrics that tell us whether a model is genuinely better
• Partner with the behavior and controls teams to connect model outputs to planning and control
• Present your findings and provide a solid foundation for the team to build upon

What they're looking for:

• Currently pursuing a BS, MS, or PhD in computer science, machine learning, robotics, or a related field — or bringing equivalent research or industry experience
• Strong Python and hands-on experience training models in PyTorch (or equivalent)
• Ability to read a paper and turn it into a working implementation
• Comfort with messy, real-world data — this is not a curated benchmark
• Sound experimental judgment understanding the difference between a real result and noise
• Experience with world models, video prediction, or sequence modeling (e.g., DreamerV3, V-JEPA 2, Genie 3, GAIA-2, NVIDIA Cosmos, etc.)
• Background in reinforcement learning or imitation learning
• Prior work with robotics or autonomous vehicle data
• Familiarity with large-scale training infrastructure

Benefits:

• Hybrid work arrangement