← Search

Behavior Machine Learning Engineer Intern - World Models

Bedrock Robotics · AI/ML/Data

Location
San Francisco, CA
Metro
San Francisco Bay Area
Workplace
On-site
Level
Intern
Type
Internship
Posted
1 month ago
Source
Simplify internships list

Read the full posting & apply ↗

PythonPyTorch

What the posting covers

Join the team bringing advanced autonomy to the built world

At Bedrock, we're moving AI out of the lab and into the real world. Our team includes veterans who helped launch Waymo, scaled Segment to a $3.2B acquisition, and grew Uber Freight to $5B in revenue. Today, we're deploying autonomous systems on heavy…

About the Role & Team

Predicting what happens next on a construction site is one of the hardest and most valuable problems at Bedrock, whether that's soil dynamics, machine interactions, or the people moving around the site. Every scoop of dirt we move rewrites the scene we have…

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…

Required

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…

Preferred

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…