Machine Learning Engineer: Imitation and Reinforcement Learning for Robotics
Core
Developing and scaling state-of-the-art learning architectures for behavior cloning and reinforcement learning to enable autonomous systems on heavy construction machinery.
Role type
Machine Learning Engineer (Behavior Learning & Reinforcement Learning)
Builds
Autonomous control systems for heavy construction machinery
Domain
Robotics, Construction, Autonomous Systems
Deliverable
production ML models
Required skills
Behavior cloning, Reinforcement learning, Deep Learning frameworks (PyTorch/Tensorflow/JAX), Data pipeline engineering, Model evaluation in simulation and real-world, System integration, Latency optimization
Preferred skills
Diffusion policies, Vision-Language-Action (VLA) models, Publications in ICRA/IROS/CoRL/CVPR/ECCV/ICML/NeurIPS
Technologies
PyTorch, TensorFlow, JAX
Responsibilities
Design, train, validate, and launch models for behavior cloning and reinforcement learning; Build and maintain data ingestion, labeling, and management pipelines; 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 models in real-world environments addressing latency and hardware constraints
Seniority
Mid-Senior, hands-on IC
