Senior Software Engineer - AI/ML Systems (Onsite)

Walmart Inc.; Sunnyvale (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

401(k) match

Stock purchase plan

Paid maternity and parental leave

PTO

Multiple health plans

Job summary

Walmart Inc. is seeking a seasoned software engineer for an onsite Sunnyvale role. You will design high-throughput services and streaming pipelines, collaborate with data scientists to productionize ML models, and advance AI-assisted development.

You’ll lead reviews and shape scalable, trustworthy systems used for offer selection and pricing. You’ll also own cross‑team discussions across Search, Item Page, and Recommendations, helping drive impact with data-driven experiments and

Qualifications

• 5+ years building production software with 3+ years in Java and/or Python.
• Strong foundation in distributed systems, data structures, and algorithms.
• Experience building and operating microservices (REST/GraphQL) that are highly available and scalable.
• Hands-on experience with streaming systems such as Kafka or Flink.
• Experience with a public cloud (GCP or AWS).
• Ability to explain complex technical ideas clearly to engineers and non‑engineers alike.

Responsibilities

• Design and build high-throughput, low-latency services that power offer selection and pricing at scale.
• Build and evolve the streaming and batch pipelines (Kafka, Flink, BigQuery, Airflow) that feed our algorithms and AI/ML models.
• Partner with data scientists to take ML models from experiment to production, and measure real customer impact through A/B tests.
• Build with AI agents and AI-assisted development, and help shape how the team uses them.
• Lead design and code reviews, and help us move faster, from PR to build to canary deploy.
• Build the tooling that keeps our decisions trustworthy: flip analysis, parity testing, and experiment dashboards.
• Drive technical discussions with product, science, and partner teams across Search, Item Page, and Recommendations.

Skills

Java

Python

Distributed systems

Microservices

Cloud (GCP/AWS)

Data structures & algorithms

Education

Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area

Tools

Kafka

Flink

BigQuery

Airflow

REST/GraphQL

Job description

Walmart Inc. is seeking a seasoned software engineer for an onsite Sunnyvale role. You will design high-throughput services and streaming pipelines, collaborate with data scientists to productionize ML models, and advance AI-assisted development.

You’ll lead reviews and shape scalable, trustworthy systems used for offer selection and pricing. You’ll also own cross‑team discussions across Search, Item Page, and Recommendations, helping drive impact with data-driven experiments and