Staff, Software Engineer

Walmart

Actively hiring

(Usa) Crossman Respect Building Ca Sunnyvale Home Office, US Posted 52 days ago $143,000–$286,000 / year

View original post Log in to save

At a glance

AI generated

TL;DR

As a Staff Software Engineer (Machine Learning Engineering) in the Spark Driver Applied AI team at Walmart, you will lead the design and development of large-scale, production-grade ML systems for driver search and matching. This role involves architecting scalable, low-latency ML systems capable of handling millions of real-time requests per second, while also leading the end-to-end productionization of ML models including serving, feature pipelines, and online inference systems. You will collaborate closely with Applied Scientists to translate advanced ML models into robust, high-performance production systems, ensuring system reliability, monitoring, and observability at Walmart scale. The ideal candidate has a Ph.D. or Master's in Computer Science and extensive experience building and scaling real-time distributed systems, proficiency in languages like Java, Python, or C++, and deep expertise in large-scale data and streaming systems such as Kafka, Spark, and Flink.

Skills

Python Java C++ Kafka Spark Flink GCP Azure ML CI/CD MLOps Docker Kubernetes Prometheus Grafana PostgreSQL Redis Git Jenkins

What you'll do

Design and build scalable, low-latency ML systems for real-time driver search and matching.
Lead the productionization of ML models, including serving, feature pipelines, and online inference systems.
Optimize distributed systems for high-throughput, real-time decision-making in driver dispatch.
Ensure system reliability, monitoring, and observability to maintain high availability at scale.
Drive best practices in MLOps, including CI/CD, model versioning, experimentation, and safe deployment.

What we're looking for

Ph.D. or Master’s in Computer Science or related field.
5+ years of experience in software engineering with focus on backend systems and ML engineering.
Proven experience building and scaling real-time, distributed systems in production environments.
Strong experience with ML production systems including model serving, feature stores, and inference pipelines.
Deep understanding of system design, scalability, and performance optimization for large-scale applications.
Proficiency in programming languages such as Java, Python, or C++.
Experience with large-scale data and streaming systems like Kafka, Spark, Flink, and cloud platforms.

Market check

Salary context

This $143,000–$286,000 range sits above 83% of similar postings on FindRole.

Peer median band

$131,650–$234,000

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$159,480–$214,500

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

Employer

About Walmart

Walmart Inc. is the world''s largest retailer by revenue, operating a chain of hypermarkets, discount department stores, and grocery stores, as well as a growing e-commerce presence through Walmart.com. Industry: General Merchandise & Grocery Retail

Walmart currently has 495 open roles on FindRole.

Listed pay typically runs $117,000–$234,000 across 487 roles with salary data.

Most-posted roles

View all roles at Walmart

Similar roles

Staff, Software Engineer

Walmart

(Usa) Crossman Service Building Ca Sunnyvale Home Office, US 35 days ago $143,000–$286,000

Java Spring Hibernate Tomcat Azure Cosmos DB Kafka Selenium Mockito JUnit TestNG Mocha Chai JIRA SonarQube Git Jenkins CI/CD Docker Kubernetes Oracle DB Reactive Programming Adaptive Programming ELK Splunk

Staff, Software Engineer

Walmart

(Usa) J Street Office Space Ar Bentonville Home Office, US 28 days ago $110,000–$220,000

Azure PostgreSQL Cosmos DB Cassandra Spanner Kubernetes CI/CD Python Docker Prometheus Grafana AWS GitOps

Staff, Software Engineer

Walmart

(Usa) Sunnyvale Tech Corners Bldg 6 Ca Sunnyvale Home Office, US 44 days ago $143,000–$286,000

Java Javascript React Node Spring CI/CD Agile Docker Kubernetes Performance Profiling Tools Multi-cloud Environments Distributed Systems Architecture CICD Automation Feature Flags Backwards Compatibility Performance Testing Frameworks Load Generation Tools Problem Solving Skills

Staff, Software Engineer

Walmart

(Usa) Crossman Excellence Building Ca Sunnyvale Home Office, US 29 days ago $143,000–$286,000

Java Spring NoSQL Apache Hadoop MapReduce Apache Spark Kafka Machine Learning Data Science CI/CD Agile Scalability Distributed Systems Fault Tolerance AWS Google Cloud Azure

Staff, Software Engineer

Walmart

(Usa) Crossman Service Building Ca Sunnyvale Home Office, US 41 days ago $143,000–$286,000

Java Spark Cassandra Kafka Redis Distributed Systems Big Data Azure Google Cloud Platform CI/CD

Staff, Software Engineer

Walmart

(Usa) David Glass Technology Center Property Costs Ar Bentonville Home Office, US 44 days ago $110,000–$220,000

Java Spring Spring Boot Spring MVC JUnit Mockito Docker Jenkins RDBMS NoSQL Wireshark Charles Kubernetes Splunk Confluence Apache Kafka Git GitHub Maven Azure Google Cloud Platform