Walmart

Principal Data Scientist – Gen AI Vision

Walmart

full-time

Posted on:

Location Type: Office

Location: Sunnyvale • California • 🇺🇸 United States

Visit company website
AI Apply
Apply

Salary

💰 $143,000 - $286,000 per year

Job Level

Lead

Tech Stack

CloudNumpyOpen SourcePandasPythonPyTorch

About the role

  • Develop and apply state-of-the-art deep learning based generative models and multi-modal models such as text generation, image generation, video generation and capturing multimodal data relationships
  • Proficient in evaluating trade-offs between open source vs Cloud APIs for multimodal Gen AI in terms of performance, cost efficiencies
  • Work with large scale data and curate in house domain datasets from scratch
  • Collaborate with Engineering teams to define interfaces, understanding and resolving dependencies
  • Standardize and build scalable and highly available systems that follow operational best practices
  • Develop visuals and reports to efficiently communicate model results and insights to business teams
  • Maintain world-class academic credentials through patent applications, publications, conference presentations, and external collaboration and service to the research community

Requirements

  • Expertise in working with deep-learning based generative/multi-modal models or similar research experience with deep-learning/AI
  • Experience in multimodal Generative Models for Text generation, Image generation, Audio and Video generation
  • Experience in VLMs for Image classification, object detection, scene/semantic segmentation, keypoint detection, CLIP/similar for similarity
  • Experience in models for multimodal synthetic data generation
  • Experience with large deep-learning model fine-tuning and deployment
  • Experience with prompt engineering and agentic AI design
  • Solid understanding of Python and expertise in working with data science related packages (Numpy, Pandas, Pytorch etc.)
  • Advanced degree in computer science, statistics, data science, mathematics or other disciplines that expose scientific methods of research and analysis. PhD degree is a plus
  • Publications in top ML/AI/NLP conferences (NeurIPS, ICML, EMNLP, ACL, ICLR, AAAI, IJCAI, CVPR, ICCV, ECCV, KDD etc.) are a plus
Benefits
  • 401(k) match
  • Stock purchase plan
  • Paid maternity and parental leave
  • PTO
  • Multiple health plans
  • Incentive awards for performance
  • Health insurance
  • Vision coverage
  • Dental coverage
  • Short-term and long-term disability
  • Company-paid life insurance
  • Company discounts
  • Military Leave Pay
  • Adoption and surrogacy expense reimbursement
  • Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
  • Live Better U education benefit program for full-time and part-time associates

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
deep learninggenerative modelsmulti-modal modelstext generationimage generationvideo generationlarge deep-learning model fine-tuningprompt engineeringagentic AI designdata science
Soft skills
collaborationcommunicationproblem-solvinganalytical thinkingorganizational skills
Certifications
PhD in computer sciencePhD in statisticsPhD in data sciencePhD in mathematics