AI Research Engineer

Position Overview

Agility Robotics is a pioneer delivering cutting-edge robotic technology through its robot, Digit – the first robot sold into workplaces globally. Our team specializes in imagining, engineering, and delivering robots with advanced mobility, dexterity, intelligence, and efficiency to work alongside people in human-centric spaces. Our mission is to make businesses more productive and people’s lives more fulfilling.

In this role, you will join the AI innovation team to develop and test cutting-edge methods in imitation learning and reinforcement learning for humanoid robots and help transition these solutions into production.

Key Responsibilities

• Design, train, and deploy robust policies for locomotion, manipulation, and dynamic interactions with the environment.
• Develop core reinforcement learning infrastructure, including scalable training pipelines and evaluation frameworks.
• Design and implement new simulation environments and tasks to support training and deployment of control policies.
• Develop and test imitation learning methods.
• Collaborate with Robotics Software and AI engineering teams to ensure transferability of policies into production.

Required Qualifications

• 3+ years of experience in developing and deploying learning-from-demonstration techniques.
• Strong programming skills in Python and proficiency with deep learning frameworks, such as PyTorch.
• Experience with modern learning-from-demonstration tools like DiffusionPolicy.
• Experience with robot data collection, training, and testing on hardware for manipulation tasks.
• Ability to work collaboratively in a fast-paced environment to deliver safe, high-quality software.

Preferred Qualifications

• Advanced degree (MS or PhD) in Robotics, Computer Science, or a related field.
• Publications in top ML or robotics conferences (e.g. NeurIPS, ICML, CoRL, RSS, ICRA).
• Familiarity with robot simulation environments (e.g. Mujoco, Isaac Sim) and sim-to-real transfer techniques.
• Experience with modern reinforcement learning techniques for locomotion, manipulation, and whole-body control.
• Experience with writing performant, high quality software in C++.

Benefits & Perks

• 401(k) Plan: Includes a 6% company match.
• Equity: Company stock options.
• Insurance Coverage: 100% company-paid medical, dental, vision, and short/long-term disability insurance for employees.
• Benefit Start Date: Eligible for benefits on your first day of employment.
• Well-Being Support: Employee Assistance Program (EAP).
• Time Off: For exempt employees, flexible, unlimited PTO and 10 company holidays (including a winter shutdown); for non-exempt employees, 10 vacation days, paid sick leave, and 10 company holidays annually.
• On-Site Perks: Catered lunches four times a week and a variety of healthy snacks and refreshments at select locations.
• Parental Leave: Generous paid parental leave programs.
• Work Environment: A culture that supports flexible work arrangements.
• Growth Opportunities: Professional development and tuition reimbursement programs.
• Relocation Assistance: Provided for eligible roles.