Machine Learning Engineer - Robot Manipulation

Company Overview

Maven Robotics is building the world’s leading general-purpose AI robots.

We are currently operating in stealth and are growing the world’s best team in AI robotics. We are looking for self-starters that are the world’s best in their field, who can innovate from a deep understanding of the fundamentals, and who share our values of unwavering truth seeking and integrity, humility, curiosity, and relentless determination.

Role Description

We are looking to recruit an exceptional Machine Learning Engineer - Robot Manipulation to design, implement, test, and deploy robot manipulation algorithms that enable assembly and material movement tasks.

In this role you will:

• Design and implement machine learning algorithms, with a focus on reinforcement learning (RL) and imitation learning (IL), to enable robotic manipulators to perform complex tasks in dynamic environments.
• Translate high-level objectives into machine learning problems and deploy robust, scalable models to real-world robotic systems.
• Integrate your ML solutions into existing robotics workflows, ensuring that models are performant in both simulated and real-world settings.
• Drive innovation by incorporating the latest research in machine learning into practical applications that push the boundaries of robotic manipulation.
• Take ownership of critical ML projects, seeing them through from conception to successful deployment.
• Collaborate across disciplines to ensure seamless integration of ML models and provide technical mentorship to junior engineers.

Qualifications

Must-have:

• MS or PhD in machine learning, computer science, robotics, or a related field.
• Strong practical experience in training and deploying machine learning models for real-world applications.
• Deep understanding of reinforcement learning (RL) and imitation learning (IL) and their application to robotics.
• Proficiency in programming languages and tools commonly used in machine learning (e.g., Python, PyTorch).
• Experience with data collection, preprocessing, and management in the context of training ML models.
• Self-starter attitude with strong ability to identify problems, prioritize them, then plan and execute working solutions.
• Enthusiasm for working in a fast paced startup environment and eagerness to support the team on a variety of topics.

Nice-to-have:

• Familiarity with robotic simulation environments (e.g., Gazebo, MuJoCo) and experience in sim-to-real transfer.
• Experience in:
• Designing and implementing reward functions for complex manipulation tasks.
• Developing models that can handle noisy, incomplete, or sparse data.
• Deployment of ML models to edge devices for real-time inference.
• Accelerating ML training processes using GPU, TPU, or other HW accelerators.
• Using reinforcement learning frameworks, e.g. Stable Baselines, RLlib, or similar.
• General knowledge of robotics principles, including kinematics, dynamics, and control.
• Publications or contributions to the machine learning community, particularly in areas related to robotics or reinforcement learning.