Senior, Software Engineer - Gen AI (Evergreen)

Company Description

About Walmart:

Fifty years ago, Sam Walton started a single mom-and-pop shop and transformed it into the world's biggest retailer. Since those founding days, one thing has remained consistent: our commitment to helping our customers save money so they can live better. Today, we're reinventing the shopping experience and our associates are at the heart of it. You'll play a crucial role in shaping the future of retail, improving millions of lives around the world. This is that place where your passions meet purpose. Join our family and create a career you're proud of.

Job Description

Job Description

Are you seeking an opportunity to unleash your expertise in cutting-edge AI technologies? Envision yourself as a Senior Machine Learning Engineer, where you'll orchestrate the design and deployment of AI systems powered by Large Language Models (LLMs), Retrieval-Augmented Generation (RAG) frameworks, and advanced agentic techniques. In this pivotal role, you'll embark on a journey to construct scalable pipelines, seamlessly integrate external data sources, and collaborate with cross-functional teams to deliver high-impact, AI-driven products that redefine the boundaries of innovation.

Key Responsibilities:

AI Development & Agentic Frameworks:

• Design and implement Generative AI solutions utilizing RAG for summarization, and knowledge retrieval.
• Build and integrate LLM-based agents that utilize vector search technologies to provide dynamic, real-time responses.
• Fine-tune and optimize models for specific business needs and tasks.

System Design & Scalability:

• Contribute to building scalable, distributed systems that integrate machine learning pipelines.
• Translate high-level designs into detailed solutions, including mock-ups, pseudo-code, and modular components.
• Design and implement MVPs to validate and refine AI-driven features.

Collaboration & Integration:

• Support architectural decisions to ensure alignment with both functional and non-functional requirements.
• Participate in code reviews and ensure adherence to best practices.

Leadership & Mentorship:

• Drive technical direction for large-scale AI initiatives and mentor junior engineers.
• Establish best practices for building and maintaining AI-driven systems.

Qualifications

Minimum Qualifications:

• 6+ years of experience in machine learning engineering, with a 2+ years focus on Generative AI and NLP.
• Proficiency in Python, ML frameworks (PyTorch, TensorFlow), and NLP libraries like Hugging Face Transformers.
• Hands-on experience with RAG frameworks, vector search tools and knowledge grounding techniques.
• Familiarity with distributed systems, scalable architectures, and cloud platforms (GCP, Azure).
• Strong understanding of software design patterns, disaster recovery strategies, and non-functional requirements.

Preferred Qualifications:

• Experience with multi-modal models and advanced optimization techniques.
• Experience with Generative AI models (e.g., GPT, Gemini) and their fine-tuning.
• Contributions to open-source projects or AI research publications.
• Strong problem-solving and debugging skills in distributed environments.

Qualifications

Azure, GCP, Gemini, Gen AI, Python

Additional Information

All your information will be kept confidential according to EEO guidelines.