Distinguished, Software Engineer -AI/ML Engineer- Walmart Connect

Walmart

Actively hiring
(Usa) Crossman Service Building Ca Sunnyvale Home Office, US Posted 56 days ago $169,000$338,000 / year

At a glance

AI generated

TL;DR

As a Distinguished Software Engineer - AI/ML at Walmart Connect, you will lead the technical vision for advanced Gen AI and ML solutions that enhance Walmart’s advertising ecosystem. Your daily responsibilities include architecting scalable retrieval-augmented generation pipelines, deploying autonomous multi-agent systems, and optimizing generative models to meet ultra-low latency requirements. You will work closely with product, engineering, and data science teams to integrate large language models into core ad surfaces like search, display, and video ads. Key skills required are deep expertise in training and deploying Large Language Models, proficiency in Python, Go, Rust, or C++, and experience with cloud-native architectures and real-time streaming pipelines. This role offers a unique opportunity to shape the future of agentic retail media at enterprise scale, leveraging Walmart’s vast first-party data for autonomous advertising experiences that drive superior ROAS and revenue growth.

Skills

Python PyTorch TensorFlow JAX LangChain LlamaIndex Kafka Flink Milvus Pinecone AWS GCP Azure Rust Go C++ LLM orchestration Retrieval-Augmented Generation Autonomous AI agents Real-Time Bidding Programmatic advertising TensorRT CI/CD

What you'll do

  • Shape the long-term AI engineering roadmap for Walmart Ads to align with the organization's vision.
  • Lead strategy and execution for pre-training, fine-tuning, and aligning foundation models for retail media.
  • Architect scalable Retrieval-Augmented Generation pipelines that integrate real-time inventory and pricing data.
  • Develop autonomous multi-agent systems for complex reasoning and dynamic optimization in advertising campaigns.
  • Drive inference optimization to ensure generative models are cost-efficient and meet ultra-low latency requirements.
  • Ensure AI capabilities are decoupled yet integrated with core ad servers, handling millions of queries per second.

What we're looking for

  • 12+ years of ML engineering experience with at least 5 years in a senior technical leadership role.
  • Deep expertise in training, fine-tuning, and deploying Large Language Models (LLMs) and Foundation Models into production.
  • Proven track record designing complex systems using LLM orchestration frameworks and multi-agent architectures.
  • Strong background in designing highly available, low-latency distributed systems handling 100K+ queries per second.
  • Expert proficiency in Python, Go, Rust, or C++, alongside modern ML frameworks like PyTorch, JAX, TensorFlow.
  • Experience with scalable vector databases, feature stores, and real-time streaming pipelines.
  • Strong foundational knowledge of programmatic advertising, RTB, click-through rate prediction, and dynamic creative optimization.

Market check

Salary context

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

Peer median band

$160,450$258,750

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$176,000$246,150

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

Distinguished, Software Engineer -AI/ML Engineer – Agentic Systems

Walmart

(Usa) Crossman Excellence Building Ca Sunnyvale Home Office, US 70 days ago $169,000$338,000
AI/ML TensorFlow PyTorch MLOps Kubernetes Docker AWS Azure GCP CI/CD Prometheus Grafana Python Agentic AI systems Multi-agent frameworks LLM-based agents Reliability Engineering Cloud Native AI services Observability Distributed tracing Metrics Logs APM AI-driven anomaly detection Infrastructure as code Service mesh architectures API gateways

Sr. Software Engineer - Applied AI

GEICO

Remote (Ca Palo Alto Office, US) 42 days ago $80,000$215,000
Python LangChain HuggingFace OpenAI Kubernetes CI/CD Docker Prometheus Grafana PostgreSQL Redis Apache Kafka Spring AI LangGraph LangSmith LlamaIndex Anthropic APIs Vector databases Knowledge graphs Java Spring生态系统
Remote

Lead AI Software Engineer- (Hybrid Baltimore, MD)

T. Rowe Price

Baltimore, Md, US 17 days ago $145,000$247,000
AWS Python Java JavaScript Kubernetes Docker CI/CD Prometheus Grafana PostgreSQL Redis Git GitHub Jenkins Ansible Terraform Linux RESTful APIs JSON YAML Swagger GraphQL Selenium JUnit Mockito Spring Boot Microservices Maven Gradle Docker Compose Kafka Zuul Elasticsearch Logstash Kibana ELK Stack GitLab Confluence Jira Slack Zoom

Software Development Engineer (AI & ML)

Fiserv

Alpharetta, Georgia, US 12 days ago
Python TensorFlow Keras Scikit-learn Apache Kafka Airflow Spark Docker Kubernetes PostgreSQL MongoDB Cassandra SQL NoSQL

Senior Software Engineer - Applied AI/ML

Motorola Solutions

Chicago, Il, US 15 days ago $135,000$155,000
Python SQL Docker Kubernetes AWS Azure GCP MLOps CI/CD PyTorch Tensorflow Databricks MLFlow AWS SageMaker Hugging Face Apache Airflow Temporal RF rRay

Sr. Software Engineer - AI

Adobe

Lehi, US 73 days ago $139,000$139,000
JavaScript React TypeScript Python NodeJS AWS Azure Lit CI/CD DOM WebSecurity Asynchrony Workers PerformanceOptimization GenerativeAI CloudAutomation