Staff, Software Engineer– GenAI Initiative

Walmart; Sunnyvale (CA); On-site; USD 120,000 - 180,000; Full time

Job summary

An established industry player is seeking a seasoned developer to join their Global Sourcing Backend team. This role focuses on building scalable APIs and applications that will transform purchase order management systems. You will tackle high-impact software monitoring issues and lead architecture decisions while collaborating across teams to drive innovation. Join a dynamic environment where your expertise in machine learning and Generative AI will directly influence the future of retail technology. This is a unique opportunity to work with cutting-edge technologies and make a significant impact in a fast-paced setting.

Benefits

Competitive Pay

Performance Bonuses

401(k) Match

Stock Purchase Plan

Paid Parental Leave

PTO

Health Plans

Qualifications

• 9+ years in machine learning/software engineering with 2+ years in Generative AI.
• Strong experience with RAG frameworks and Python ML frameworks.

Responsibilities

• Design and develop scalable AI systems and deploy Generative AI models.
• Drive technical direction for large-scale AI initiatives and mentor junior engineers.

Skills

Machine Learning

Generative AI

Python

Distributed Computing

Cloud Platforms (GCP, Azure)

RAG Frameworks

Knowledge Grounding

Multi-step Reasoning

System Design

Education

Bachelor’s degree in Computer Science

Master’s degree in relevant fields

Tools

PyTorch

TensorFlow

Hugging Face Transformers

Job description

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 94086-4828, United States of America