Agility Robotics Logo

Agility Robotics

Staff AI Software Engineer

Reposted Yesterday
Remote
Hiring Remotely in USA
Senior level
Remote
Hiring Remotely in USA
Senior level
Lead the development of tools and infrastructure for AI models to ensure Digit operates autonomously and efficiently. Create data pipelines, validation tools, and robust monitoring systems.
The summary above was generated by AI

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 Role

The AI Engineering team builds the models and infrastructure that enable Digit to operate autonomously, efficiently, and safely. As a Staff AI Core Engineer, you will lead the development of the foundational tools, libraries, and data systems that power our AI teams to deliver robust and scalable learning-based capabilities for Digit.

This is a remote-friendly role with a preference for candidates located near one of our offices for in-person collaboration, meetings, and events.

About The Work

  • 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.

About You

  • 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.

Bonus Points

  • Experience deploying machine learning models into real-time or embedded systems.
  • Experience managing cloud or on-premise GPU clusters, distributed training, or high-performance compute environments.
  • Familiarity with experiment tracking tools (e.g. MLflow, Weights & Biases) and orchestration systems (e.g.Kubernetes).
  • Background in robotics, perception, or control system applications.

Full-time Employees are eligible for 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.

All of our roles are U.S.-based. Applicants must have current authorization to work in the United States.

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.


Agility Robotics does not accept unsolicited referrals from third-party recruiting agencies.  We prioritize direct applicants and encourage all qualified candidates to apply directly through our careers page.  If you are represented by a third party, your application may not be considered.  To ensure full consideration, please apply directly.


Apply Now: https://grnh.se/b444bbd04us

Top Skills

Kubernetes
Mlflow
Python
PyTorch
TensorFlow
Weights & Biases

Similar Jobs

Yesterday
Remote or Hybrid
Santa Clara, CA, USA
173K-303K Annually
Senior level
173K-303K Annually
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Design and develop AI/ML solutions, focusing on scalable software and user experience, while collaborating across teams to integrate generative AI technologies.
Top Skills: AngularJavaKubernetesLlmsOpentelemetryReactVue
2 Days Ago
Easy Apply
Remote
USA
Easy Apply
218K-257K Annually
Senior level
218K-257K Annually
Senior level
Artificial Intelligence • Blockchain • Fintech • Financial Services • Cryptocurrency • NFT • Web3
As a Staff Software Engineer, you'll lead AI adoption, design prototypes, collaborate across teams, and ensure responsible AI integration at Coinbase.
Top Skills: AIAWSGenerative AiGoLarge Language ModelsMicroservices ArchitecturePython
8 Days Ago
Remote or Hybrid
Santa Clara, CA, USA
188K-328K Annually
Senior level
188K-328K Annually
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
The role involves designing AI agents, building intelligent search capabilities, collaborating on scalable solutions, and delivering high-quality products while mentoring colleagues.
Top Skills: AngularCi/CdCloudJavaJavaScriptReactRelational DatabasesUnix-Based SystemsVue

What you need to know about the NYC Tech Scene

As the undisputed financial capital of the world, New York City is an epicenter of startup funding activity. The city has a thriving fintech scene and is a major player in verticals ranging from AI to biotech, cybersecurity and digital media. It also has universities like NYU, Columbia and Cornell Tech attracting students and researchers from across the globe, providing the ecosystem with a constant influx of world-class talent. And its East Coast location and three international airports make it a perfect spot for European companies establishing a foothold in the United States.

Key Facts About NYC Tech

  • Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
  • Key Industries: Artificial intelligence, Fintech
  • Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
  • Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account