Senior AI Research Engineer

Agility Robotics

. Design, train, and deploy robust policies for locomotion, manipulation, and dynamic interactions with the environment.

Posted 4/21/2026full-timeCalifornia, Oregon, Pennsylvania • 🇺🇸 United StatesSenior💰 $185,000 - $288,000 per yearWebsite

Tech Stack

Tools & technologies

PythonPyTorch

About the role

Key responsibilities & impact

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, design, and test imitation learning methods
Collaborate with Robotics Software and AI engineering teams to develop policies which can be transferred to production

Requirements

What you’ll need

3+ years of experience developing and deploying learning-from-demonstration
Strong programming skills in Python, with proficiency in 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 to perform manipulation tasks.
Ability to work collaboratively in a fast-paced environment to deliver safe, high-quality software
MS in Robotics, Computer Science, or a related field.

Benefits

Comp & 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:
Exempt Employees: Flexible, unlimited PTO and 10 company holidays, including a winter shutdown.
Non-Exempt Employees: 10 vacation days, paid sick leave, and 10 company holidays, including a winter shutdown, annually.
On-Site Perks: Catered lunches four times a week and a variety of healthy snacks and refreshments at our Salem and Pittsburgh 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.
Annual Discretionary Bonus: Provided for eligible roles.

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

Pythonreinforcement learningimitation learningdeep learningscalable training pipelinesevaluation frameworksrobot data collectionmanipulation taskssimulation environmentsDiffusionPolicy

Soft Skills

collaborationfast-paced environmenthigh-quality software delivery

Certifications

MS in RoboticsMS in Computer Science