Company Description

Walmart Global Tech is at the forefront of AI innovation, leveraging cutting-edge technologies to redefine customer experiences. We are seeking a Senior Machine Learning Engineer with deep expertise in Generative AI, LLMs, and scalable system design to drive impactful solutions.

Job Description

We are seeking a Senior Machine Learning Engineer to design and deploy AI systems using LLMs, RAG frameworks, and advanced agentic techniques. In this role, you will focus on building scalable pipelines, integrating external data sources, and collaborating with teams to deliver high-impact AI-driven products.

Key Responsibilities:

AI Development & Agentic Frameworks:

• Design and implement Generative AI solutions utilizing RAG for summarization, and knowledge retrieval.
• Build and integrate LLM-based agents that utilize vector search technologies to provide dynamic, real-time responses.
• Fine-tune and optimize models for specific business needs and tasks.

System Design & Scalability:

• Contribute to building scalable, distributed systems that integrate machine learning pipelines.
• Translate high-level designs into detailed solutions, including mock-ups, pseudo-code, and modular components.
• Design and implement MVPs to validate and refine AI-driven features.

Collaboration & Integration:

• Support architectural decisions to ensure alignment with both functional and non-functional requirements.
• Participate in code reviews and ensure adherence to best practices.

Leadership & Mentorship:

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

Qualifications

Required Skills & Experience:

• 6+ years of experience in machine learning engineering, with a 2+ years focus on Generative AI and NLP.
• Proficiency in Python, ML frameworks (PyTorch, TensorFlow), and NLP libraries like Hugging Face Transformers.
• Hands-on experience with RAG frameworks, vector search tools and knowledge grounding techniques.
• Familiarity with distributed systems, scalable architectures, and cloud platforms (GCP, Azure).
• Strong understanding of software design patterns, disaster recovery strategies, and non-functional requirements.

Preferred Qualifications:

• Experience with multi-modal models and advanced optimization techniques.
• Experience with Generative AI models (e.g., GPT, Gemini) and their fine-tuning.
• Contributions to open-source projects or AI research publications.
• Strong problem-solving and debugging skills in distributed environments.

Additional Information

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 that is primarily in office coupled with virtual when not onsite. Our campuses serve as a hub to enhance collaboration, bring us together for purpose and deliver on business needs. This approach helps us make quicker decisions, remove location barriers across our global team and be more flexible in our personal lives.