Staff, Data Scientist

Walmart · Retail · Sunnyvale, CA

Staff Applied Scientist role at Walmart Last Mile Delivery focused on developing and deploying machine learning and GenAI solutions for delivery operations. The role involves designing, training, evaluating, and monitoring production models, with a strong emphasis on building scalable, efficient, and impactful end-to-end systems. Experience with Agentic AI and deploying ML models in production is required.

What you'd actually do

  1. Work with a team of applied scientists in developing innovative AI-driven solutions and provide technical leadership as domain expert
  2. Research and implement state-of-the-art machine learning techniques.
  3. Collaborate cross-functionally with business, product, and engineering stakeholders to ensure AI solutions align with Walmart’s strategic objectives.
  4. Design and optimize machine learning pipelines to ensure scalability, efficiency, and real-time processing capabilities.
  5. Build and deploy end-to-end models, balancing accuracy, computational efficiency, and measurable business impact with deep expertise in best practice of MLOps

Skills

Required

  • Ph.D. or equivalent advanced degree in Computer Science, Statistics, Operations Research, Economics, or a related field. Or Master degree with at least 5 years of relevant working experience
  • Extensive experience in machine learning, GenAI, and Agentic AI.
  • 5+ Years of experience of deploying ML models in production, preferably in high-scale environments
  • Strong programming skills in Python, Java, C, or C++
  • Experience with large-scale data systems (Spark/Hive) and cloud computing (GCP/Azure)
  • Demonstrated ability to drive projects from conception to production in a fast-paced, dynamic environment
  • Excellent communication and leadership skills, with the ability to influence and inspire cross-functional teams

What the JD emphasized

  • extensive experience in machine learning, GenAI, and Agentic AI
  • 5+ Years of experience of deploying ML models in production, preferably in high-scale environments

Other signals

  • Develop machine learning and GenAI solutions for last mile delivery.
  • Design, train, evaluate, deploy, and monitor production models.
  • Build and deploy end-to-end models balancing accuracy, efficiency, and business impact.
  • Hands-on experience in building GenAI applications.
  • Extensive experience in machine learning, GenAI, and Agentic AI.
  • Deploying ML models in production, preferably in high-scale environments.