Senior Software Engineer, AI/ML Platform
Agility Robotics3 days ago
Pittsburgh, PA, USA +2 moreSenior
Base Salary
$197k - $307k/yr
Responsibilities
- Design and implement the ML platform for end-to-end data processing, training, evaluation, and deployment workflows.
- Develop reliable workflows across cloud compute, Kubernetes, and continuous automation.
- Build model registry, feature store, and experiment tracking infrastructure.
- Own developer-facing APIs and CLI tools for simple and reproducible ML workflows.
- Implement continuous integration and delivery for ML, including retraining, automated testing, and production model delivery.
- Integrate ML orchestration and metadata tracking with existing data lakes and pipelines.
- Apply MLOps practices including reproducibility, lineage, rollback, monitoring, and governance.
- Mentor junior engineers and influence the cloud platform roadmap and architecture.
- Collaborate with AI researchers, robotics engineers, data engineers, and autonomy teams.
Requirements
- At least 5 years of software engineering experience, including at least 2 years working on ML infrastructure, data platforms, or MLOps systems in production.
- Experience building modern ML platform components such as experiment tracking, model registries, training pipelines, or deployment systems.
- Familiarity with ML orchestration and tracking tools such as MLflow, WandB, Airflow, and Kubeflow.
- Proficiency with cloud-native platforms including AWS, GCP, or Azure, plus containers and infrastructure as code 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 to deliver shared infrastructure.
- Experience with robotics, autonomous vehicles, drones, or embedded ML is preferred.
- Open-source contributions to ML infrastructure or MLOps tooling are a plus.
Benefits
- Remote-friendly role with a fully distributed team and flexible work arrangements.
- 401(k) plan with a 6% company match, company stock options, and an annual discretionary bonus for eligible roles.
- Company-paid medical, dental, vision, and short- and long-term disability insurance, with benefits beginning on the first day.
- Employee Assistance Program, paid parental leave, and paid time off or vacation and sick leave depending on exemption status.
- Twelve company holidays, including a winter shutdown.
- Catered lunches four times a week and snacks at Salem and Pittsburgh locations.
- Professional development, tuition reimbursement, and relocation assistance for eligible roles.
Tech Stack
Categories
Data EngineeringML Engineering
About Agility Robotics
Agility Robotics builds and deploys Digit, a bipedal humanoid robot for logistics and manufacturing tasks such as tote handling and material flow in warehouses and distribution centers. The privately held company sells robots, software, and support services to industrial operators integrating automation alongside human workers. Founded in 2015 and headquartered in Salem, Oregon, it has commercially deployed systems in real-world facilities.