Walmart Global Tech is seeking a Staff, Software Engineer to lead the design, development, and deployment of advanced AI systems, leveraging cutting-edge technologies to redefine customer experiences.
Requirements
- 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.
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.
Other
- 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.
- 6 years’ experience in software engineering or related area.
- 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.
- Knowledge of accessibility best practices and experience creating inclusive digital experiences.
- Ability to work in Sunnyvale, CA
