Walmart Global Tech is seeking a Staff, Software Engineer with deep expertise in Generative AI, LLMs, Agentic and RAG frameworks to lead the design, development, and deployment of advanced AI systems to redefine customer experiences and build highly scalable and reliable solutions for the next generation AI system.
Requirements
- Strong experience with Agentic Frameworks, 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
- Deep knowledge of open-source projects or publications in Generative AI, RAG, or agentic frameworks.
- Strong backend Programming experience in Python / Golang or Java / Type script
- Experience in Gen AI – Agentic Frameworks, RAG Frameworks
- Cloud architecture and deployments: Kubernetes, Docker, micro-services architectures (REST)
Responsibilities
- Design, Develop , 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.
- Develop scalable, distributed AI systems with a focus on performance, fault tolerance, and disaster recovery.
- Develop detailed design specifications, including modular, extensible, and functional component
- Develop MVP's into Shippable product to validate requirements and refine designs iteratively.
- Provide deeper Insights on Containerization/Docker (and) experience in CI/CD and deployment pipelines
Other
- Clear Communication, including the ability to build design gaps
- Ability to multitask between several different requirements and features concurrently
- Ability to perform in a team environment and have a true passion about Technology. It’s a must!
- Work cross-functionally to design solutions that address both functional and non-functional requirements.
- Drive technical direction for large-scale AI initiatives and mentor junior engineers.
