Senior Software Engineer, AI/ML Platform
Agility Robotics4 hours ago
Base Salary
$197k - $307k/yr
Responsibilities
- Design and implement an ML platform orchestrating data processing, training, evaluation, and deployment workflows.
- Develop reliable workflows across cloud compute, Kubernetes, and continuous automation.
- Build model registry, feature store, experiment tracking, and related ML infrastructure components.
- Own developer-facing APIs and CLI tools that make ML workflows simple and reproducible.
- Implement continuous integration and delivery for ML, including continuous retraining, automated testing, and model delivery to production.
- Integrate ML orchestration and metadata tracking tools with existing data lakes and pipelines.
- Apply MLOps practices including reproducibility, lineage, rollback, monitoring, and governance.
- Collaborate with AI researchers, robotics engineers, data engineers, and platform teams to deliver scalable systems.
- Mentor junior engineers and influence the broader cloud platform organization’s roadmap.
Requirements
- 5+ years of software engineering experience, including at least 2+ years working on ML infrastructure, data platforms, or production MLOps systems.
- Experience building and maintaining ML platform components such as experiment tracking, model registries, training pipelines, or deployment systems.
- Familiarity with MLflow, WandB, Airflow, Kubeflow, or similar orchestration and tracking tools.
- Proficiency with AWS, GCP, or Azure, containers, and infrastructure-as-code tools such as CDK or Terraform.
- Hands-on experience processing or modeling multimodal data such as sensor logs, camera streams, and behavior traces.
- Ability to collaborate with research scientists, data engineers, and robotics or autonomy teams.
- Experience with robotics, autonomous vehicles, drones, or embedded ML is a bonus.
- Contributions to open-source ML infrastructure or MLOps tooling are a bonus.
- Applicants must have current authorization to work in the United States.
Benefits
- Remote-friendly role with a fully distributed team and flexible work arrangements.
- 401(k) plan with a 6% company match.
- Company stock options, company-paid medical, dental, vision, and short- and long-term disability insurance.
- Benefits eligibility begins on the first day of employment.
- Employee Assistance Program, paid parental leave, professional development, tuition reimbursement, and relocation assistance for eligible roles.
- Flexible unlimited PTO and 12 company holidays for exempt employees, or 10 vacation days, paid sick leave, and 12 company holidays for non-exempt employees.
- Catered lunches four times a week and snacks at Salem and Pittsburgh locations.
- Annual discretionary bonus for eligible roles.
- All roles are U.S.-based.
Tech Stack
Categories
About Agility Robotics
Industrial humanoids hard at work.