Senior Staff AI Engineer

Agility Robotics

Agility Robotics

Software Engineering, Data Science
San Francisco, CA, USA · Pittsburgh, PA, USA
Posted on Thursday, April 11, 2024

Agility Robotics is a pioneer. Our robot, Digit, is the first to be sold into workplaces across the globe. Our team is differentiated by its expertise in imagining, engineering, and delivering robots with advanced mobility, dexterity, intelligence, and efficiency -- robots specifically designed to work alongside people, in spaces built for people. Every day, we break through engineering challenges and invent new solutions and capabilities that will one day make robots commonplace and approachable. This work is our passion and our responsibility: our mission is to make businesses more productive and people’s lives more fulfilling.

About the Team

Agility Robotics is the first company in the world to deploy humanoid robots in real customer use cases, doing something useful in the world. Our mission is to build robot partners to augment the human workforce. This means developing generally intelligent, generally capable robots to solve the world’s labor shortage and rid the world of dull, dirty and dangerous tasks. We are building a small, independent, and fast-moving innovation team of creative engineers and technologists who are shielded from the daily fires of product deployment, focused on developing the next generation of hardware and physical intelligence.

About The Role

As a Senior/Staff AI Engineer, you will be a team leader, working with experienced hardware and software engineers to develop cutting-edge autonomous capabilities for our humanoid robots. You will focus on areas such as training foundational visuomotor policies for manipulation and locomotion, planning through learned embeddings of goals, tasks and environments and leveraging text and image foundation models for multi-modal conditioning. You will utilize supervised, unsupervised and reinforcement learning methods on both simulation and hardware data.

About the Work

  • Full-time role, 40 hrs / week
  • Team leadership: roughly 50% individual contributor, 50% managing a highly capable team
  • Stay up to date with the state of the art in reinforcement learning, imitation learning and generative models
  • Apply, implement and improve state of the art machine learning algorithms and models and test them in real-world environments and hardware
  • Own and pursue a research agenda, aligned with the innovation leadership, picking the most promising and impactful problems to work on
  • Train models on collocated and cloud compute clusters with data from a variety of sources

About You

You love pursuing novel ideas and making them real, are a great communicator, and enjoy working with a highly capable team. You are a fast learner and interested in reading and adapting the latest techniques and models in the emerging AI landscape to new applications. You are a creative engineer who builds systems from the ground up.

  • PhD in machine learning, computer science, math or related field, or equivalent experience
  • 5+ years experience developing and applying AI/ML methods in industry
  • Experience developing for and operating physical robots
  • Publications in top machine learning conferences (ICML, NeurIPS, CoRL etc)
  • A track record of coming up with new ideas or improving upon existing ideas in machine learning or robotics
  • Proficiency with Python and extensive experience with hardware accelerated tensor libraries such as PyTorch or JAX


  • Experience training visuomotor policies and deploying them to hardware
  • Experience training and fine tuning large transformer models (of any modality)
  • Experience in neural network deployment toolchains such as ONNX, TensorRT

Work Environment & Physical Demands

The physical demands described here are representative of those that must be met by an employee to successfully perform the essential functions of this position. Reasonable accommodations may be made to enable individuals with disabilities to perform the functions.

While performing the duties of this job, the employee routinely is required to sit; talk and hear; use hands to touch, feel, handle, push/pull and move objects; see details of objects that are less than a few feet away; see differences between colors, shades, and brightness. The employee is occasionally required to stand, walk and move around the office; stoop, kneel, crouch, twist, crawl, reach, and stretch; and see details of objects that are more than a few feet away.

Other Duties

Please note this job description is not designed to cover or contain a comprehensive listing of activities, duties or responsibilities that are required of the employee for this job as this is subject to change at any time with or without notice. Employees will be required to follow any other job-related instructions and to perform any other job-related duties requested by any person authorized to give such instructions or assignments.


  • Competitive Compensation Package
  • 100% Company Paid Disability, Life, Health, Dental & Vision Insurance (for AR employees)
  • Employer Matched Retirement Plan
  • Five Weeks of Paid Time Off Annually
  • Paid Parental Leave
  • Employee Referral Bonus Program
  • Relocation Assistance (for certain roles)
  • Stock Options (for certain roles)
  • A culture that encourages flexible work schedules
  • A culture that embraces individuality and authenticity

Agility Robotics is committed to a work environment in which all individuals are treated with respect and dignity. Each individual has the right to work in a professional atmosphere that promotes equal employment opportunities and prohibits unlawful discriminatory practices, including harassment. Therefore, it is the policy of Agility Robotics to ensure equal employment opportunity without discrimination or harassment on the basis of race, color, religion, sex, sexual orientation, gender identity or expression, age, disability, marital status, citizenship, national origin, genetic information, or any other characteristic protected by law. Agility Robotics prohibits any such discrimination or harassment.

Apply Now: