You and the process

With a person
Anna, in-house recruiter
9 years hiring engineers
30 minutes with a real recruiter
They read your CV with you, on a call, and say where the offers are being lost.
Didn’t find what you were looking for? Tell us what to build
Walmart

Senior Data Scientist – Supply Chain Optimization

  • Office
  • 6+ years

Salary

$110,302/ year

AI summary

For members

The whole posting in a few lines. Sign up to read it here and on every role you open.

Sign Up to Read

Description

This senior role focuses on supply chain optimization, using AI agents and machine learning to improve delivery promises, fulfillment routing, and root-cause analysis.

The work moves agentic and ML systems into production and mixes LLM-based development with classical modeling on large datasets.

Responsibilities

  • Build goal-driven AI agents that improve customer delivery promises, fulfillment routing, and root-cause analysis across end-to-end supply chain workflows.
  • Create agents that reason, plan, call tools, and manage short- and long-term memory, using LangChain, LangGraph, Google ADK, and Pydantic.
  • Work with LLMs such as GPT, Claude, and Gemini, and train custom models in Hugging Face, PyTorch, or TensorFlow for NLP work covering context engineering, chain-of-thought reasoning, and tool calling.
  • Deploy AI/ML and agentic systems to production with Docker, Kubernetes, GCP or Azure, and Airflow, aiming for scalability, reliability, and low latency, backed by CI/CD pipelines for ongoing updates.
  • Build scalable RAG systems that give real-time access to logistics policies, SOPs, and network constraints.
  • Run vector databases (Pinecone, Milvus, ChromaDB) and knowledge graphs to sharpen decisions on fulfillment, routing, and issue resolution.
  • Apply classical and advanced machine learning, such as XGBoost, K-Means, regression, SVMs, and random forest, to large structured and unstructured datasets to improve delivery ETA prediction and reliability.
  • Process data and build models with Pandas, NumPy, and scikit-learn.
  • Build and tune PostgreSQL workflows for transactional data and Google BigQuery workflows for analytics, keeping data quality, consistency, and performance at the company's scale.
  • Build evaluation and observability tools with Arize Phoenix and LangFuse to track production accuracy, hallucination rate, latency, and cost for models and agents.
  • Turn research from venues such as NeurIPS, ICML, ICLR, ACL, and EMNLP into production-ready enterprise solutions for the supply chain.

Requirements

  • Bachelor's degree or equivalent in computer science, analytics, engineering (any), or a related field, plus one year of work in analytics or a similar area; or a Master's degree or equivalent in those fields.
  • Hands-on experience building deep learning architectures in Python with PyTorch, TensorFlow, and LangChain.
  • Experience pretraining and fine-tuning Hugging Face or TFHub language models for NLI and question answering on open corpora such as SQuAD and MNLI.
  • Experience with GPT-family generative models and encoder models such as BERT, RoBERTa, and DistilBERT for language reasoning.
  • Experience using computational semantics tools (WordNet, spaCy, NLTK) for linguistic analysis.
  • Experience with statistical and modeling methods such as graphical models, Bayesian inference, neural networks, SVM, and random forest, using scikit-learn and SciPy.
  • Experience validating models through testing, tuning, and metrics such as chi-square, ROC curves, and root mean square error.
  • Experience running trend, sentiment, and cluster analysis (Word2Vec, TF-IDF, K-Means, hierarchical clustering, PCA, FAMD) to answer business questions and make recommendations.
  • Experience with exploratory data analysis using Pandas, NumPy, Matplotlib, and Plotly.
  • Experience with CI/CD, testing, and documentation, including UML, state and activity workflows, Astah, and Java.
  • Graduate coursework, graduate research, or professional experience with these skills counts toward the requirements, in any amount.

Conditions

Based at 811 Excellence Dr, Bentonville, AR 72716.

Pay is $110,302.00 to $180,000.00 per year, plus annual or quarterly performance incentives.

The company is an equal opportunity employer and runs a drug-free workplace with zero tolerance for illegal drugs and alcohol on the job.

Eligibility for some benefits depends on job classification and length of employment, and benefits and plan terms may change.

Benefits

  • Competitive pay plus performance-based incentive awards.
  • Medical, vision, and dental coverage.
  • 401(k), stock purchase, and company-paid life insurance.
  • PTO including sick leave, parental leave, family care leave, bereavement, jury duty, and voting time.
  • Short-term and long-term disability.
  • Education assistance covering 100% of company-paid college degrees.
  • Company discounts, military service pay, adoption expense reimbursement, and more.

Where you’d work

From the office

About the company

Walmart

  • Industry: E-commerce

Office in United States

Also hiring in Bellevue, United States, Sunnyvale, United States, Canada and 4 more places

1 of their 219 open roles is remote

Your chances

Worth a look before you spend an evening tailoring a CV for it.

  • 22 checks run
  • 1 red flag

Still hiring?

15 checks

1 red flag

How crowded?

7 checks

No red flags

Fits Me

How well does this role fit you?

Answer a few questions or drop your CV, and every role gets a fit score with the reasons, this one first.

  • Your field
  • Level
  • Stack
  • Work model
  • Salary floor
  • Must-haves
Details12 facts · Role, Location, Compensation, Employment, Company
Tech stack
  • Python
  • PostgreSQL
  • Kubernetes
  • Docker
  • PyTorch
  • Azure
  • Airflow
  • BigQuery
  • LLMs
  • RAG
  • LangChain
  • Vector databases
  • TensorFlow
  • scikit-learn
  • NLP
Seniority
Senior
Type
Full-time
Industry
E-commerce
Specialty
Data Science
Region
United States
Show 6 more factsShow less

Role

Category
Data & Analytics
Specialty
Data Science
Seniority
Senior
Experience
1+ years
Tech stack
  • Python
  • PostgreSQL
  • Kubernetes
  • Docker
  • PyTorch
  • Azure
  • Airflow
  • BigQuery
  • LLMs
  • RAG
  • LangChain
  • Vector databases
  • TensorFlow
  • scikit-learn
  • NLP

Location

Work model
Office
Region
United States
Office
  • United States

Compensation

Salary
$110,302 / year
Pay period
Annual

Employment

Type
Full-time

Company

Industry
E-commerce

Something wrong with this vacancy?

Similar vacancies

Share this vacancy

What's wrong with it?

The employer never sees who reported.

Reason