Agility Robotics

Staff AI Software Engineer

Agility Robotics

full-time

Posted on:

Location Type: Remote

Location: Remote • 🇺🇸 United States

Visit company website
✨ AI Apply
Apply

Job Level

Lead

Tech Stack

PythonPyTorchTensorflow

About the role

  • Develop data pipelines and curation workflows to maintain high-quality datasets for perception, locomotion, and manipulation tasks.
  • Design, build, and maintain scalable AI infrastructure, including shared libraries, model training pipelines, and evaluation frameworks.
  • Build dashboards, visualization tools, and reusable metrics to monitor and improve model performance and operational readiness.
  • Define and implement testing, validation, and reproducibility tools to ensure reliable deployment of models on Digit.
  • Collaborate with skills and control teams to integrate models efficiently into production systems, including real-time execution environments.
  • Promote best practices for data management, experiment tracking, and ML operations across the AI Engineering organization.

Requirements

  • 8+ years of professional experience in AI infrastructure, ML engineering, or MLOps roles.
  • Strong software engineering skills, particularly in Python; proficiency in developing scalable, maintainable libraries and tools.
  • Experience building and maintaining ML pipelines, from dataset curation to training, evaluation, and deployment.
  • Familiarity with training frameworks such as PyTorch or TensorFlow.
  • Practical understanding of data-driven model development, including dataset management, reproducibility, and performance monitoring.
  • Excellent communication skills and a collaborative mindset for working across AI, controls, and robotics software teams.
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.
  • 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.

Applicant Tracking System Keywords

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

Hard skills
PythonAI infrastructureML engineeringMLOpsML pipelinesdataset curationmodel trainingmodel evaluationmodel deploymentdata management
Soft skills
communicationcollaborationproblem-solvingteamworkorganizational skills