Staff, Data Scientist

About the role

At Walmart, we prioritize innovation and data security. Our team is dedicated to maintaining a secure operating environment and preserving the trust of our customers, associates, and stakeholders. We combine a range of services and expertise to prevent fraud, detect threats, and manage digital risk and access. Our focus is on mitigating attack risks, securing cloud transformation, and fostering a culture of security and reliability within our team

Responsibilities

• Design, train, and deploy deep learning models for object detection, image-based personalization, including product discovery, visual recommendations, and generative image content.
• Build and scale foundation model pipelines that leverage multimodal signals - including text, images, and customer embeddings - to power highly personalized visual experiences.
• Lead innovation efforts on image generative modeling, including diffusion models, CNNs, or text-to-image synthesis aligned with retail context.
• Implement efficient image understanding models for visual similarity, visual search, or classification across Walmart's vast product catalog.
• Design model workflow that integrates with MLOps to optimize model performance and latency, especially for customer-facing features with visual components.
• Contribute to cutting-edge research and internal knowledge by publishing work or presenting findings on vision-language models, multimodal personalization, and generative AI for e-commerce.

Requirements

• MS or (preferably) PhD in Computer Science, Machine Learning, Computer Vision, Applied Mathematics, or a related technical field, with 2-4 years of industry or research experience.
• Demonstrated expertise in deep learning for computer vision, including CNNs, vision transformers (ViTs), and diffusion-based generative models.
• Experience building and deploying image generation, manipulation, or understanding systems in production - such as visual search, personalized image recommendations.
• Proficiency in Python and strong familiarity with deep learning frameworks such as PyTorch, TensorFlow, and image modeling libraries like OpenCV, torchvision, or Hugging Face.
• Familiarity with multimodal models (e.g., CLIP, BLIP, Flamingo) that combine vision and language for richer customer understanding and content personalization.
• Experience working with large-scale datasets, data pipelines, and distributed training.
• About Walmart Global Tech
• Our hubs: Walmart is a global company with offices across the United States and around the world. Our global headquarters is in Bentonville, Arkansas, with primary hubs in the San Francisco Bay area and New York/New Jersey.

Benefits

Additional Information

Position Summary... What you'll do... We are looking for a computer vision scientist to solve some real retail problems/challenges. Utilize Computer Vision algorithms, Generative AI methods, deep learning architectures, VLMs and state-of-the-art techniques to build and deploy advanced retail solutions.

Your Match

How well this role fits your profile.

Company Intel

What employees say

Worked at Walmart? Share your experience