Staff Data Scientist: Vision & Generative AI Lead

Walmart Global Tech; Bentonville (AR); On-site; USD 150,000 - 230,000; Full time

Benefits offered by this job

Stock program

Health benefits

401(k) match

Paid maternity and parental leave

PTO

Job summary

Walmart Global Tech in Bentonville, AR is seeking a Staff Data Scientist to advance computer vision and generative AI initiatives. You will design, train, and deploy models, drive production-grade ML pipelines, and guide junior team members in a fast-paced retail technical environment.

Collaborating with cross-functional partners, you will translate business challenges into data-driven solutions, optimize models for latency and scale, and contribute to Walmart’s data strategy and innovative

Qualifications

• MS or PhD in Computer Science, Machine Learning, Computer Vision, Applied Mathematics, or a related field.
• 5-8 years of industry or research experience.
• Deep learning expertise in computer vision (CNNs, ViTs, diffusion models).
• Experience deploying image generation or understanding systems in production (visual search, personalized image recommendations).

Responsibilities

• Design, train, and deploy computer vision models for object detection and image-based personalization.
• Build and scale foundation model pipelines that leverage multimodal signals (text, images, embeddings).
• Mentor team members and lead innovation in image generative modeling (diffusion, CNNs, text-to-image).
• Design ML workflows with MLOps for production deployment and monitoring.
• Collaborate with cross-functional teams to translate business requirements into data-driven solutions.

Skills

Deep learning for CV

Python programming

Mentoring / leadership

Multimodal models

Model deployment / MLOps

Cross-functional collaboration

Research experience

Data sourcing / validation

Education

MS or PhD in CS/ML/CV

Tools

PyTorch

TensorFlow

OpenCV

Torchvision

Hugging Face

GPU deepstream

Job description

Walmart Global Tech in Bentonville, AR is seeking a Staff Data Scientist to advance computer vision and generative AI initiatives. You will design, train, and deploy models, drive production-grade ML pipelines, and guide junior team members in a fast-paced retail technical environment.

Collaborating with cross-functional partners, you will translate business challenges into data-driven solutions, optimize models for latency and scale, and contribute to Walmart’s data strategy and innovative