Principal AI Scientist – Lead Large-Scale ML

Walmart Inc.

Bentonville (AR)

On-site

USD 110,000 - 220,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Performance bonus

Stock options

Comprehensive benefits

Job summary

Walmart Inc. is seeking a Principal Data Scientist to lead AI/ML development and deployment across Walmart’s operations. You will design innovative algorithms, tackle large-scale data challenges, and collaborate with cross-functional teams to drive measurable business outcomes.

The role emphasizes scalable cloud solutions, state-of-the-art frameworks, and strong leadership to mentor others while delivering high-impact results at Walmart.

Qualifications

Bachelor’s or higher in a related field with strong AI/ML foundations.
10+ years in software development and data science focusing on AI/ML.
Proven experience with Python, C++, CUDA, TensorFlow and PyTorch.

Responsibilities

Lead end-to-end AI/ML projects from ideation to deployment.
Design scalable cloud-based ML solutions on GCP or similar.
Develop novel algorithms for large-scale data challenges.
Build parallel compute environments for CPU/GPU workloads.
Collaborate with engineering, business and operations teams.
Advocate CI/CD, testing and documentation practices.
Mentor junior data scientists and foster innovation.

Skills

Python

C++

CUDA

TensorFlow

PyTorch

NumPy

Pandas

GCP

Terraform

Pyspark

Scala

NLP

Computer Vision

Education

Bachelor’s degree in Computer Science, Data Science, or related field

Master’s or Ph.D. preferred

Tools

Terraform

Cloud platforms (GCP)

APIs

Job description

The role emphasizes scalable cloud solutions, state-of-the-art frameworks, and strong leadership to mentor others while delivering high-impact results at Walmart.

Get your free, confidential resume review.

or drag and drop your file here.