Staff, Software Engineer– GenAI Initiative

Position Summary...

As part of Walmart's Global Sourcing Backend team, you will be building highly scalable and reliable APIs, services, and applications that will drive the next generation purchase order management system. You will independently handle high-impact, critical software/system monitoring issues, troubleshoot business and production issues. Working within this fast-moving, entrepreneurial team, you'll contribute to innovation and development using best-in-class methodologies, impacting perception and business transformation.

What you'll do...

Design and develop scalable services, solve complex software problems using advanced technology, and gain exposure to emerging tech in Advertising and E-commerce. Provide technical guidance, lead architecture decisions, and demonstrate your expertise as an experienced developer. Drive high-visibility projects, participate in complex cross-functional projects, translate requirements into technical solutions, design architectures, develop code, conduct testing, communicate progress, collaborate across teams, troubleshoot issues, improve designs, and ensure timely delivery.

Key Responsibilities: AI Model Development & Agentic Frameworks

• Design, fine-tune, and deploy Generative AI models focusing on RAG frameworks for enhanced information retrieval and generation.
• Develop and optimize LLM-based agents for knowledge grounding, multi-step reasoning, and decision-making.
• Integrate external data sources with LLMs to build dynamic, context-aware systems.

• Architect scalable, distributed AI systems prioritizing performance, fault tolerance, and disaster recovery.
• Create detailed design specifications, develop MVPs for validation, and refine designs iteratively.

• Partner with product teams for AI integration into customer-facing applications.
• Collaborate cross-functionally to meet functional and non-functional requirements.

• Drive technical direction for large-scale AI initiatives and mentor junior engineers.
• Establish best practices for AI system development and maintenance.

What you'll bring

• 9+ years in machine learning/software engineering, with 2+ years in Generative AI or LLMs.
• Strong experience with RAG frameworks, vector search, knowledge grounding, Python, ML frameworks (PyTorch, TensorFlow), Hugging Face Transformers.
• Proficiency in distributed computing, cloud platforms (GCP, Azure), scalable architectures.
• Familiarity with disaster recovery principles, resilient systems, multi-modal models, and advanced optimization techniques is preferred.
• Experience with Generative AI models like GPT, Gemini, and contributions to open-source or publications are a plus.
• Deep understanding of system design, non-functional requirements, and design patterns preferred.

About Walmart Global Tech

Join a team where one line of code can impact hundreds of millions. We are software engineers, data scientists, cybersecurity experts, and service professionals at the forefront of retail innovation. We value people-led, tech-empowered environments, offering roles for all career stages, with opportunities to learn, grow, and impact the future of retail.

Benefits

Competitive pay, performance bonuses, 401(k) match, stock purchase plan, paid parental leave, PTO, health plans, and more. We support a diverse and inclusive culture, valuing all individuals and their unique contributions.

Minimum Qualifications

Outlined below are the minimum qualifications for this position. If none are listed, there are no minimum qualifications.

• Bachelor’s degree in computer science, engineering, or related field + 4 years’ experience, or 6 years’ experience in software engineering or related area.

Preferred Qualifications

Master’s degree in relevant fields + 2 years' experience in software engineering or related areas. Knowledge in accessibility standards (WCAG 2.2 AA), assistive technologies, and inclusive digital experience is valued. Experience creating accessible products and services, following Walmart’s accessibility standards, is preferred.

Primary Location

840 W California Ave, Sunnyvale, CA , United States of America