Senior Machine Learning Engineer - Reinforcement Learning
Basis Set · Columbus, OH · 2 days ago
EngineeringInternship
About the role
As a Sr. ML Engineer focused on Reinforcement Learning, you will design, implement, and optimize RL algorithms that enable intelligent agents to operate in dynamic, unstructured environments. This role involves working closely with cross-functional teams to design, test, and deploy innovative solutions that improve the performance and capabilities of our robotic systems.
Responsibilities
- Design, implement, and evaluate RL algorithms for robotic control, motion planning, and adaptive behaviors in dynamic, unstructured environments.
- Develop and integrate RL policies with robot control systems, ensuring compatibility with hardware constraints and real-time requirements.
- Collaborate with perception teams to fuse RL with vision, depth, and sensor data for robust decision-making.
- Build and maintain sim-to-real pipelines, including domain randomization and transfer learning techniques.
- Conduct experiments on physical robots, including designing safety protocols and monitoring for unexpected behaviors.
- Leverage simulation environments (e.g., MuJoCo, Isaac Gym, Gazebo, PyBullet) for large-scale training before real-world validation.
- Continuously improve model efficiency to operate within compute and latency constraints on embedded robotic systems.
Requirements
- Master’s or PhD in Computer Science, Robotics, Machine Learning, or related field, or equivalent practical experience.
- Experience developing and deploying reinforcement learning algorithms on real-world systems.
- Proficiency in Python and deep learning frameworks such as PyTorch or TensorFlow.
- Experience with simulation environments (e.g., MuJoCo, Isaac Gym).
- Solid understanding of probability, statistics, and optimization.
- Experience with training and deploying ML models in production systems.
Qualifications
- Expertise in reinforcement learning and its applications to robotics.
- Strong background in machine learning, computer science, or a related field.
- Experience with robotics, sensor fusion, and control systems.
- Ability to work effectively in a fast-paced, interdisciplinary team environment.
Skills
- Python programming.
- Deep learning frameworks (PyTorch, TensorFlow).
- Simulation environments (MuJoCo, Isaac Gym).
- Probability, statistics, and optimization.
- Robotics and control systems.
Benefits
- Daily free lunch to keep you fueled and connected with the team.
- Flexible PTO so you can take the time you need, when you need it.
- Comprehensive medical, dental, and vision coverage.
- 6 weeks fully paid parental leave, plus an additional 6–8 weeks for birthing parents (12–14 weeks total).
- 401(k) retirement plan through Empower.
Pay
$100,000 - $150,000 annually based on experience.
Schedule
This role can be located in our Columbus, Ohio Headquarters or Remote.