W

Walmart

Sells groceries and essentials in stores and online

Principal Data Scientist

Natural Language Processing & LLMsPosted on Nov 11
Full-timeBengaluru, Karnataka, India

About the Job

Requirements

  • A Bachelor's, Master's, or PhD in Computer Science, Machine Learning, Statistics, Mathematics, or a related field.
  • At least 10 years of experience applying cutting-edge deep learning, with deep expertise in at least two of recommender systems, language models, multimodal models, and deep reinforcement learning.
  • Experience building large-scale GPU training and/or inference systems, including hands-on distributed GPU training with data, tensor, and pipeline parallelism, FSDP/DeepSpeed/Megatron-style frameworks, and mixed precision.
  • Experience with low-latency inference serving, such as vLLM, TensorRT-LLM, or Triton Inference Server, and model compression.
  • Ability to profile and debug performance across the stack and optimize kernels in CUDA and/or Triton.
  • A track record of taking deep learning models from idea to production at scale with measurable business impact.
  • Expertise in Python and PyTorch or JAX, and familiarity with large-scale data processing such as Spark or BigQuery.
  • Ability to lead through influence, set technical direction, and drive adoption across teams without direct authority.
  • A Bachelor's degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, or a related field and 5 years of experience in an analytics-related field; or a Master's degree in one of those fields and 3 years of analytics-related experience; or 7 years of experience in an analytics or related field.

Responsibilities

  • Define problem statements and solution direction, owning the full lifecycle from identifying high-leverage opportunities and aligning stakeholders through framing the science and leading execution to production rollout.
  • Identify patterns, problems, and root causes behind customer conversations.
  • Build alignment across product, engineering, business, and data science teams in the United States and India, using evidence to challenge long-held assumptions.
  • Translate ambiguous business goals into precise objectives, model architectures, loss functions, and offline and online evaluation frameworks.
  • Lead data engineering over billions of interactions, large-scale distributed GPU training, and inference optimization, including distillation, quantization, caching, and custom kernels.
  • Take models through A/B tests into production, measure business impact, and turn successful solutions into reusable platforms.
  • Mentor scientists and engineers, lead design and research reviews, and represent Walmart through publications, patents, and open-source work.
  • Build scalable agent evaluation frameworks using LLM-as-judge and model-based graders calibrated against human judgment, retail-specific benchmarks, simulated users and environments for multi-turn and tool-using agents, and automated red-teaming to catch regressions.
  • Build frameworks, standards, and tools that improve code quality, reliability, and operational excellence across teams.
  • Lead technical explorations, evaluate emerging technologies, and incubate solutions for future business needs.
  • Partner with product, UX, program management, and global engineering teams, fostering psychological safety, learning, and respect for diverse perspectives.
  • Embed security, performance, and fault tolerance throughout the technology stack.
  • Own and deliver solutions that improve critical business metrics, unlock growth, and create digital retail experiences for millions of customers.

About the Company

Walmart

Sells groceries and essentials in stores and online

Walmart sells groceries, household essentials, clothing, and electronics through stores and online marketplaces for consumers around the world. Its everyday low price model combines large purchasing volumes with distribution networks that move goods to stores and customer homes. Shoppers can browse online, collect orders at a nearby store, or request delivery, while third-party merchants use its marketplace and fulfillment services to reach customers. The group earns revenue from merchandise sales, memberships, advertising, and services, and also operates pharmacies, optical centers, and digital payments businesses. Walmart opened its first discount store in 1962 and is headquartered in Bentonville, Arkansas.

Company size
10,001+
Company stage
IPO
Headquarters
Bentonville, AR
Founded
1962

Recently Posted Jobs

Explore more opportunities at Walmart

See more jobs