Senior, Software Engineer - Gen AI (Evergreen)

Sunnyvale, CA, United States

Walmart

What’s a career at Walmart or Sam’s Club like? To find out, explore our culture, our opportunities and the difference you can make.

View all jobs at Walmart

Apply now Apply later

Company Description

About Walmart:

Fifty years ago, Sam Walton started a single mom-and-pop shop and transformed it into the world's biggest retailer. Since those founding days, one thing has remained consistent: our commitment to helping our customers save money so they can live better. Today, we're reinventing the shopping experience and our associates are at the heart of it. You'll play a crucial role in shaping the future of retail, improving millions of lives around the world. This is that place where your passions meet purpose. Join our family and create a career you're proud of.

Job Description

Job Description

Are you seeking an opportunity to unleash your expertise in cutting-edge AI technologies? Envision yourself as a Senior Machine Learning Engineer, where you'll orchestrate the design and deployment of AI systems powered by Large Language Models (LLMs), Retrieval-Augmented Generation (RAG) frameworks, and advanced agentic techniques. In this pivotal role, you'll embark on a journey to construct scalable pipelines, seamlessly integrate external data sources, and collaborate with cross-functional teams to deliver high-impact, AI-driven products that redefine the boundaries of innovation.


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

Minimum Qualifications:

  • 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.

Qualifications

Azure, GCP, Gemini, Gen AI, Python

Additional Information

All your information will be kept confidential according to EEO guidelines.

Apply now Apply later

* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰

Job stats:  0  0  0

Tags: Architecture Azure Distributed Systems Engineering GCP Gemini Generative AI GPT LLMs Machine Learning MVP NLP Open Source Pipelines Python PyTorch RAG Research TensorFlow Transformers

Perks/benefits: Career development

Region: North America
Country: United States

More jobs like this