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

Staff Software Engineer, ML

$285k - $340k

Apptronik

Apptronik is a human-centered robotics company developing AI-powered robots to support humanity in every facet of life. Our flagship humanoid robot, Apollo, is built to collaborate thoughtfully with people, starting with critical industries such as manufacturing and logistics, with future applications in healthcare, the home, and beyond.

We operate at the cutting edge of Applied AI, applying our expertise across the full robotics stack to solve some of society's most important problems. You will join a team dedicated to bringing Apollo to market at scale, tackling the complex challenges like safety, commercialization, and mass production to change the world for the better.

JOB SUMMARY

Apptronik is seeking a Staff MLOps Engineer to own the technical direction of our MLOps platform — the system of record for datasets, experiments, model artifacts, and serving paths that connects teleoperation data collection on one side to deployed autonomy on Apollo on the other. In this role, you will set the architecture for the platform layer above the training cluster: dataset lifecycle, experiment tracking, model registry, evaluation harnesses, and the serving / packaging path that delivers trained policies to robots in the field. You will lead by influence across MLOps, Autonomy, Data Platform, and TeleOp — establishing the standards, contracts, and tooling that turn one-off research code into a repeatable, auditable pipeline from data to deployed model. This is a hands-on technical leadership role, not a management position; you will be a primary contributor while mentoring the engineers around you, and partnering closely with the Training Infrastructure engineer who owns the cluster layer beneath the platform.

ESSENTIAL DUTIES AND RESPONSIBILITIES

Platform Architecture & Ownership

  • Technical Direction:  Own the technical direction for the MLOps platform — define subsystem interfaces, drive architecture decisions, and establish engineering standards for how datasets, experiments, and models move through Apptronik's systems.
  • Cross-Team Authority:  Serve as the primary technical point of contact for Autonomy, Data Platform, and TeleOp on all matters of model lifecycle and platform contracts.

Dataset Lifecycle & Versioning

  • Versioning & Lineage:  Design and operate the dataset layer end-to-end — versioning, lineage, splits, and labeling-integration handoff.
  • Reproducibility:  Ensure every trained model can be traced back to the exact data and code that produced it.

Model Registry & Artifact Management

  • Registry:  Build and operate a first-class model registry — versioned artifacts, metadata, evaluation results, lineage, and approval workflows.
  • Promotion Path:  Define the promotion path from "trained" to "qualified" to "deployed to robot."

Evaluation & Qualification Harnesses

  • Automated Evaluation:  Define the offline benchmarks, simulation rollouts, and policy-gating harnesses that any model must pass before reaching Apollo.
  • Metrics Framework:  Develop the metrics framework that the autonomy team trusts to gate releases.

Serving, Packaging & Deployment to Robot

  • On-Robot Path:  Own the path from registered model to running inference on Apollo — packaging (ONNX, TensorRT, torch.compile), versioning on-robot, rollback, and observability of deployed policy behavior.
  • Telemetry Seam:  Coordinate with Connect and Data Platform on the deploy-and-telemetry seam back from the fleet.

Mentorship & Cross-Functional Leadership

  • Mentorship:  Mentor mid-level and senior engineers on the MLOps team through code review, design review, and direct collaboration.
  • Influence:  Partner with the Training Infrastructure engineer on the cluster/platform contract, and influence research workflows across Autonomy to standardize on the platform's primitives.

SKILLS AND REQUIREMENTS

  • Deep proficiency in Python and at least one systems-level language (Go, Rust, or C++), with demonstrated ability to make and defend architectural tradeoffs in production ML platforms
  • Proven experience owning and delivering an MLOps platform end-to-end — dataset lifecycle, experiment tracking, model registry, evaluation, and serving — at a company that ships models to production
  • Expertise across the model lifecycle: dataset versioning (DVC, LakeFS, Delta, or equivalent), experiment tracking (MLflow, W&B, Determined), model registry, and policy serving
  • Strong background designing service-oriented systems on Kubernetes; comfortable with the contract between platform APIs and underlying compute infrastructure
  • Experience defining evaluation and qualification frameworks for ML models where the cost of a regression is high (robotics, safety-critical, or production-customer-facing)
  • Experience leading technical projects end-to-end: architecture, implementation, validation, and iteration
  • Demonstrated ability to lead by influence across teams — setting standards that other engineers adopt voluntarily, and mentoring engineers around you
  • Proficiency with cloud infrastructure (AWS, GCP, or Azure), Docker, Git, and modern CI/CD workflows

EDUCATION and/or EXPERIENCE

  • Master's degree in Computer Science, Machine Learning, or a related technical field preferred; Bachelor's considered with exceptional experience.
  • 8+ years of professional software engineering experience in ML platforms or related infrastructure, OR 4+ years of direct, hands-on experience owning an MLOps platform that shipped models to production.

Preferred Qualifications:

  • Experience deploying ML models to edge or embedded targets (on-device inference, ONNX Runtime, TensorRT, robot fleets)
  • Experience with RL training and evaluation infrastructure for embodied agents (rollout workers, replay buffers, sim-eval harnesses)
  • Familiarity with humanoid robotics, dexterous manipulation, or teleoperation data domains
  • Experience with simulation-in-the-loop evaluation (IsaacSim, MuJoCo, or equivalent)
  • Familiarity with policy gating, shadow deployments, or staged rollout strategies for autonomy
  • Open-source contributions to MLOps platform tooling (MLflow, BentoML, KServe, Ray Serve, etc.)

PHYSICAL REQUIREMENTS

  • Prolonged periods of sitting at a desk and working on a computer
  • Must be able to lift 15 pounds at times
  • Vision to read printed materials and a computer screen
  • Hearing and speech to communicate
  •  

The annual salary range is $285,000 - $340,000

 

 

*This is a direct hire. Please, no outside Agency solicitations. 

Apptronik provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.

Vacancy posted 6 days ago
Similar jobs that could be interesting for youBased on the Staff Software Engineer, ML in Sunnyvale, CA vacancy
  • $240k - $265k

     ...leveraging its commercial self-driving software to develop, test and deploy autonomous...  ...Department of Defense.We are looking for a Staff Software Engineer to help build and improve behavior...  ...across Prediction, Controls, Perception, ML, Simulation, Systems, and Safety.Define... 
    Suggested
    Temporary work
    Work at office
    Visa sponsorship

    Kodiak Robotics

    Mountain View, CA
    2 days ago
  • $143k - $286k

     ...PermanentCompany: WalmartBusiness Segment: Home OfficeRole summary:The Staff Software Engineer is responsible for leading the design, development, testing,...  ...fostering cross-functional collaboration, and leveraging AI/ML technologies to drive innovation. The position emphasizes... 
    Suggested
    Full time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    3 days ago
  • $190.9k - $334.1k

     ...DescriptionIt all started when engineer Fred Luddy wrote code that...  ...and throughput. Apply AI and ML to forecasting, anomaly detection...  ...Write and maintain declarative, software-defined infrastructure at...  ...leadership. Level expectations Senior Staff : Leads the design and... 
    Suggested
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Santa Clara, CA
    4 days ago
  •  ...inference systems that combine GPU-accelerated prefill with ultra-fast decode on the Cerebras Wafer-Scale Engine.We are hiring a Software Engineer to build and evolve the ML API layer that makes this heterogeneous serving system accessible, reliable, and easy to use. You... 
    Suggested

    Cerebras Systems

    Sunnyvale, CA
    4 days ago
  • $207k - $300k

     ...the systems, while mentoring engineers in measurement best practices....  ...experience. 8 years of experience in software development.5 years of...  ...5 years of experience leading ML design and optimizing ML infrastructure...  ...forward.We are seeking a Staff Software Engineer with a deep... 
    Suggested

    Google

    Mountain View, CA
    2 days ago
  • $176.1k - $308.2k

     ...DescriptionIt all started when engineer Fred Luddy wrote code that automated...  ...for people.Job DescriptionAs a Staff Engineer, you will...  ...junior engineers. They combine deep software engineering fundamentals with hands-on expertise in AI/ML integration, cloud architecture... 
    Work experience placement
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Santa Clara, CA
    2 days ago
  • $110k - $220k

     ...WalmartBusiness Segment: Home OfficeRole summary:Join Walmart as a Staff Software Engineer to lead the design, development, and deployment of scalable...  ...capabilities, including infrastructure, tooling, and AI/ML frameworks. This role drives the full software lifecycle,... 
    Full time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    4 days ago
  •  ...analyses to evaluate autonomous driving software performance across the autonomy stack.Lead...  ...functional efforts with autonomy, systems engineering, simulation, and data teams to embed...  ...decisions.Invent and drive new statistical and ML methods, and ML introspection techniques,... 
    Full time
    Local area
    Work from home

    General Motors

    Sunnyvale, CA
    3 days ago
  • $262k - $364k

     ...timelines. Influence and coach a distributed team of engineers.Drive technical project strategy, lead large-scale ML infrastructure optimization, and oversee the...  ...equivalent practical experience.8 years of experience in software development.7 years of experience leading... 

    Google

    Mountain View, CA
    2 days ago
  •  ...prefill with ultra-fast decode on the Cerebras Wafer-Scale Engine.We are hiring a Software Engineer to productionize and optimize our GPU serving...  ...SGLang, PyTorch, Triton, TensorRT-LLM, or another open-source ML systems project.Experience optimizing prefill-heavy or... 

    Cerebras Systems

    Sunnyvale, CA
    1 day ago
  • $207k - $300k

     ...practical experience.8 years of experience in software development.5 years of experience leading Machine Learning (ML) design and optimizing ML infrastructure (e.g.,...  ...cross-business projects.Google Cloud's software engineers develop the next-generation technologies that change... 

    Google

    Sunnyvale, CA
    5 days ago
  • $240k - $265k

     ...leveraging its commercial self-driving software to develop, test and deploy autonomous...  ...Department of Defense.We are looking for a Staff Software Engineer to help shape how learned models are...  ...Machine Learning, working closely with ML engineers and autonomy teams to bring learned... 
    Visa sponsorship

    Kodiak Robotics

    Mountain View, CA
    2 days ago
  • $262k - $364k

     ...cross-system bottlenecks; raise the bar on engineering excellence, experimentation velocity, and...  ...experience.8 years of experience with software development in one or more programming languages...  ...architecture.8 years of experience with ML/AI algorithms and tools, deep learning or... 
    Worldwide

    Google

    Mountain View, CA
    22 hours ago
  • $207k - $300k

     ...practical experience.8 years of experience in software development.5 years of experience with...  ....5 years of experience leading ML design and optimizing end-to-end Machine...  ...qualifications:Master’s degree or PhD in Engineering, Computer Science, or a related technical... 
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $143k - $286k

     ...WalmartBusiness Segment: Home OfficeRole summary: The (USA) Staff, Software Engineer plays a critical role in delivering scalable platform capabilities...  ...expertise in cloud-native architecture, API design, and AI/ML integration to build innovative, secure, and maintainable... 
    Full time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    3 days ago
  • $207k - $300k

     ...development teams on code design hardware/software interface.Build functional and cycle-...  ...qualifications:Master’s degree or PhD in Engineering, Computer Science, or a related technical...  ...including chip design, system software, ML supercomputer, compiler, and system test.... 
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $207k - $300k

     ...timelines, and influence and coach a distributed team of engineers.Leverage AI strategically to solve complex layout...  ...practical experience.8 years of experience in software development.5 years of experience with ML design and ML infrastructure (e.g., model deployment... 

    Google

    Mountain View, CA
    1 day ago
  • $110k - $220k

     ...not eligible for remote work.Role summary:Walmart is seeking a Staff Software Engineer to lead the development and delivery of scalable platform capabilities, including infrastructure, tooling, and AI/ML frameworks. This role involves driving the full software lifecycle... 
    Full time
    Contract work
    Temporary work
    Part time
    Work at office

    Walmart

    Sunnyvale, CA
    5 days ago
  • $166.5k - $291.4k

     ...DescriptionIt all started when engineer Fred Luddy wrote code that...  ...Job DescriptionThe Role As a Staff Engineer in HRSD, you set the...  ...single feature set. This is not an ML research role; you do not...  ...QualificationsWhat You Bring 7+ years of software engineering, with a record of... 
    Contract work
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Santa Clara, CA
    4 days ago
  • $176.4k - $319.72k

     ...Price, and other leading investors.About the RoleAs a Senior/Staff Software Engineer working on driving behavior verification, you are...  ...autonomous vehicle.Leverage a combination of machine learning (ML) models and safety metrics from literature to evaluate the end... 
    Odd job
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    1 day ago
  • $262k - $364k

     ...test, deploy, maintain, and enhance large scale software solutions.Provide technical leadership on high-impact...  .... Influence and coach a distributed team of engineers.Drive technical project strategy, lead large-scale ML infrastructure optimization, and oversee the design... 

    Google

    Sunnyvale, CA
    4 days ago
  • $262k - $364k

     ...Collaborate with DeepMind to solve their ML model performance challenges, such as onboarding...  ...experience.8 years of experience in software development.7 years of experience leading...  ..., Triton, CUTLASS, etc.) and performance engineering techniques.Experience with modern LLMs... 
    Immediate start
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $189k - $274k

     ...and make mobility more efficient and accessible for all. As a Staff Software Engineer focusing on Deep Learning Acceleration at Aurora, you will...  ...performance optimization. Hands-on experience in optimizing DL/ML workloads at the framework level using at least one deep... 
    Work at office
    Local area
    3 days per week

    Aurora Innovation

    Mountain View, CA
    4 days ago
  • $143k - $286k

     ...data, and platform teamsMentor engineers and lead design and code...  ...Qualifications 8+ years of professional software engineering experienceStrong...  ...with AI/ML integration and data-driven systemsExperience...  ...StockSummaryWe are seeking a Staff Software Engineer to design and... 
    Full time
    Temporary work
    Part time
    Visa sponsorship

    Walmart

    Sunnyvale, CA
    4 days ago
  • $194.6k - $293.9k

     ...0661242Team: Machine Learning and AI, Software and Services, Operations and Supply ChainLocation...  ...2026-05-07SummaryWe are looking for a Staff level Software Engineer with experience working with the...  ...for prompt optimization and tuning ML workflowsExperience with RAG and... 
    Relocation

    Apple

    Cupertino, CA
    3 days ago
  • $143k - $286k

     ...PermanentCompany: WalmartBusiness Segment: Home OfficeRole summary: This Staff Software Engineer will lead the design and development of scalable, production...  ...the technology that enables analytics, experimentation, AI/ML, and intelligent automation across Walmart eCommerce. We... 
    Full time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    2 days ago
  • $189k - $280k

     ...systems to intuitive design, intelligent software, and next-generation safety and entertainment...  ...AI Behavior Planning group to build ML software and solutions for autonomous driving. In this hands-on Staff Software/ML Engineer role, you will partner with other ML engineers... 
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    22 hours ago
  • $143k - $286k

     ...WalmartBusiness Segment: Home OfficePlatform Engineering Execution: Delivers platform...  ..., including infrastructure, tooling, and ML frameworks (for example, CI/CD, SDKs, model...  ...resilience, and agility across systems. Software Development & Lifecycle Ownership: Drives... 
    Full time
    Temporary work
    Part time
    Local area

    Walmart

    Sunnyvale, CA
    4 days ago
  • $207k - $300k

     ...equivalent practical experience.8 years of experience in software development.5 years of experience leading ML design and optimizing end-to-end Machine Learning (...  ...projects.Experience as a full-lifecycle ML engineer in training dataset curation, modern neural architectures... 

    Google

    Mountain View, CA
    22 hours ago
  • $152k

     ...world. Role Overview The Resource Fabric Engineering team builds and operates the...  ...innovation at scale. We are looking for a Staff Software Engineer to lead the design and evolution...  ...and Compliance teams to define end-to-end ML model development lifecycle experiences.... 
    Temporary work
    Flexible hours

    Coupang

    Mountain View, CA
    4 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Software Engineer, ML. Be the first to apply!