Walmart

Staff Data Scientist, Conversational AI

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

Python

About the role

  • Develop and customize state-of-the-art deep learning based generative models and multimodal models to support conversational customer shopping experience.
  • Utilize cutting-edge machine learning/statistical/optimization techniques to deliver solutions for projects in the domain of conversational AI.
  • Build, validate, test and deploy language models combined with different machine learning techniques to explain or predict behavior and solve a variety of business and engineering problems.
  • Mine massive data, derive business insights, and propose interesting modeling projects that have the potential to have high business impact.
  • 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.

Requirements

  • Rich experience with NLP and language models (e.g., Transformers, BERT, Llama, GPTs, Gemini), including but not limited to prompt engineering, fine-tuning and efficient deployment.
  • Experience with agentic system development, including but not limited to static/agentic flow design, agentic system evaluation.
  • Solid programming skills of Python and familiar with challenges working with large data.
  • Excellent oral and written communication skills.
  • PhD degree in computer science, statistics, data science, mathematics or other discipline that exposes scientific methods of research and analysis.
  • Strong academic and publication record is required (NeurIPS, ICML, EMNLP, ACL, ICLR, AAAI, KDD etc.).
Benefits
  • Health benefits include medical, vision and dental coverage.
  • 401(k) match.
  • Stock purchase plan.
  • Paid maternity and parental leave.
  • PTO, multiple health plans, and much more.

Applicant Tracking System Keywords

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

Hard skills
deep learninggenerative modelsmultimodal modelsmachine learningNLPlanguage modelsPythonprompt engineeringfine-tuningdata mining
Soft skills
communicationcollaborationproblem-solvinganalytical thinking
Certifications
PhD in computer sciencePhD in statisticsPhD in data sciencePhD in mathematics