Agility Robotics Logo

Agility Robotics

Senior Software Engineer, AI/ML Platform

Posted 2 Days Ago
Remote
Hiring Remotely in USA
197K-307K Annually
Senior level
Remote
Hiring Remotely in USA
197K-307K Annually
Senior level
Build and scale the ML infrastructure platform for humanoid robotics, including data processing, training, evaluation, deployment, model management, observability, and reproducible workflows. Develop cloud-native Kubernetes workflows, model registries, feature stores, experiment tracking, developer APIs, CLI tools, and ML CI/CD systems. Collaborate with AI researchers, robotics engineers, and data platform teams while mentoring junior engineers and shaping platform architecture, reliability, and scalability.
The summary above was generated by AI

Agility’s commercially deployed humanoids operate alongside teams in warehouses, manufacturing facilities, and distribution centers—tackling physically demanding and repetitive tasks while enabling workers to focus on higher-value work. With industry-leading safety standards and years of proven deployment data, we're pioneering a new era of automation that enhances human potential.

About The Role

Join the team building the machine learning platform to power fleet-scale humanoid robotics. As a senior engineer on the ML Infrastructure and Platform group, you will help architect and build the foundational infrastructure for AI and machine learning operations at Agility. This includes the platform layer for data collection and processing, training, sim and real evaluation, and model management and observability.

Your work will empower our AI teams across perception, controls, skills, and innovation to build and deploy next-generation robot foundation models and end-to-end policies for humanoid robots by providing tools to develop and operationalize machine learning at scale. 

Key Responsibilities

Execution and Technical Ownership

    • Contribute to the design and implementation of the ML platform for orchestrating the end to end AI flywheel of data processing, training, evaluation, and deployment 
    • Develop reliable workflows across cloud compute, Kubernetes, and continuous automation
    • Build core infrastructure components such as the model registry, feature store and experiment tracking tooling.
    • Own developer-facing APIs and CLI tools that make ML workflows simple and reproducible. 
    • Implement the CI/CD lifecycle for ML that enable continuous retraining, automated testing, and seamless model delivery to production environments

Collaboration

    • Work closely with the Staff ML Infra Engineer and cross-functional stakeholders (AI researchers and robotics engineers) to understand requirements and translate them into scalable solutions/systems.
    • Partner with data platform engineers to integrate ML orchestration and metadata tracking tools with our existing data lake and pipelines.

Engineering Excellence, Growth and Impact

    • Apply MLOps best practices: reproducibility, lineage, rollback, monitoring and governance. 
    • Mentor junior engineers and influence the broader cloud platform organization’s roadmap.
    • Contribute to internal discussions on platform architecture, reliability, and scalability alongside the broader ML and data platform team
What We’re Aiming For (MLOps Level 2)
  • Version-controlled ML pipelines (data, code, and config)
  • Automated and reproducible model training and evaluation
  • Continuous integration and delivery for ML workflows
  • Centralized experiment tracking and performance visualization
  • Standardized model packaging and deployment to production
  • Monitoring of models post-deployment
Required Qualifications
  • 5+ years of software engineering experience, with at least 2+ years working on ML infrastructure, data platforms or MLOps systems in production environments.
  • Experience building and maintaining components of modern ML platforms—such as experiment tracking, model registries, training pipelines, or deployment systems
  • Familiarity with orchestration and tracking tools (MLflow, WandB, Airflow, Kubeflow, etc.)
  • Proficiency with cloud-native platforms (AWS, GCP, or Azure), containers, and IaC (e.g., CDK, Terraform)
  • Hands-on experience with processing or modeling multimodal data(sensor logs, camera streams, behaviour traces etc).
  • Comfortable collaborating cross-functionally with research scientists, data engineers, and robotics/autonomy teams to ship infrastructure used by others
Bonus Qualifications
  • Experience with robotics, autonomous vehicles, drones or embedded ML.
  • Contributions to open-source ML infrastructure or MLOps tooling a plus. 
Why This Role?
  • Build from the start Join at a pivotal moment - help shape the ML platform layer as its being defined, not inherited. 
  • High impact: Your work will directly enable faster, safer, and more intelligent robotic behaviors at scale
  • Technical Frontier: You will work directly on enabling the next frontier of AI in real production settings
  • Remote-friendly with a strong engineering culture and a fully distributed team.

The final salary offered to a successful candidate will be dependent on several factors that may include but are not limited to: market location, job-related knowledge, skills, and experience. This range may change based on geographical location and may be modified in the future.

Anticipated Base Salary Range
$197,000$307,000 USD

In addition to base pay, our competitive total rewards package consists of the following for full-time employees:

  • 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 12 company holidays, including a winter shutdown.
    • Non-Exempt Employees: 10 vacation days, paid sick leave, and 12 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.
  • Annual Discretionary Bonus: 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.


Similar Jobs

18 Days Ago
Remote
United States
220K-260K Annually
Senior level
220K-260K Annually
Senior level
Artificial Intelligence • Marketing Tech • Mobile • Software
As a Senior Software Engineer, you'll build and operate the ML platform, optimizing data operations, managing the feature store, and supporting ML initiatives while driving architectural roadmap decisions.
Top Skills: AirflowSparkAWSCloudflareDynamoDBGradleGraphQLHuggingfaceJavaKafkaKinesisKubernetesMetaflowPandasPlanetscalePlaywrightPostgresPythonPyTorchRayReactRedisSpark Structured StreamingSpring BootTensorFlowTerraformTypescriptVite
5 Minutes Ago
Easy Apply
Remote
United States
Easy Apply
146K-225K Annually
Junior
146K-225K Annually
Junior
Big Data • Fintech • Mobile • Payments • Financial Services
Develop and ship full-stack web and mobile experiences across Affirm’s consumer app and website. Collaborate with product, design, analytics, and engineering teams to break down projects, make technical decisions, write tested and extensible code, launch scalable APIs, monitor service health, support on-call operations, review code, and participate in team development and interviewing.
Top Skills: APIsKotlinPythonReactReact NativeVue
8 Minutes Ago
Remote
United States
85K-115K Annually
Senior level
85K-115K Annually
Senior level
Software
Manage the full collections and accounts receivable lifecycle, including recovery strategy, balance holds, ACH returns, write-offs, journal entries, forecasting, and risk controls. Lead collections analysts and specialists, monitor aging and recovery KPIs, handle escalations, and partner with Finance, Risk, Sales, Operations, Disputes, Underwriting, and Engineering. Use AI to automate correspondence, triage cases, identify delinquency patterns, reduce manual work, and improve cash flow and operational efficiency.
Top Skills: AdyenAi ToolsExcelGoogle SheetsSalesforceTsysWorkday

What you need to know about the Chicago Tech Scene

With vibrant neighborhoods, great food and more affordable housing than either coast, Chicago might be the most liveable major tech hub. It is the birthplace of modern commodities and futures trading, a national hub for logistics and commerce, and home to the American Medical Association and the American Bar Association. This diverse blend of industry influences has helped Chicago emerge as a major player in verticals like fintech, biotechnology, legal tech, e-commerce and logistics technology. It’s also a major hiring center for tech companies on both coasts.

Key Facts About Chicago Tech

  • Number of Tech Workers: 245,800; 5.2% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: McDonald’s, John Deere, Boeing, Morningstar
  • Key Industries: Artificial intelligence, biotechnology, fintech, software, logistics technology
  • Funding Landscape: $2.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Pritzker Group Venture Capital, Arch Venture Partners, MATH Venture Partners, Jump Capital, Hyde Park Venture Partners
  • Research Centers and Universities: Northwestern University, University of Chicago, University of Illinois Urbana-Champaign, Illinois Institute of Technology, Argonne National Laboratory, Fermi National Accelerator Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account