Note: The job is a remote job and is open to candidates in USA. Agility Robotics develops commercially deployed humanoid robots that work alongside teams in warehouses, manufacturing facilities, and distribution centers. The Senior Software Engineer will architect and build foundational machine learning infrastructure for data processing, training, evaluation, deployment, model management, and observability at fleet scale, while supporting AI and robotics teams.
Responsibilities
- 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
- 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
- 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
Skills
- 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
- All of our roles are U.S.-based. Applicants must have current authorization to work in the United States
- Experience with robotics, autonomous vehicles, drones or embedded ML
- Contributions to open-source ML infrastructure or MLOps tooling a plus
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).
- 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.
Company Overview