Senior Data Scientist, Computer Vision

About the Role

Key Responsibilities

• Design, train, and deploy deep learning models for 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 workflows that integrate 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

• A Master’s or 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.

Nice to Have

• 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 assessments in Python, Spark, Scala, or R.
• Experience using open source frameworks (e.g., scikit-learn, TensorFlow, Torch).
• Background in creating inclusive digital experiences and implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and digital accessibility best practices.

Qualifications

• Option 1: Bachelor’s degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, or related field and 3 years of experience in an analytics related field.
• Option 2: Master’s degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, or related field and 1 year of experience in an analytics related field.
• Option 3: 5 years of experience in an analytics or related field.

Benefits & Perks

• Incentive awards for performance.
• 401(k) match, stock purchase plan.
• Paid maternity and parental leave.
• Paid time off including PTO, sick leave, holidays, and other purposes.
• Multiple health plans.
• Health benefits including medical, vision, and dental coverage.
• Financial benefits such as 401(k), stock purchase, and company-paid life insurance.
• Walmart-paid education benefit program Live Better U.
• Additional perks like short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement.