Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Senior Software Engineer, AI/ML Platform

$197k - $307k
Full-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

Collaboration

    • 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.
Engineering Excellence, Growth and Impact :

    • 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
What We're Aiming For (MLOps Level 2)

  • 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

Required Qualifications

  • 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

Bonus Qualifications

  • Experience with robotics, autonomous vehicles, drones or embedded ML.
  • Contributions to open-source ML infrastructure or MLOps tooling a plus.

Why This Role?

  • 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.

The final salary offered to a successful candidate will be dependent on several factors that may include but are not limited to: market location, job-related knowledge, skills, and experience. This range may change based on geographical location and may be modified in the future.

Anticipated Base Salary Range

$197,000—$307,000 USD

In addition to base pay, our competitive total rewards package consists of the following for full-time employees:

  • 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.
All of our roles are U.S.-based. Applicants must have current authorization to work in the United States.

Agility Robotics is committed to a work environment in which all individuals are treated with respect and dignity. Each individual has the right to work in a professional atmosphere that promotes equal employment opportunities and prohibits unlawful discriminatory practices, including harassment. Therefore, it is the policy of Agility Robotics to ensure equal employment opportunity without discrimination or harassment on the basis of race, color, religion, sex, sexual orientation, gender identity or expression, age, disability, marital status, citizenship, national origin, genetic information, or any other characteristic protected by law. Agility Robotics prohibits any such discrimination or harassment.

Agility Robotics does not accept unsolicited referrals from third-party recruiting agencies. We prioritize direct applicants and encourage all qualified candidates to apply directly through our careers page. If you are represented by a third party, your application may not be considered. To ensure full consideration, please apply directly.
Vacancy posted 3 days ago
Similar jobs that could be interesting for youBased on the Senior Software Engineer, AI/ML Platform in California vacancy
  • $175k - $250k

     ...Swayable Swayable is a fast-growing AI and automated data science platform that measures public opinion...  ...Swayable is seeking a Senior Engineer blending Python software development expertise with scientific...  ...constantly evolving toolset for ML and AI Ops. You are... 
    Platform
    Senior
    Full time

    Swayable

    San Francisco, CA
    1 day ago
  •  ...usher in this new era, we seek AI-native thinkers across every...  ...Build and scale the orchestration engines that execute complex agentic...  ...backend infrastructure for AI/ML products. Technical Stack:...  ...of AI: vector indices, agent platforms, and building scalable data pipelines... 
    Platform
    Senior
    Full time

    Snowflake

    Menlo Park, CA
    1 day ago
  • $190k - $280k

     ...About Sentry Bad software is everywhere, and we’re tired of it. Sentry is on a mission...  ...tools. About the role As a Senior Software Engineer on Sentry’s AI/ML team, you’ll be directly responsible for developing the platform used by our debugging agents. This role... 
    Platform
    Senior
    Hourly pay
    Full time
    Work at office

    Sentry

    San Francisco, CA
    1 day ago
  • $181k - $220k

     ...Aircall is a unicorn, AI-powered customer communications platform used by 22,000+ companies worldwide to drive...  ...home here. We are looking for a Senior Software Engineer to help build agentic systems...  ...least 2 years working directly on AI/ML-powered applications. ~ Strong... 
    Platform
    Senior
    Full time
    Worldwide

    Aircall

    San Francisco, CA
    1 day ago
  • $143k - $191k

     ...next generation of Agentic AI products that help millions...  ...decisions. We’re looking for a Software Engineer to design and build highly scalable...  ...managers, designers, and AI/ML engineers to bring...  ...Evaluation Familiarity with cloud platforms (AWS, GCP, or Azure)... 
    Platform
    Senior
    Full time
    Work at office
    Remote work
    Flexible hours

    Eightfold

    Santa Clara, CA
    1 day ago
  • $144k - $270k

     ...Impact at LILA We are seeking a Senior Software Engineer to join our Applied AI group and help build the next...  ...generation of our AI-driven scientific platform. In this role, you will design and...  ..., and vectorized data for AI/ML workloads. Database & Knowledge... 
    Platform
    Senior
    Full time
    Work at office
    Local area
    Flexible hours

    Lila Sciences

    San Francisco, CA
    1 day ago
  •  ...in this new era, we seek AI-native thinkers across...  ..., collaborative team of engineers on a journey to provide...  ...collaboration at Snowflake. AS A SENIOR SOFTWARE ENGINEER IN DATA CLEAN...  ...APIs to expand platform capabilities and support custom AI/ML workflows. Partner closely... 
    Platform
    Senior
    Full time
    Shift work

    Snowflake

    Menlo Park, CA
    1 day ago
  • $170k - $210k

     ...protocols, and a modern technology platform to deliver care that has...  ...and scaled remotely. AI-native engineering. We expect every engineer to...  ...the frontier of how we build software. Low-ego, high-curiosity....  ...building production LLM / ML features ~ Strong hands-on... 
    Platform
    Senior
    Full time
    Work at office
    Immediate start
    Remote work
    Shift work
    2 days per week

    Midi Health

    San Francisco, CA
    1 day ago
  • $152k - $241.5k

    We are seeking a Senior Software Engineer to drive integration of the NVIDIA Grove...  ...set of leading open-source AI frameworks. In this role, you...  ...ecosystem, or related ML infrastructure projects.Experience...  ..., or multi-tenant AI platforms.Experience building SDKs/APIs... 
    Platform
    Senior
    Full time

    Nvidia

    Santa Clara, CA
    1 day ago
  •  ...the enterprise sustainability platform. Companies like Airbnb,...  ...Watershed is building the AI suite for companies to measure...  ...business. We're looking for software engineers to help build the AI platform...  ...in backend, platform, or AI/ML engineering ~ Experience building... 
    Platform
    Senior
    Full time
    Work at office
    Remote work

    Watershed

    San Francisco, CA
    1 day ago
  • $174k - $252k

     ...learning, and generative AI, from concept to...  ...roadmap for key ML components, ensuring...  ...exploration.Mentor engineers and promote best...  ...of experience with software development using one...  ...and production ML platforms (e.g., TFX, AdBrain...  ...technology forward.As a Senior Software Engineer... 
    Platform
    Senior

    Google

    Mountain View, CA
    2 days ago
  •  ...businesses build and maintain trust, with AI at the core of their next phase of innovation. The platform is redefining how critical...  ...Requirements ~6+ years of software engineering experience, including 3+ years working on ML or AI systems ~ Proven experience... 
    Platform
    Senior

    Harnham

    San Francisco, CA
    17 hours ago
  • $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... 
    Platform
    Senior
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • $193.3k - $261.5k

     ...about large-scale systems, AI innovation, and...  ...Architect and build scalable ML infrastructure powering...  ...excellence* Raise the engineering bar through technical...  ...internship professional software development experience-...  ...SGLang, TensorRT or similar platforms in production... 
    Platform
    Senior
    Internship
    Local area
    Flexible hours

    Amazon

    Palo Alto, CA
    1 day ago
  • $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... 
    Platform
    Senior
    Full time
    Work at office
    Local area
    Remote work
    Flexible hours

    Chime

    San Francisco, CA
    2 days ago
  • $145k - $182k

    Make Your Mark: We’re looking for a Senior AI/ML Engineer to design, build, and optimize data pipelines...  ...seamless integration with different platforms and technologies.Define and manage MCP...  ..., and efficiency.Collaborate with software engineers to integrate machine learning... 
    Platform
    Senior
    Full time
    Temporary work
    Work at office
    Shift work
    3 days per week

    BlackLine

    Pleasanton, CA
    3 days ago
  •  ...money. We are looking for engineers who enjoy building and...  ...entire 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/... 
    Platform
    Senior
    Full time
    Worldwide
    Relocation

    Lightspark

    Culver City, CA
    1 day ago
  •  ...an innovative and highly skilled Software Engineer III specializing in agentic AI tool building to lead the development...  ...APIs, web scraping libraries, and ML frameworks for predictive...  ...and research workflows.Enhance F5 platform and portfolio integration, including... 
    Platform
    Senior
    Full time
    Local area

    F5 Networks

    San Jose, CA
    1 day ago
  • $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... 
    Platform
    Senior
    Local area
    Worldwide

    DataBricks

    San Francisco, CA
    1 day ago
  •  ...role Watershed is building the AI suite for companies to measure their...  ...their business. We’re looking for software engineers to help build the AI platform that powers our agents product. You...  ...experience in backend, platform, or AI/ML engineering ~ Experience building... 
    Platform
    Senior
    Work at office
    Remote work

    WaterSHED

    San Francisco, CA
    3 days ago
  • $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... 
    Platform
    Senior
    Worldwide

    Google

    San Jose, CA
    4 days ago
  • $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... 
    Platform
    Senior
    Worldwide

    Google

    Irvine, CA
    1 day ago
  • $119.8k - $234.7k

     ...Less than 25%Profession: Software EngineeringDiscipline:...  ...builds the end-to-end AI stack and is core to Azure...  ...as a Service, Azure ML, Cognitive Services, and...  ...a Principal Software Engineer - Responsible AI who is...  ...Safety and governance platforms for AI models and agents... 
    Platform
    Senior
    Ongoing contract
    Work at office
    Local area
    3 days per week

    Microsoft

    Mountain View, CA
    2 days ago
  • $159k - $178k

    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... 
    Platform
    Senior
    Full time
    Work at office
    Local area
    Work from home
    Relocation

    Flock Freight

    Encinitas, CA
    4 days ago
  • $162.15k - $259.44k

     ...our ecosystem of devices and cloud software. Like our products, we work better together...  ...at a company where you matter.Senior Software Engineer, AI Platform (Corporate AI Team)Team & Role OverviewAxon...  ...(e.g., OpenAI, AWS Bedrock, Azure ML) for fit, performance, and cost;... 
    Platform
    Senior
    Work experience placement
    Work at office

    Axon

    San Francisco, CA
    2 days ago
  • $184k - $287.5k

     ...seeking highly skilled and motivated software engineers to join us and build AI inference systems that serve large-...  ...pareto frontier for the field of ML Systems; survey recent publications...  ...containerd/CRI-O/CRIU.Experience with cloud platforms (AWS/GCP/Azure), infrastructure as... 
    Platform
    Senior
    Full time

    Nvidia

    Santa Clara, CA
    1 day ago
  • $224k - $356.5k

     ...unlimited potential of AI to define the next era...  ...AI team is building the software stack that makes large...  ...experience. We own the platform — performance, CI/CD pipelines...  ...Science, Computer Engineering, Electrical Engineering...  ...in GPU computing, ML systems, or high-performance... 
    Platform
    Senior
    Full time
    Local area

    Nvidia

    Santa Clara, CA
    1 day ago
  • $174k - $252k

     ...development, and optimization of software components critical for Large...  ...integration across core platform services, including billing,...  ...Kubernetes, Google Kubernetes Engine (GKE)), networking infrastructure...  ...development, including AI/ML applications.Preferred qualifications... 
    Platform
    Senior

    Google

    Sunnyvale, CA
    1 day ago
  • $193.3k - $261.5k

     ...Amazon Neuron, the software development kit used...  ...and Trainium ML accelerators. This...  ...software boundary, our engineers build systematic infrastructure...  ...what's possible in AI acceleration.As...  ...mentorship. Our senior members enjoy one-...  ...or similar platforms in production environments... 
    Platform
    Senior
    Work experience placement
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    1 day ago
  •  ...TeamDoorDash’s GenAI Platform team sits within...  ...infrastructure for Generative AI at DoorDash,...  ...and inference engines, fine-tuning and training...  ...is ideal for a senior engineer who enjoys...  ...technical bar for — ML engineers, product...  ...industry experience in software engineeringDeep... 
    Platform
    Senior
    Hourly pay
    Work at office
    Local area
    Remote work
    Flexible hours

    Doordash

    San Francisco, CA
    3 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Senior Software Engineer, AI/ML Platform. Be the first to apply!