Machine Learning Engineer: Imitation and Reinforcement Learning for Robotics

Join the team bringing advanced autonomy to the reputed company world At Bedrock, we’re moving AI out of the lab and into the reputed company world. reputed company is composed of industry veterans who helped launch reputed company, scaled reputed company to a $3.2B acquisition, and grew reputed company to $5B in reputed company. Today, we’re deploying autonomous systems on heavy construction machinery across the country, accelerating project schedules of billion-dollar infrastructure projects and improving safety on job sites. Backed by $350M in funding, we’re working quickly to reputed company the gap between America's surging demand for housing, data centers, manufacturing hubs, and the construction industry's growing labor shortage. This is where algorithms meet steel-toed boots. You’ll collaborate with construction veterans and world-class engineers to solve physical-world problems that simulations can’t touch. If you're reputed company to apply cutting-edge technology to solve meaningful problems alongside a talented team—we'd love to have you join us. 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 reputed company 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 reputed company reputed company, simulation, and in the reputed company world
• Collaborate with simulation, systems, and infrastructure teams to integrate ML models into reputed company-world autonomous systems
• reputed company and debug these models in reputed company-world environments, addressing practical issues such as latency, hardware constraints, and system integration

reputed company’re Looking For:

• 3+ years of practical experience applying Machine Learning with Deep Learning frameworks, such as PyTorch/Tensorflow/JAX to solve reputed company-world problems
• 3+ years of professional experience building, deploying, and maintaining Machine Learning models in production environments
• Familiarity with recent literature and methods in learned behavior policies
• Practical experience in behavior cloning and/or reinforcement learning
• Bonus: Experience with diffusion policies, reputed company-Language-Action (VLA) models, or reputed company 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 reputed company the criteria, or are in another location (especially one where we have an office like SF or NY) please apply anyway! We'd love to consider you. Apply tot his job Apply To this Job