Position Summary...
What you'll do...
About team:
Walmart’s AI & Data organization is building platform capabilities and enterprise services that accelerate the journey from idea to AI-powered solution safely and cost-effectively for builders across the enterprise. This includes foundational data platforms, ML infrastructure, AI productization, governance, and Associates builder solutions that enable millions of associates, and data practitioners.
We are looking for a Staff Data Scientist to set the technical direction for GenAI, agentic AI, and ML initiatives that accelerate the journey from idea to AI-powered solution across the enterprise — safely, cost-effectively, and at Walmart scale. You will architect production-scale, multi-agent and stateful AI systems on top of our foundational data platforms and ML infrastructure, lead applied research on frontier techniques, evolve our LLMOps and governance capabilities in partnership with platform teams,



and mentor data scientists and ML engineers across levels — multiplying the impact of the AI productization and Associates builder solutions that empower millions of Walmart associates and data practitioners.
What you will do
Set technical direction for GenAI, agentic AI, and AI/ML initiatives across the team; partner with engineering, product, and leadership on multi-quarter roadmaps
Architect end-to-end, production-scale GenAI and agentic AI systems — including multi-agent orchestration, tool integration, long-term memory, retrieval, evaluation, observability, safety, and cost optimization — using frameworks such as LangChain, LangGraph, or equivalents
Lead applied research and rapid prototyping on frontier techniques such as advanced reasoning and planning, fine-tuning (SFT, LoRA/QLoRA, DPO, RLHF/RLAIF), distillation, multi-modal models, and emerging agentic patterns
Partner with platform and infrastructure teams to evolve MLOps and LLMOps capabilities, including distributed training, low-latency serving, evaluation

📌 (IND) STAFF, DATA SCIENTIST (Bengaluru)
🏢 Walmart
📍 Bengaluru