Internship 2026 Machine Learning: World Models

Bedrock is seeking a motivated intern to help design and train a world model that powers simulation, evaluation, and training for autonomous excavation. Over the course of a 10-12 week internship, you’ll prototype architectures, mine real-world data, and develop generative representations to accelerate our autonomy stack. 

You’ll work closely with our Machine Learning team to build models that capture the underlying dynamics of complex environments. Expect hands-on mentorship, regular code reviews, and the opportunity to contribute directly to a foundational piece of Bedrock’s ML infrastructure.

We’re looking for students who are excited about world models, simulation, and autonomy.

Required:

• Currently pursuing a BS, MS, or PhD in machine learning or related technical field.
• Familiarity with recent world model literature (e.g., Dreamer-series, Genie, PlaNet, DINO-WM, etc.).
• Experience training and evaluating ML models end-to-end.
• First author on at least two peer-reviewed papers.
• Strong Python and deep learning skills. PyTorch (preferred), JAX, or tensorflow.
• Relevant GitHub portfolio showcasing your work.
• Strong documentation and communication skills.
• Ability to work ~10-12 weeks in our San Francisco office during Summer 2026.

Preferred:

• Familiarity with vehicle dynamics, stability analysis, or multibody simulation.
• Production-level coding experience.
• Robotics or simulation background.
• Experience in self-supervised learning.
• Publications in top-tier venues (NeurIPS, ICLR, ICML, ICCV, CVPR, etc.).
• Experience with reinforcement learning or simulation-based training.

Our roles are often flexible. If you don't fit all the criteria, or are in another location (especially one where we have an office like SF of NY) please apply anyway! We'd love to consider you. 

At Bedrock, we've assembled one of the most experienced autonomous technology teams in the industry, with deep expertise scaling breakthroughs across transportation, infrastructure, and enterprise software. Our leaders helped put the first self-driving cars on public roads at Waymo, scaled systems for Segment's $3.2B acquisition, and grew Uber Freight to $5B in revenue.

While others debate the future of AI, we're deploying it in the real world. Our systems are already installed on heavy machines across the country, learning on real construction sites and working to reshape the earth with survey-grade precision and exceptional safety. This isn't a simulation—it's autonomous intelligence working on billion-dollar infrastructure projects.

In just over a year, we've raised $80M, put our equipment into the field, and established partnerships with forward-thinking contractors who are integrating our technology into their operations. We're working quickly to close the gap between America's surging demand for housing, data centers, manufacturing hubs, and the construction industry's growing labor shortage.

Here, algorithms meet steel-toed boots. You'll collaborate with both construction veterans and experienced engineers, tackling problems where your work directly impacts how the physical world get built. If you're interested in applying cutting-edge technology to solve meaningful problems alongside a talented team—we'd love to have you join us.