Senior Software Engineer, AI/ML Platform
$197k - $307kFull-time
Agility Robotics
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.
About The Role
Join the team building the machine learning platform to power fleet-scale humanoid robotics. As a senior engineer on the ML Infrastructure and Platform group, you will help architect and build the foundational infrastructure for AI and machine learning operations at Agility. This includes the platform layer for data collection and processing, training, sim and real evaluation, and model management and observability. Your work will empower our AI teams across perception, controls, skills, and innovation to build and deploy next-generation robot foundation models and end-to-end policies for humanoid robots by providing tools to develop and operationalize machine learning at scale. 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
- 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
- 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
- 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
- Experience with robotics, autonomous vehicles, drones or embedded ML.
- Contributions to open-source ML infrastructure or MLOps tooling a plus.
- 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.
- 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.
Vacancy posted a month ago
Similar jobs that could be interesting for youBased on the Senior Software Engineer, AI/ML Platform in California vacancy
- ...-performance computing, cloud, and AI. Whether you’re designing next-gen... ...and technically exceptional PMTS AI/ML Compiler Engineer to join AMD's AI Software organization. In this role, you will... ...and inference workloads on AMD GPU platforms. You will work at the intersection...PlatformSenior
$174k - $252k
...one or more specialized ML areas, leverage ML... ...years of experience with software development in one or more... ...JAX) and production ML platforms (e.g., TFX, AdBrain).... ...systems.Google's software engineers develop the next-... ...is at the forefront of AI innovation, applying cutting...PlatformSenior$174k - $252k
...mechanisms).Develop and integrate AI-agentic platforms and toolchains to automate... ...analytical and optimization engines (e.g., Mixed-Integer... ...practical experience.5 years of software engineering (e.g., C++ and Python... ...parallelism) across modern ML frameworks (e.g., JAX,...PlatformSenior$174k - $252k
...GenAI solutions, leverage ML infrastructure, and... ...maintaining, or launching software products, and 1 year of... ....Google's software engineers develop the next-generation... ...technology forward.The AI and Infrastructure team... ...the essential platforms that enable developers...PlatformSeniorWorldwide$143k - $191k
...the next generation of Agentic AI products that help millions... ...decisions. We’re looking for a Software Engineer to design and build highly... ...managers, designers, and AI/ML engineers to bring intelligent... ...EvaluationFamiliarity with cloud platforms (AWS, GCP, or Azure)...PlatformSeniorWork at officeRemote workFlexible hours$164k
About the roleAs a Senior Software Engineer on the AI Enablement team, you will build the internal platforms, primitives, and guardrails that make AI a trusted co-pilot for employees... ...-on experience building with LLMs or ML services and shipping production-quality features...PlatformSeniorFull timeWork at officeLocal areaRemote workFlexible hours$193.3k - $261.5k
...systems that enable Amazon’s AI to learn from real-world... ...improve at massive scale.As a Senior Software Development Engineer, you will lead the design... ...problems and build scalable platforms that transform billions of... ...Experience with CUDA kernels or ML/low-level kernelsPreferred...PlatformSeniorInternshipLocal areaFlexible hours$119.77k - $140.9k
...maintenance of best in class software experiences. The candidate... ...to provide guidance on engineering best practices and compliance... ...Strong backend + ML systems experienceGenAI exposure (required)ML platform engineering experienceStrong AI/ML understanding (incl. embeddings...PlatformSeniorFull timeWork experience placementLocal area3 days per week$160k - $225k
...running the world's best data and AI infrastructure platform so our customers can use deep data... ...sophisticated AI teams in the world.As a Senior Software Engineer for AI Runtime, you will play a... ...infrastructure, high-performance computing, or ML systems.Experience with distributed...PlatformSeniorLocal areaWorldwide$155k - $175k
Senior Software Engineer, Applied AIRole OverviewFlock Freight is looking for a Senior... ...Software Engineer, Applied AI to help the company adopt AI... ...You will work closely with Platform Ops and business teams to... ...year delivering applied AI, ML, or intelligent automation in...PlatformSeniorFull timeWork at officeLocal areaWork from homeRelocation$174k - $252k
...testability, and adherence to engineering best practices.... ...Triage, debug, and resolve platform issues, analyzing root... ...or more specialized ML areas, leverage ML infrastructure... ...of experience with software development in one or... ...is building the core AI-driven platform that...PlatformSeniorWorldwide$174k - $252k
...and develop foundational AI feedback loop with foundational... ...years of experience with software development in one or... ...etc), applied ML or Core ML development, or ML platform/infrastructure (e.g., model... ...areas.Google's software engineers develop the next-generation...PlatformSenior$174k - $252k
...one or more specialized ML areas, leverage ML... ...maintaining, or launching software products, and 1 year of... ...technologies.Google's software engineers develop the next-... ...technology forward.The AI and Infrastructure team... ...the essential platforms that enable developers...PlatformSeniorWorldwide$200k - $238k
...money.We are looking for engineers who enjoy building and shipping... ...company operates. As a Senior Software Engineer on the Data & AI team, you'll expand... ...our internal data and AI platform, into the layer that lets... ...systems; data engineering, ML engineering, and backend/...PlatformSeniorWorldwideRelocation$164k
About the role As a Senior Software Engineer on the AI Enablement team, you will build the internal platforms, primitives, and guardrails that make AI a trusted co-pilot for employees... ...Hands-on experience building with LLMs or ML services and shipping production-quality...PlatformSeniorFull timeFlexible hours- C3 AI (NYSE: AI), is the Enterprise AI application software company. C3 AI delivers a family of fully... ...the C3 Agentic AI Platform, an end-to-end platform... ...3 AI is looking for Senior Software Engineers to join the rapidly growing... ...and extensible AI/ML platform for large-...PlatformSeniorWork experience placement
- Instawork Robotics is seeking a Senior Software Engineer to contribute to next-generation robotics products. You will work across distributed systems... ..., Kubernetes, and cloud infrastructure to deliver scalable AI-powered solutions that process massive data workloads. You’...PlatformSenior
$240k
Title : Senior AI Engineer Location : San Francisco, CA (Hybrid) Compensation... ...phase of innovation. The platform is redefining how critical enterprise... ...Requirements 6+ years of software engineering experience, including 3+ years working on ML or AI systems Proven...PlatformSenior$174.92k - $209.91k
...ready to query, with no engineering or maintenance... ...the context teams and AI systems rely on. Fivetran... ...Fivetran is looking for a Senior R&D Software Engineer to join our... ...SRE work to keep the platform trustworthy at scaleWrite... ...reading AI/ML research and turning promising...PlatformSeniorFull timeWork at officeRemote workFlexible hoursShift work$153.2k - $234.1k
...of the Autonomous Vehicle (AV) software stack through intelligent automation, AI-enabled engineering workflows, and data-driven validation... ...Engineers, DevOps, QA, and AI/ML teams to design and implement... ..., dashboards, observability platforms, and cloud services into unified...PlatformSeniorFull timeLocal areaWork from homeFlexible hours$173k - $251k
Everlaw is looking for a Senior Software Engineer, AI Platform with experience building and operating AI platform capabilities that power user-facing software... ...user interfacesYou have practical experience with AI/ML-powered systems such as retrieval pipelines, semantic...PlatformSeniorFull timeWork experience placementWork at officeLocal areaRemote work2 days per week3 days per week$167.7k - $245.2k
...close on: 10/26/2026Meet the TeamThe AI Software & Platform team delivers AI products and platform... ...resilience of our AI platform. Partner with Engineering, Product, UX, DevOps, Documentation,... ...testing Generative AI, AI/ML platforms, AI agents, LLM-powered applications...PlatformSeniorFull timeTemporary workLocal areaFlexible hours- Design, build, and operate agentic AI systems and secure, scalable... ...product teams, and strengthen engineering quality through reviews,... ...Requirements: Requires 5+ years of software engineering experience... ...delivering LLM applications or AI/ML systems in production. Candidates...PlatformSenior
$119.77k - $140.9k
...maintenance of best in class software experiences. The... ...reviews to provide guidance on engineering best practices and... ...applications Strong backend + ML systems experience GenAI exposure (required) ML platform engineering experience Strong AI/ML understanding (incl. embeddings...PlatformSeniorTemporary workWork experience placementLocal area3 days per week- ...centers and a data platform processing... ...of events daily. Engineering is not a support function... ...We are embracing AI not as a trend, but... ...it to ship better software, faster, while never... ...researcher or an ML specialist. You are... ...decisions to senior leadership with confidence...PlatformSeniorFull timeWork at officeLocal area
- ...computing experiences—from AI and data centers, to... ...critical to AMD’s AI software roadmap.AMD GPUs are an... ...Triton/Gluon kernels for ML kernels powering the... ...GPUs the best-in-class platform for Triton-powered AI workloads... ..., and performance engineering. You are comfortable...PlatformSenior
- C3 AI (NYSE: AI), is the Enterprise AI application software company. C3 AI delivers a family of fully integrated products including the C3 Agentic AI Platform, an end-to-end platform for developing, deploying... ...looking for a seasoned software engineer experienced in the field...PlatformSeniorWork experience placement
$182k - $242k
...CoreWeave is The Essential Cloud for AI™. Built for pioneers by pioneers, CoreWeave delivers a platform of technology, tools, and... ...role We're looking for a Senior Engineer to be a driving force on CoreWeave... ...ideally for infrastructure or ML systems rather than general-...PlatformSeniorPermanent employmentFull timeTemporary workCasual workWork at officeFlexible hours$174k - $252k
...integration of the Expert Intelligence platform with flagship Google GenAI products (e.g... ...technical leadership and mentorship to other engineers within a rapidly growing organization,... ...experience.5 years of experience with software development in C++ and JavaExperience in...PlatformSenior$170k - $205k
...only vertically integrated AI infrastructure company built... ...an ambitious and experienced Senior Software Engineer to join their team. You'll have... ...teams to define the AI platform roadmap. Influence the long... ...stage projects quickly. ~ AI/ML Expertise: Experience with...PlatformSeniorFull timeTemporary workWork at office
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior Software Engineer, AI/ML Platform. Be the first to apply!
Related searches
- software engineer amazon California
- information technology software engineer California
- federal - software developer California
- rust software engineer California
- software engineer remote California
- junior software developer internship California
- remote entry level software developer California
- software engineer contract California
- network software engineer California
- intel software engineer California



