Software Engineer, Agility Robotics

• 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

Desirable

• Our Autonomy Perception Team is growing rapidly and we are looking for talented engineers to join our small, agile, world-class team
• In this role you will support efforts to build perception features that will enable Digit to autonomously interact with its environment and manipulate objects during real-world deployments with industry partners
• 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