(IND) STAFF, SOFTWARE ENGINEER / Staff ML Engineer

Position Summary...

What you'll do...

About Team

Walmart’s Global Sourcing Technology team is responsible for building and maintaining software that enables Walmart to Source the best quality products at the lowest possible price and pass that savings to our customers so they can Save Money and Live Better. Walmart is investing in transforming the Global Sourcing business by enhancing the user experience with the latest technology and we are looking for talented individuals who can use their engineering and leadership skills to help our teams solve complex problems. You will be part of the application development team leading software design and implementation, problem solving, as our customers evolve and adapt, we are taking it a few notches further here. We’re changing what customers can expect from the experience of shopping, from the physical stores, to mobile, social and even online; we’re not just ready for the future of shopping, we’re creating it.
We take great pride in our entrepreneurial culture and the strength of our engineering team. If that sounds exciting to you, we want to talk to you about joining us.

Walmart Global Tech is at the forefront of AI innovation, leveraging cutting-edge technologies to redefine customer experiences. We are seeking a Staff Machine Learning Engineer with deep expertise in Generative AI, LLMs, and RAG frameworks to lead the design, development, and deployment of advanced AI systems. This role involves architecting scalable solutions, integrating agentic frameworks, and mentoring engineering teams to push the boundaries of AI-driven innovation.

 

What You Will Do:

AI Model Development & Agentic Frameworks:

• Design, fine-tune, and deploy Generative AI models with a focus on RAG frameworks to enhance information retrieval and generation.
• Develop and optimize LLM-based agents for tasks like knowledge grounding, multi-step reasoning, and decision-making.
• Integrate external data sources (e.g., vector databases) with LLMs to build dynamic, context-aware systems.

Architecture & Scalability:

• Architect scalable, distributed AI systems with a focus on performance, fault tolerance, and disaster recovery.
• Develop detailed design specifications, including modular, extensible, and functional components.
• Build MVPs to validate requirements and refine designs iteratively.

Collaboration & Product Integration:

• Partner with product teams to integrate AI into customer-facing applications.
• Work cross-functionally to design solutions that address both functional and non-functional requirements.

Leadership & Mentorship:

• Drive technical direction for large-scale AI initiatives and mentor junior engineers.
• Establish best practices for building and maintaining AI-driven systems.

 

What You will Bring

• Overall 9+ years of experience in machine learning / software engineering, with 2+ years in Generative AI or LLMs.
• Strong experience with RAG frameworks, vector search technologies and knowledge grounding techniques.
• Expertise in Python, ML frameworks (PyTorch, TensorFlow), and libraries like Hugging Face Transformers.
• Proficiency in distributed computing, cloud platforms (GCP, Azure), and scalable architectures.
• Familiarity with disaster recovery principles and building resilient, fault-tolerant systems.

 

Preferred Qualifications:

• Experience with multi-modal models (text, image, video) and advanced optimization techniques (e.g., quantization, distillation).
• Experience with Generative AI models (e.g., GPT, Gemini) and their fine-tuning.
• Contributions to open-source projects or publications in Generative AI, RAG, or agentic frameworks.
• Strong understanding of system design, non-functional requirements, and design patterns.

About Walmart Global Tech
Imagine working in an environment where one line of code can make life easier for hundreds of millions of people.  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. We have roles for those chasing their first opportunity as well as those looking for the opportunity that will define their career. 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
We use a hybrid way of working with primary in office presence coupled with an optimal mix of virtual presence. We use our campuses to collaborate and be together in person, as business needs require and for development and networking opportunities. This approach helps us make quicker decisions, remove location barriers across our global team, be more flexible in our personal lives.

Benefits

Beyond our great 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.

Belonging

We aim to create a culture where every associate feels valued for who they are, rooted in respect for the individual. Our goal is to foster a sense of belonging, to create opportunities for all our associates, customers and suppliers, and to be a Walmart for everyone.

At Walmart, our vision is "everyone included." By fostering a workplace culture where everyone is—and feels—included, everyone wins. Our associates and customers reflect the makeup of all 19 countries where we operate. 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.

Minimum Qualifications...

Outlined below are the required minimum qualifications for this position. If none are listed, there are no minimum qualifications.

Minimum Qualifications:Option 1: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 4 years’ experience in software engineering or related area.Option 2: 6 years’ experience in software engineering or related area.

Preferred Qualifications...

Outlined below are the optional preferred qualifications for this position. If none are listed, there are no preferred qualifications.

Master’s degree in Computer Science, Computer Engineering, Computer Information Systems, Software Engineering, or related area and 2 years' experience in software engineering or related area

Primary Location...

Pardhanani Wilshire Ii, Cessna Business Park, Kadubeesanahalli Village, Varthur Hobli , India