Senior Software Engineer, AI/ML Platform

Agility Robotics; Fremont (CA); Remote; USD 197,000 - 307,000; Full time

Benefits offered by this job

401(k) Plan with company match

Equity (stock options)

Full medical/dental/vision coverage

Job summary

Agility Robotics seeks a seasoned software engineer to shape our ML platform. You will architect end-to-end data processing, training, evaluation, and deployment workflows, building core infra like model registries and feature stores.

You will own APIs and CI/CD for ML, enabling reliable retraining and production delivery. You will collaborate with researchers, data engineers, and robotics teams, applying MLOps best practices and guiding platform scalability.

Qualifications

• 5+ years of software engineering experience, with 2+ years on ML infra, data platforms or MLOps in production.
• Experience building ML platform components—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 (CDK, Terraform).
• Hands-on experience with multimodal data (sensor logs, camera streams, behavioural traces).
• Collaborating cross-functionally with researchers, data engineers, and robotics teams to ship infrastructure.

Responsibilities

• Contribute to design and implementation of the ML platform for end-to-end data processing, training, evaluation, and deployment.
• Develop reliable workflows across cloud compute, Kubernetes, and automated pipelines.
• Build core infrastructure components like the model registry, feature store, and experiment tracking tooling.
• Own developer-facing APIs and CLI tools for reproducible ML workflows.
• Implement CI/CD for ML enabling retraining, testing, and model delivery to production.

Skills

ML Platform

Cloud native

CI/CD for ML

Experiment Tracking

Model Registry

Automation

Terraform

Tools

MLflow

WandB

Airflow

Kubeflow

AWS

GCP

Job description

About the Role

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.

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.

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

$197,000—$307,000 USD

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