Jobs · Engineering

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.

Similar jobs