Senior, Data Scientist - Agentic AI

Walmart • San Bruno, CA

159d • $117,000 - $234,000

About The Position

We're looking for a highly skilled and motivated Senior Data Scientist to lead the development of our agentic AI platforms. This role is crucial for our E-commerce Analytics team, where you'll be instrumental in shaping the future of our AI/ML infrastructure. You will leverage your expertise to build and extend agentic frameworks and create robust, scalable CI/CD pipelines to support production-grade agentic workloads. Your work will involve close collaboration with data analytics, machine learning, and product teams to integrate and scale innovative AI/ML solutions across our E-commerce ecosystem.

Requirements

• Hands-on experience deploying, managing, and operating AI/ML workloads in a production environment.
• Hands-on experience developing, deploying, and maintaining AI agents in production environments.
• Deep understanding of ML frameworks and commercial AI/ML infrastructure, such as PyTorch, TensorFlow, KubeFlow/MLFlow, and HuggingFace, and agentic frameworks like Langchain, ADK, AutoGen, etc.
• Proven experience with containerization and orchestration technologies, such as Docker and Kubernetes.
• Experience with distributed training techniques like data parallelism, tensor parallelism, and pipeline parallelism.
• Strong technical acumen with the ability to act as a credible technical advisor, setting and enforcing high-quality standards for code and system design.
• Demonstrate ability to develop, drive, and execute a technical vision and roadmap that aligns with business objectives.
• Experience building high-traffic distributed systems and ML infrastructure, ensuring scalability and resilience.
• Exceptional communication and collaboration skills, with the ability to articulate complex technical concepts clearly to audiences.

Nice To Haves

• Experience with the design and development of Generative AI models and technologies is preferred.
• Data science, machine learning, optimization models, Master's degree in Machine Learning, Computer Science, Information Technology, Operations Research, Statistics, Applied Mathematics, Econometrics.
• Successful completion of one or more assessments in Python, Spark, Scala, or R.
• Using open source frameworks (for example, scikit learn, tensorflow, torch).
• Knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly.

Responsibilities

• Architect, design, and develop the next generation of our AI/ML infrastructure, focusing on capabilities that support agentic frameworks and production workloads.
• Develop and execute a comprehensive technical roadmap, defining success metrics and championing the platform's adoption across the organization.
• Transition AI/ML models from proof-of-concept to production-ready pipelines using MLOps frameworks, ensuring solutions are scalable, reliable, and maintainable.
• Partner with data analytics and data science teams to understand their requirements, iterate on solutions, and provide expert technical guidance.
• Stay on the cutting edge of Generative AI and other emerging technologies, contributing to continuous platform improvement and fostering a culture of technical excellence.
• Champion best practices for containerization and orchestration technologies to optimize data processing and inference for large-scale models.
• Effectively communicate project progress, technical challenges, and solutions to a wide range of audiences, from technical teams to senior leadership.

Benefits

• Competitive pay as well as performance-based bonus awards.
• Health benefits including medical, vision and dental coverage.
• Financial benefits including 401(k), stock purchase and company-paid life insurance.
• Paid time off benefits including PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
• Short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement.
• Live Better U education benefit program for full-time and part-time associates.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Industry: General Merchandise Retailers
• Education Level: Bachelor's degree