Senior, Data Scientist (MLE) | Last Mile Delivery | Sunnyvale

Walmart • Sunnyvale, CA

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

About The Position

Walmart Global Tech Applied AI Last Mile team is looking for a passionate Senior Data Scientist to join world's largest retailer to shape the future of retail by working on various Last-Mile related problems. You will have the chance to leverage advanced machine learning and optimization methods to tackle a wide range of real-world problems, such as routing, trip pricing, driver search, driver matching, supply/demand forecast, ETA prediction, incentive strategy, and capacity optimization. This role offers an opportunity to work on large-scale, real-time data pipelines and deploy ML solutions that directly impact millions of customers and drivers. The Spark Driver Platform operates across all 50 U.S. states, empowering millions of drivers to provide local delivery and making it Walmart's largest local delivery service provider. The Spark Driver Applied AI team is dedicated to enhancing efficiency and profitability by pushing the boundaries of machine learning, data science, and economics. We build highly reliable and scalable production systems that drive the growth of the Spark Driver Platform. Our key focus areas include Capacity Planning, Driver Search, Driver Matching, Driver Offer Pricing, Driver Surge Pricing, Incentive Strategy, ETA Prediction, and GenAI Automation for Driver Support.

Requirements

• Ph.D. or master in Computer Sciences, Statistics, Operations Research, Economics, or other quantitative fields.
• Strong knowledge of machine learning, deep learning, reinforcement learning, and optimization.
• Proven track record of deploying ML models in production, preferably in high-scale environments.
• Strong proficiency in programming languages, such as Python, Java, TensorFlow, PyTorch, or other ML frameworks.
• Hands-on experience with cloud platforms such as GCP, or Azure.
• Knowledge of big data technologies like Spark, Kafka, and SQL.
• Proficiency in MLOps practices, including CI/CD for ML pipelines and model monitoring.
• Ability to drive projects from conception to production in a fact-paced, dynamic environment.
• Capability of conducting independent research to keep track of latest development and trend in the industry and academia.
• Excellent communication and leadership skills, with the ability to influence and inspire cross-functional teams.

Nice To Haves

• 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 of accessibility best practices and experience in creating inclusive digital experiences.

Responsibilities

• Build production models on large-scale datasets to help Walmart Last mile business use cases, by utilizing statistical modeling, machine learning and optimization techniques.
• Build data pipeline from multiple sources and extract data insight with data exploration and visualization techniques.
• Design, implement, deploy, and scale data science products with consideration of optimality and computation complexity.
• Collaborate with stakeholders, including product, engineering, operations team to ensure that our data science product supports Walmart's business objectives.
• Explore and latest trends of machine learning and optimization techniques to come up with solutions for complex data challenges.

Benefits

• Incentive awards for performance.
• 401(k) match.
• Stock purchase plan.
• Paid maternity and parental leave.
• PTO, multiple health plans.
• Medical, vision and dental coverage.
• Short-term and long-term disability.
• Company discounts.
• Military Leave Pay.
• Adoption and surrogacy expense reimbursement.
• Live Better U education benefit program.

What This Job Offers

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