The Foundation Models team builds shared, reusable AI that the rest of Walmart builds on. Our models serve Walmart US, International markets, and Sam's Club in many areas: online search and recommendations, sponsored products, catalog understanding, store and inventory planning and optimization, and Walmart's conversational agents for customers, sellers, engineers, and associates. You'll own the full lifecycle: spotting the highest-leverage bets, getting stakeholders on board, framing the science, and leading execution through data engineering, large-scale training, inference optimization, and production rollout. We'll judge success by what ships and what customers and the business can measure, not by prototypes.
Find where foundation models can bring step-change gains across Walmart, then write the science vision and multi-quarter roadmap.
Build alignment across product, engineering, ads, merchandising, and supply chain teams in the US and India. Turn ambiguous business goals into precise objectives, model architectures, loss functions, and evaluation frameworks, both offline and online.
Execute end to end: Lead data engineering over billions of interactions, large-scale distributed GPU training, and inference optimization (distillation, quantization, caching, custom kernels).
Bachelor's, Master's, or PhD in Computer Science, Machine Learning, Statistics, Mathematics, or a related field.
10+ years of experience applying cutting-edge deep learning, with deep expertise in at least two of: Recommender Systems, Language Models, Multimodal Models, Deep Reinforcement Learning.




Experience building large-scale GPU training and/or inference systems:
Hands-on experience with distributed GPU training, including data, tensor, and pipeline parallelism, FSDP/DeepSpeed/Megatron-style frameworks, and mixed precision.
Experience with low-latency inference serving (e.g. vLLM, TensorRT-LLM, Triton Inference Server) and model compression.
Able to profile and debug performance across the stack, and to optimize kernels in CUDA and/or Triton.
Expert in Python and PyTorch (or JAX), and comfortable with large-scale data processing (Spark, BigQuery, or similar).
That's what we do at Walmart Global Tech. We're a team of software engineers, data scientists, cybersecurity expert's and service professionals within the world's leading retailer who make an epic impact and are at the forefront of the next retail disruption. People are why we innovate, and people power our innovations. We are people-led and tech-empowered.
We train our team in the skillsets of the future and bring in experts like you to help us grow. Here, you can kickstart a great career in tech, gain new skills and experience for virtually every industry, or leverage your expertise to innovate at scale, impact millions and reimagine the future of retail.
Flexible,



hybrid work
Walmart's culture sets us apart, and we know being together helps us innovate, learn and grow great careers. This role is based in our [Bangalore/Chennai] office for daily work, with the flexibility for associates to manage their personal lives.
Beyond our excellent compensation package, you can receive incentive awards for your performance. Other great perks include a host of best-in-class benefits maternity and parental leave, PTO, health benefits, and much more.
By fostering a workplace culture where everyone is—and feels—included, everyone wins. By making Walmart a welcoming place where all people feel like they belong, we're able to engage associates, strengthen our business, improve our ability to serve customers, and support the communities where we operate.
Walmart, Inc., We believe we are best equipped to help our associates, customers and the communities we serve live better when we really know them. That means understanding, respecting and valuing unique styles, experiences, identities, ideas and opinions – while being inclusive of all people.
Bachelors degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology or related field and 5 years' experience in an analytics related field. Option 2: Masters degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology or related field and 3 years' experience in an analytics related field. Option 3: 7 years' experience in an analytics or related field.

📌 Principal Data Scientist - Tech (Bengaluru)
🏢 Walmart Global Tech India
📍 Bengaluru