Technical leader architecting and deploying advanced Gen AI and ML solutions for Walmart Connect. Driving innovation in ad targeting, personalization, and campaign analytics.
Responsibilities
Shape the long-term AI engineering roadmap for Walmart Ads, aligning with the org vision of becoming the definitive AI-native omnichannel RMN.
Lead the strategy for pre-training, fine-tuning, and aligning (RLHF/DPO) open-weight foundation models explicitly for the retail media domain.
Architect highly scalable Retrieval-Augmented Generation pipelines that ground LLM outputs in real-time inventory, pricing, and hyper-local store data.
Architect and deploy autonomous AI agents capable of complex reasoning, tool-use, and multi-step planning.
Collaborate cross-functionally with Ads Product, Data Science, and Engineering to integrate LLMs & agentic AI into core ad surfaces (Search, Display, Video, In-Store) and omnichannel measurement.
Drive inference optimization (quantization, vLLM, TensorRT) to ensure generative models are cost-efficient and meet the strict ultra-low latency requirements (sub-100ms) of programmatic ad exchanges and real-time bidding (RTB) environments.
Ensure all AI capabilities are cleanly decoupled yet tightly integrated with core ad servers, SSPs, DSPs, and yield management engines handling millions of queries per second (QPS).
Requirements
PhD or Master’s degree in Computer Science, Machine Learning, Statistics, or related field.
12+ years of ML engineering experience, with at least 5+ years operating at a Principal, Staff, or Distinguished level in a high-scale tech environment.
Deep hands-on expertise in training, fine-tuning, and deploying Large Language Models and Foundation Models into production.
Proven track record of designing complex systems using LLM orchestration frameworks (e.g., LangChain, LlamaIndex), reasoning frameworks (Chain-of-Thought, ReAct), and multi-agent architectures.
Strong background in designing highly available, low-latency distributed systems (handling 100K+ QPS) using cloud-native architectures (GCP, Azure, or AWS).
Expert-level proficiency in Python, Go, Rust, or C++, alongside modern ML frameworks (PyTorch, JAX, TensorFlow).
Experience with scalable vector databases (Milvus, Pinecone), feature stores, and real-time streaming pipelines (Kafka, Flink).
Strong foundational knowledge of programmatic advertising, Real-Time Bidding (RTB), click-through rate (CTR) prediction, and dynamic creative optimization.
Patents or publications in top-tier AI conferences (NeurIPS, ICML) are a strong plus.
Benefits
Health benefits include medical, vision and dental coverage.
Financial benefits include 401(k), stock purchase and company-paid life insurance.
Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.
Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.
Senior Scientist II leading innovative AI and machine learning projects in oncology at Tempus. Collaborating with teams to advance predictive modeling and drug R&D initiatives.
MLOps Engineer scaling AI/ML solutions across game studios at Stillfront. Collaborating with teams to operationalize machine learning solutions for a diverse gaming portfolio.
Machine Learning Engineer supporting the Enterprise Machine Learning team in developing advanced solutions. Collaborating with stakeholders and driving data science initiatives in a hybrid work environment.
Contract AI/ML Engineer focused on delivering machine learning projects for clients at AND Digital. Aiming to enhance AI - powered solutions and contribute to digital skills development.
Senior Machine Learning Engineer at TomTom building data pipelines for autonomous vehicle mapping solutions. Collaborating in a diverse team to innovate and implement machine learning technologies.
Machine Learning Manager improving automated decision - making and managing a team. Driving innovations in credit risk modeling for Monzo’s borrowing products.
Machine Learning Engineer responsible for ML model design and deployment at SiGMA Group. Enhancing event experiences through AI - driven solutions in iGaming and tech.
Machine Learning Intern at Nomagic tackling physical manipulation challenges in AI robotics with top professionals. Engage in innovative projects while shaping robotic technology.
Machine Learning Engineer developing and deploying AI systems for Personio's HR platform. Collaborating with teams to integrate ML features and products ensuring data privacy and security.
Principal Machine Learning Engineer leading AI and Machine Learning systems at Bumble for recommendations and personalization. Driving improvements in user engagement and safety across Bumble products.