Software Engineer III, Machine Learning

Summary

Join Agility Robotics' Autonomy Perception Team as a Machine Learning Engineer to develop perception features for Digit, the company's humanoid robot. You will support the training, optimization, and deployment of machine learning models, develop systems to monitor perception performance, and create data collection and annotation protocols. Collaborate with other engineering teams to solve complex problems. The ideal candidate possesses a BS or MS in computer science or a related field, 3+ years of relevant experience, and expertise in machine learning libraries like PyTorch and Slurm. The role requires proficiency in Python, Git, and effective communication skills. Agility Robotics offers a comprehensive benefits package including 401k matching, equity, insurance, paid time off, and more.

Requirements

• BS or MS in computer science or related field or have equivalent industry experience
• 3+ years of relevant experience as a machine learning engineer
• Experience training ML models (cluster-based training is a plus)
• Familiarity with common machine learning libraries such as (but not limited to) PyTorch and Slurm
• Software experience in Python
• Experience with version control (Git)
• Ability to communicate technical information effectively both orally and in writing with other team members
• Ability to work independently and as part of a team with a demonstrated ability to establish and maintain effective working relationships
• Authorization to work in the USA

Responsibilities

• Support training, optimization, and deployment of machine learning models on a large fleet of state-of-the-art humanoid robots
• Train and deploy custom machine learning models to improve robot perception and task performance for specific customer environments and applications
• Develop systems to monitor perception performance for robots in the field
• Develop robot data collection, annotation, and analysis protocols to improve machine learning models and perception performance for robots in the field
• Support cluster hardware, software, and protocols for training and evaluating machine learning models
• Support machine learning performance tests on secondary cluster that uses robot compute and GPU hardware
• Work with other teams in software (Manipulation, Navigation, Planning, and Controls) to solve complex cross functional problems

Preferred Qualifications

• Experience setting up compute cluster for simultaneously training and validation of multiple ML models
• Experience deploying neural networks with limited compute (e.g., TensorRT, ONNX, Pytorch Mobile, Tensorflow Lite, etc.)
• Experience with robotics hardware and embedded systems
• Experience with data annotation services
• Experience with GPU Programming (eg. CUDA, OpenGL)
• Experience with modern C++
• Exposure to standard agile and software development processes (issue tracking tools (Jira), feature branching, version control (Git), continuous integration and deployment

Benefits

• 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
• Time Off: Non-Exempt Employees: 10 vacation days, paid sick leave, and 10 company holidays, including a winter shutdown, annually
• On-Site Perks: Catered lunches twice 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