Staff, Software Engineer

Walmart

Actively hiring
(Usa) Crossman Excellence Building Ca Sunnyvale Home Office, US Posted 73 days ago $143,000$286,000 / year

At a glance

AI generated

TL;DR

As a Staff Software Engineer at Walmart Global Tech, you will join the cutting-edge AI innovation team to lead the design and deployment of advanced generative AI systems. Your responsibilities include architecting scalable solutions using agentic frameworks like RAG, developing LLM-based agents for knowledge grounding and decision-making, and integrating external data sources with LLMs. You will work closely with engineering teams to enhance developer experience and build highly reliable AI-driven solutions. Key skills required are 8+ years of machine learning or software engineering experience, including 2+ years in generative AI or LLMs, expertise in Python, PyTorch, TensorFlow, and Hugging Face Transformers, proficiency in distributed computing and cloud platforms like GCP and Azure, and strong collaboration and mentoring abilities. This role offers the opportunity to contribute to large-scale AI initiatives at one of the world’s largest retailers.

Skills

Python PyTorch TensorFlow Hugging Face Transformers RAG frameworks Agentic Frameworks GCP Azure Kubernetes Docker CI/CD REST Go Java TypeScript

What you'll do

  • Design and develop generative AI models with a focus on RAG frameworks to enhance information retrieval.
  • Develop and optimize LLM-based agents for knowledge grounding, multi-step reasoning, and decision-making tasks.
  • Integrate external data sources like vector databases with LLMs to build dynamic, context-aware systems.
  • Create scalable, distributed AI systems focusing on performance, fault tolerance, and disaster recovery.
  • Mentor junior engineers and drive technical direction for large-scale AI initiatives.

What we're looking for

  • 8+ years of experience in machine learning and software engineering.
  • Deep expertise in Generative AI, LLMs, Agentic frameworks, and RAG technologies.
  • Proficiency in Python, PyTorch, TensorFlow, and Hugging Face Transformers.
  • Strong background in distributed computing, cloud platforms (GCP, Azure), and scalable architectures.
  • Experience in designing and deploying containerized applications with Docker and CI/CD pipelines.
  • Ability to mentor junior engineers and drive technical direction for large-scale AI initiatives.

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

More like this

Similar roles

Staff, Software Engineer

Walmart

(Usa) Gec Sunnyvale Bus Pk Bldg M Ca Sunnyvale Home Office, US 64 days ago $143,000$286,000
Java SpringBoot Kafka RabbitMQ Docker Kubernetes Azure Google Cloud Platform OpenAI Anthropic LLM APIs CI/CD DevOps RESTful web services Messaging systems Microservices Cloud Native Architecture Service Registry 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) 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 Respect Building Ca Sunnyvale Home Office, US 63 days ago $143,000$286,000
Java Spring Boot Apache Kafka Apache Spark Kubernetes Docker AWS Cassandra BigQuery HBase CI/CD Prometheus Grafana

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) Crossman Respect Building Ca Sunnyvale Home Office, US 65 days ago $143,000$286,000
Java Spring Spring Boot React TypeScript JavaScript HTML CSS Kubernetes Docker AWS GCP Azure Prometheus Grafana ELK Security最佳实践 微前端架构 Micro-frontend architectures 大数据技术(Spark, Hadoop, Flink) AI/ML集成