Senior, Software Engineer - Gen AI

World's leading retailer transforming through technology, focusing on innovative solutions to help people save money and live better.
$117,000 - $234,000
Machine Learning
Senior Software Engineer
Hybrid
5,000+ Employees
6+ years of experience
AI · Enterprise SaaS · Retail

Description For Senior, Software Engineer - Gen AI

Walmart Global Tech is seeking a Senior Machine Learning Engineer to join their innovative AI team. This role focuses on developing cutting-edge Generative AI solutions and Large Language Models (LLMs) to transform retail experiences. As a senior engineer, you'll work on designing and implementing scalable AI systems, utilizing RAG frameworks, and building sophisticated machine learning pipelines that impact millions of customers.

The position offers an exciting opportunity to work at the intersection of retail and artificial intelligence, applying advanced technologies like LLMs and vector search to solve real-world problems at scale. You'll be responsible for developing AI-driven features, optimizing models, and establishing best practices while working with a talented team of engineers.

Walmart provides a comprehensive benefits package including competitive salary, performance bonuses, health benefits, 401(k) matching, and stock purchase options. The work environment is hybrid, combining in-office collaboration with flexible remote work options. The company's Silicon Valley location offers access to a vibrant tech ecosystem and various cultural and recreational activities.

This role is perfect for experienced ML engineers passionate about Generative AI who want to make a significant impact by building AI solutions that serve hundreds of millions of customers. You'll have the opportunity to work with state-of-the-art technologies while contributing to the future of retail technology at one of the world's largest companies.

Last updated 7 hours ago

Responsibilities For Senior, Software Engineer - Gen AI

  • Design and implement Generative AI solutions utilizing RAG for summarization and knowledge retrieval
  • Build and integrate LLM-based agents using vector search technologies
  • Fine-tune and optimize models for specific business needs
  • Contribute to building scalable, distributed systems that integrate machine learning pipelines
  • Translate high-level designs into detailed solutions
  • Design and implement MVPs to validate AI-driven features
  • Drive technical direction for large-scale AI initiatives
  • Mentor junior engineers
  • Establish best practices for AI-driven systems

Requirements For Senior, Software Engineer - Gen AI

Python
  • 6+ years of experience in machine learning engineering, with 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
  • Bachelor's degree in computer science or related field (or equivalent experience)

Benefits For Senior, Software Engineer - Gen AI

Medical Insurance
Dental Insurance
Vision Insurance
401k
Equity
Parental Leave
  • Multiple health plan options including vision & dental
  • 401(k) and stock purchase plans
  • Life insurance
  • Education assistance for Associate and dependents
  • Parental Leave
  • Military service pay
  • Paid Time off (vacation, sick, parental)
  • Short-term and long-term disability
  • Associate discounts
  • Performance-based bonus awards

Interested in this job?

Jobs Related To Walmart Senior, Software Engineer - Gen AI

Senior, Software Engineer (Search MLOps)

Senior Software Engineer position at Walmart Global Tech, focusing on Search MLOps and machine learning infrastructure, offering competitive salary and comprehensive benefits in Silicon Valley.

Data Scientist III - Machine Learning Engineer - Data Ventures

Senior Machine Learning Engineer role at Walmart focused on building and deploying scalable ML solutions for data products and insights.

Software Engineer III, Machine Learning Engineer

Senior Machine Learning Engineer role at Walmart focusing on building state-of-the-art bidding optimization and user prediction systems for display ads platform.

Senior, Data Scientist - GenAI Platform

Senior Data Scientist position at Walmart focusing on GenAI Platform development, building LLM-powered solutions to enhance employee experience and productivity.

Senior, Data Scientist – Conversational AI

Senior Data Scientist position at Walmart focusing on Conversational AI and Natural Language Understanding, building intelligent shopping assistants and customer interaction systems.