Machine Learning Engineer II - Learned Planning (Reinforcement Learning)
Torc Robotics3 hours ago
Remote, United StatesMid Level
H1B sponsor
Base Salary
$153k - $184k/yr
Responsibilities
- Develop and train learned behavior models using behavior cloning, imitation learning, reinforcement learning, and related approaches.
- Implement production-quality machine learning code for model training, evaluation, and inference within the autonomy stack.
- Analyze model performance and failure modes and improve robustness and generalization across driving scenarios.
- Build training pipelines and data workflows using behavior datasets from simulation, fleet logs, and on-vehicle data.
- Collaborate with simulation, validation, and autonomy engineering teams to test and integrate learned behavior models.
- Develop tooling and infrastructure that improve experimentation speed, reproducibility, and model iteration.
- Contribute to technical discussions about model architectures and training strategies.
Requirements
- Bachelor’s degree in Computer Science, Robotics, Electrical Engineering, Machine Learning, or a related technical field with 4+ years of industry experience, or a master’s degree with 2+ years of experience.
- Experience applying imitation learning, reinforcement learning, or sequence modeling to robotics, autonomous systems, or complex control environments.
- Strong programming skills in Python and PyTorch and experience writing production-quality machine learning code.
- Experience training and evaluating machine learning models with large datasets and scalable compute environments.
- Understanding of autonomy-related ML architectures such as transformers, graph neural networks, and sequence models.
- Experience debugging model behavior, analyzing performance metrics, and iterating on training pipelines.
- Ability to collaborate across teams and integrate ML models into larger software systems.
- Preferred experience includes autonomous driving, robotics, simulation-based training, reinforcement learning frameworks, distributed training systems such as Ray, vehicle dynamics, motion planning, multi-agent decision-making, and production ML deployment.
Benefits
- Competitive compensation package with bonus and stock options.
- 100% paid medical, dental, and vision premiums for full-time employees.
- 401(k) plan with a 6% employer match.
- Flexible schedule and generous paid vacation available immediately after the start date.
- AD&D and life insurance.
- Hybrid work is available from the Ann Arbor, Michigan office, or the role may be performed remotely within the United States.
Categories
ML EngineeringRobotics
About Torc Robotics
Torc Robotics develops autonomous driving software for heavy-duty trucks, offering a full self-driving stack, mapping, and perception for freight carriers and OEMs. Founded in 2005 and headquartered in Blacksburg, Virginia, it was acquired by Daimler Truck in 2019. As a subsidiary, Torc focuses on commercializing Level 4 trucking with its parent’s vehicle platforms and industry partnerships in North American freight transport.