Job Details

The Role

We're looking for a Machine Learning Engineer with a focus on behavior learning, specifically data-driven behavior policies and robust data infrastructure. In this role, you'll be responsible for developing and scaling state-of-the-art learning architectures, while also building the data systems that make these models reliable, scalable, and reproducible in production.

The Role

We're looking for a Machine Learning Engineer with a focus on behavior learning, specifically data-driven behavior policies and robust data infrastructure. In this role, you'll be responsible for developing and scaling state-of-the-art learning architectures, while also building the data systems that make these models reliable, scalable, and reproducible in production.

What you'll do:

• Design, train, validate, and launch models for behavior cloning and reinforcement learning
• Build and maintain data ingestion, labeling, and management pipelines to ensure high-quality training datasets
• Build metrics to evaluate model performance in open loop, simulation, and in the real world
• Collaborate with simulation, systems, and infrastructure teams to integrate ML models into real-world autonomous systems
• Deploy and debug these models in real-world environments, addressing practical issues such as latency, hardware constraints, and system integration
  
  

What we're looking for:

• Practical experience applying machine learning with deep learning frameworks, such as PyTorch, to solve real-world problems
• Proficiency in Python and comfort with at least one systems language (e.g., C++, Rust)
• Familiarity with recent literature and methods in learned behavior policies
• Practical experience in behavior cloning and/or reinforcement learning
• Bonus: Experience with diffusion policies, Vision-Language-Action (VLA) models, or related technologies
• Bonus: Published work in conferences such as ICRA, IROS, CoRL, CVPR, ECCV, ICCV, ICML, NeurIPS, …
  
  

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.

Join the team bringing advanced autonomy to the built world

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.

Seniority level

• Seniority level

  Not Applicable

Employment type

• Employment type

  Full-time

Job function

• Job function

  Engineering and Information Technology

Referrals increase your chances of interviewing at Bedrock Robotics by 2x

Get notified about new Machine Learning Engineer jobs in San Francisco, CA

San Francisco, CA $130,000.00-$230,000.00 6 months ago

San Francisco, CA $140,670.00-$195,400.00 19 hours ago

Research Engineer - Machine Learning (ML)

Machine Learning Engineer (I, II, or Sr.)

San Francisco, CA $140,000.00-$180,000.00 5 months ago

San Francisco, CA $140,000.00-$160,000.00 5 months ago

Redwood City, CA $123,000.00-$185,000.00 5 days ago

San Francisco, CA $140,000.00-$215,000.00 1 month ago

Machine Learning Scientist, NLP (All Levels)

San Francisco, CA $200,000.00-$300,000.00 4 months ago

Machine Learning Engineer (I, II, or Sr.)

Perception Software or Machine Learning Engineer

San Francisco, CA $140,000.00-$160,000.00 4 months ago

Machine Learning Scientist, NLP (All Levels)

San Francisco, CA $200,000.00-$300,000.00 4 months ago

San Francisco, CA $150,000.00-$260,000.00 4 months ago

San Francisco, CA $100,000.00-$180,000.00 1 year ago

San Francisco, CA $100,000.00-$300,000.00 5 days ago

San Francisco, CA $88,000.00-$140,000.00 1 month ago

San Francisco, CA $140,000.00-$290,000.00 7 months ago

San Francisco, CA $150,000.00-$250,000.00 1 month ago

San Francisco, CA $115,000.00-$185,000.00 2 hours ago

Machine Learning Engineer, AI (FULLY REMOTE)

San Francisco, CA $176,600.00-$225,900.00 5 days ago

San Francisco, CA $190,000.00-$355,000.00 3 weeks ago

San Mateo, CA $140,000.00-$210,000.00 1 month ago

San Francisco, CA $175,000.00-$250,000.00 2 weeks ago

San Francisco, CA $175,000.00-$225,000.00 8 months ago

San Francisco, CA $140,000.00-$200,000.00 2 weeks ago

Machine Learning Engineer, GenAI Applied ML

San Francisco, CA $176,000.00-$220,000.00 1 month ago

San Francisco, CA $100,000.00-$300,000.00 4 weeks ago

We're unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

J-18808-Ljbffr