Senior Software Engineer, AI/ML Platform
New
A
Agility RoboticsRobotics, AI
RemoteFull-TimeSenior
Salary$197,000 — $307,000 USD
Apply NowOpens the employer's application page
Job Details
- Experience
- 5+ years of software engineering experience, with at least 2+ years working on ML infrastructure, data platforms or MLOps systems
- Required Skills
- AWSGCPKubeflowKubernetesMLFlowAirflowAzureTerraformMLOps
Requirements
- 5+ years of software engineering experience
- 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)
- Experience with containers and IaC (e.g., CDK, Terraform)
- Hands-on experience with processing or modeling multimodal data (sensor logs, camera streams, behaviour traces etc.)
- Experience collaborating cross-functionally with research scientists and data/robotics engineers
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 to translate requirements into scalable systems
- Partner with data platform engineers to integrate ML orchestration tools with existing data pipelines
- Mentor junior engineers and influence the broader cloud platform organization’s roadmap
View Full Description & ApplyYou'll be redirected to the employer's site