Apply on Employer Site

Walmart Canada · 4 months ago

(USA) Staff, Software Engineer - AI-RAG Expert

United States

Full-time

Remote

Senior Level, Lead/Staff

$143K/yr - $286K/yr

4+ years exp

Walmart Canada is at the forefront of AI innovation, leveraging cutting-edge technologies to redefine customer experiences. They are seeking a Staff, Software Engineer with expertise in Generative AI and RAG frameworks to lead the design and development of advanced AI systems, while mentoring engineering teams and architecting scalable solutions.

DeliveryRetailShopping

No H1B

Responsibilities

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

Assist in Architecting 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

Partner with architects to design AI for platform consumption

Partner with teams to integrate AI into platforms

Work cross-functionally to design solutions that address both functional and non-functional requirements

Assist in establishing best practices for building and maintaining AI-driven systems

Act as a mentor, fostering technical growth within the team and contributing to a culture of continuous learning and innovation

Qualification

Generative AIRAG frameworksLLMsPythonML frameworksDistributed computingCloud platformsScalable architecturesSystem designMentorshipCollaboration

Required

Overall 4+ 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

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

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

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

Background in creating inclusive digital experiences, demonstrating knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly

Knowledge of accessibility best practices and join us as we continue to create accessible products and services following Walmart's accessibility standards and guidelines for supporting an inclusive culture

Benefits

Health benefits include medical, vision and dental coverage.

Financial benefits include 401(k), stock purchase and company-paid life insurance.

Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.

Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.

Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.