Data Scientist III

Walmart Walmart · Retail · Bentonville, AR

Data Scientist III at Walmart Global Tech focused on driving initiatives across forecasting, pricing, and space optimization. The role involves implementing MLOps solutions and designing, developing, and deploying AI agents leveraging LLMs and advanced ML/DL models to serve millions of customers. Responsibilities include problem formulation, data exploration, advanced AI/ML development, model deployment with MLOps best practices, and research in AI/ML. Requires expertise in Python, ML libraries, cloud environments (GCP), MLOps tools, ML pipelines, ETL/ELT, transformers, embeddings, and DevOps. Nice-to-haves include agentic frameworks, RAG, safety mechanisms, experimentation platforms, and serverless architecture.

What you'd actually do

  1. Design and develop state-of-the-art algorithms, AI/ML models, and optimization techniques to generate actionable business insights.
  2. Design, develop, and deploy AI agents leveraging LLMs and advanced ML/DL models, that serve millions of our customers globally.
  3. Validate and monitor AI/ML models post-deployment, leveraging MLOps best practices such as CI/CD pipelines, performance tuning, and effective systems integration.
  4. Identify, evaluate, and leverage diverse datasets.
  5. Translate complex business challenges into analytical solutions.

Skills

Required

  • Python
  • Pandas/Polars
  • NumPy
  • PySpark
  • TensorFlow
  • PyTorch
  • GPU utilization
  • software development
  • mathematical/scientific computing
  • deploying large-scale AI/ML/DS solutions
  • parallelized cloud compute environments
  • GCP
  • MLOps concepts
  • Google Vertex
  • MLFlow
  • ClearML
  • building and orchestrating ML pipelines
  • ETL/ELT data pipelines
  • data-driven applications at scale
  • transformers
  • embeddings
  • DevOps tools
  • Git/GitHub/GitHub Actions
  • infrastructure-as-code methodologies
  • Master’s degree in computer science, Machine Learning, Operations Research, Statistics, Optimization, Data Analytics, Mathematics, or a closely related quantitative field.
  • 3 years of professional experience in an ML or Optimization Model development-focused role.

Nice to have

  • agentic and multi-agent frameworks
  • tool-use
  • retrieval-augmented generation
  • policy learning
  • safety mechanisms
  • evaluation harnesses
  • experimentation platforms
  • sequential decisioning
  • A/B testing
  • contextual bandits
  • designing metrics for uplift measurement
  • serverless architecture
  • microservices implementation
  • API Gateway
  • FastAPI
  • VM-hosted solutions
  • logistics
  • supply chains
  • delivery operations
  • PhD in Computer Science, Machine Learning Operations Research, Statistics, Optimization, Data Analytics, Mathematics, or a comparable quantitative field.
  • publications in leading academic conferences and journals
  • open-source code contributions
  • intellectual property generation

What the JD emphasized

  • deploying large-scale AI/ML/DS solutions
  • deploying AI agents leveraging LLMs
  • MLOps solutions
  • advanced ML/DL models

Other signals

  • deploying AI agents leveraging LLMs
  • MLOps solutions
  • advanced ML/DL models
  • scalable, data-driven solutions
  • deploying large-scale AI/ML/DS solutions