Staff, Software Engineer

Walmart · Retail · Sunnyvale, CA

Staff Software Engineer role focused on designing, developing, and deploying advanced AI systems using Generative AI, LLMs, Agentic, and RAG frameworks. The role involves architecting scalable solutions, integrating agentic frameworks, mentoring teams, and building highly scalable and reliable solutions for Walmart's Global Technology Platform team.

What you'd actually do

  1. Design, Develop , fine-tune, and deploy Generative AI models with a focus on RAG frameworks to enhance information retrieval and generation.
  2. Develop and optimize LLM-based agents for tasks like knowledge grounding, multi-step reasoning, and decision-making.
  3. Integrate external data sources (e.g., vector databases) with LLMs to build dynamic, context-aware systems.
  4. Develop scalable, distributed AI systems with a focus on performance, fault tolerance, and disaster recovery.
  5. Drive technical direction for large-scale AI initiatives and mentor junior engineers.

Skills

Required

  • Python
  • ML frameworks (PyTorch, TensorFlow)
  • Hugging Face Transformers
  • distributed computing
  • cloud platforms (GCP, Azure)
  • scalable architectures
  • Python / Golang or Java / Type script

Nice to have

  • Gen AI – Agentic Frameworks
  • RAG Frameworks
  • Cloud architecture and deployments: Kubernetes
  • Docker
  • micro-services architectures (REST)

What the JD emphasized

  • Generative AI
  • LLMs
  • Agentic
  • RAG frameworks
  • Generative AI models
  • LLM-based agents
  • Agentic Frameworks
  • RAG frameworks
  • Generative AI
  • RAG
  • agentic frameworks

Other signals

  • Generative AI
  • LLMs
  • Agentic frameworks
  • RAG frameworks