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

Senior Software Engineer, AI/ML Platform [Remote]

$197k - $307k
Full-time

Agility Robotics

California
  • Remote job
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 16 days ago
Similar jobs that could be interesting for youBased on the Senior Software Engineer, AI/ML Platform [Remote] 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... 
    Platform
    Senior

    AMD

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

    Google

    Mountain View, CA
    23 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
    23 hours ago
  • $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)... 
    Platform
    Senior
    Work at office
    Remote work
    Flexible hours

    Eightfold

    Santa Clara, CA
    4 days ago
  • $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... 
    Platform
    Senior
    Internship
    Local area
    Flexible hours

    Amazon

    Palo Alto, CA
    4 days 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
    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
    4 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
    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
    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
    15 hours 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
    4 days ago
  •  ...Walmart Inc. in Sunnyvale, CA is seeking a Senior Software Engineer to lead scalable software delivery, owning feature development, testing, and...  ...support within cross-functional teams. You will drive AI/ML integration, solution design, and technical leadership to align... 
    Platform
    Senior

    Jobleads-US

    Sunnyvale, CA
    4 days 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
    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
    1 day ago
  •  ...Qualcomm Atheros, Inc. seeks a Senior Software Engineer to design, develop, and validate embedded and cloud edge software for wireless networking platforms, including AI/ML-based traffic optimization and high-performance data paths on Linux. You will collaborate with... 
    Platform
    Senior

    Jobleads-US

    Santa Clara, CA
    2 days ago
  • $123k - $190.9k

     ...DescriptionThe Cyber Analytics and AI Innovation team is...  ...by a passion for AI/ML technology and a...  ...delivering exceptional platforms and services to our business...  ...team of architects, engineers, data scientists, and product...  ...pipelines.Strong software engineering background... 
    Platform
    Senior
    Full time
    Work experience placement
    Work at office
    Local area

    Visa

    San Mateo, CA
    1 day 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

    Sunnyvale, CA
    2 days ago
  • $144k - $270k

     ...Senior Software Engineer, Applied AI Cambridge, MA USA; San Francisco, CA USA Your Impact at LILA...  ...generation of our AI-driven scientific platform. In this role, you will design and optimize...  ..., and vectorized data for AI/ML workloads. Database & Knowledge Systems... 
    Platform
    Senior
    Full time
    Work at office
    Local area
    Flexible hours

    Lila Sciences

    San Francisco, CA
    3 days ago
  • $220k - $300k

     ...operate. By combining frontier agentic AI, an enterprise-grade platform, and deep domain expertise, we're...  ...Representative Projects Context Engineering & Agent Infrastructure. Build the...  ...with at least 1+ year focused on AI/ML engineering. Staff candidates will typically... 
    Platform
    Senior

    Harvey

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

    Lightspark

    Los Angeles, CA
    1 day ago
  • $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... 
    Platform
    Senior
    Full time
    Flexible hours

    Chime

    San Francisco, CA
    21 days ago
  •  ...Forward-Deployed Engineering Role With Ai At Its Core This is a...  ...AI researcher or an ML specialist. You are expected...  ...Deliver reliable software with measurable...  ...architectural decisions to senior leadership with...  ...campaign analytics, A/B platforms Customer Experience... 
    Platform
    Senior

    U.S. Auto Parts

    Long Beach, CA
    4 days ago
  • $186.07k - $218.9k

     ...learn more about working at Coinbase. As a Senior Software Engineer on the Platform Core Automation team, you'll build the AI infrastructure and agentic systems that automate...  ...building LLM-based applications or AI/ML systems in production, including prompt engineering... 
    Platform
    Senior
    Local area

    Coinbase

    Sacramento, CA
    4 days ago
  • $160.5k - $200k

     ...Senior Software Engineer- AI for RTL Design & Formal Verification San Jose, California, United States...  ...Astera Labs’ Intelligent Connectivity Platform integrates CXL®, Ethernet, NVLink, PCIe...  ...Out Direct experience applying AI/ML techniques to EDA or CAD flow problems... 
    Platform
    Senior
    For contractors
    Flexible hours

    Jobleads-US

    San Jose, CA
    4 days ago
  • 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-... 
    Platform
    Senior
    Work experience placement

    C3 IoT

    Redwood City, CA
    4 days ago
  • $174k - $252k

     ...Senior Software Engineer, Search Ecosystems, AI/ML Share Senior Software Engineer, Search Ecosystems, AI/ML corporate_fare Google place Mountain View...  ...end technical integration of the Expert Intelligence platform with flagship Google GenAI products (e.g.,... 
    Platform
    Senior

    Jobleads-US

    Mountain View, CA
    4 days ago
  • $182k - $242k

     ...Senior Software and AI Engineer Livingston, NJ / New York, NY / Sunnyvale, CA / San Francisco, CA / Bellevue...  ...by pioneers, CoreWeave delivers a platform of technology, tools, and teams that...  ...shipping production-grade AI/ML/LLM-based applications, or strong desire... 
    Platform
    Senior
    Full time
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    San Francisco, CA
    23 hours ago
  • $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... 
    Platform
    Senior
    Full time
    Work at office
    Remote work
    Flexible hours
    Shift work

    Fivetran

    Oakland, CA
    1 day ago
  • $182k - $242k

     ...CoreWeave is The Essential Cloud for AI™. Built for pioneers by pioneers, CoreWeave delivers a platform of technology, tools, and...  ...at  About the Role As a Senior Software Engineer II (IC4) on the AI Workload...  ...with GPU-based workloads, ML training, or inference pipelines... 
    Platform
    Senior
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    Sunnyvale, CA
    a month ago
  • $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... 
    Platform
    Senior
    Full time
    Work experience placement
    Work at office
    Local area
    Remote work
    2 days per week
    3 days per week

    Everlaw

    Oakland, CA
    1 day ago

Do you want to receive more vacancies?

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