Taro Logo

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
This job posting is no longer active. Check out these related jobs instead:

Job Description

Walmart Global Tech is seeking a Senior Machine Learning Engineer specializing in Generative AI and Large Language Models (LLMs) to join their innovative team. This role combines cutting-edge AI development with practical business applications, focusing on building scalable solutions that impact millions of customers.

The position offers an opportunity to work on advanced AI systems, implementing RAG frameworks and agentic techniques while collaborating with a world-class team. You'll be responsible for designing and deploying AI solutions that integrate with Walmart's extensive retail infrastructure, requiring both technical expertise and strategic thinking.

As a senior engineer, you'll not only build systems but also shape the technical direction of AI initiatives and mentor other team members. The role demands expertise in Python, ML frameworks, and distributed systems, with a particular focus on Generative AI and NLP.

Walmart offers a comprehensive benefits package including competitive salary, performance bonuses, stock options, and extensive healthcare coverage. The hybrid work environment provides flexibility while maintaining collaborative opportunities at their Silicon Valley campus.

This is an excellent opportunity for an experienced ML engineer looking to make a significant impact at scale, working with one of the world's largest retailers during their technological transformation. You'll be at the forefront of applying generative AI to real-world challenges while enjoying the benefits of a stable, innovative company with extensive resources and reach.

Last updated 3 months 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 plans
  • 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