(USA) Senior, Data Scientist

Walmart

Sunnyvale, CA

Full Time

Senior

117k-234k

9 days ago

Job Description

About the Role

The International Search team at Walmart powers query understanding, relevance ranking, and search result optimization across multiple touchpoints within the Walmart International market's app and website. They work with state-of-the-art machine learning (ML) and deep learning (DL) models to better understand customer queries and retrieve the most relevant products, utilizing the latest advancements in artificial intelligence (AI) to provide a more intuitive and effective search experience for customers.

Key Responsibilities

Stay up-to-date with the latest machine learning and deep learning algorithms (Transformers, Multi-modal models) that power Search and Query Understanding at Walmart.
Utilize pre-trained embedding models such as T5, CLIP, BERT, and RoBERTa for query rewriting, intent classification, and semantic search.
Work with big data technologies like Hive, Spark, GCP, VertexAI, and MapReduce to build scalable algorithms.
Use tools like Java, Python, Bash, Scikit-Learn, PyTorch, XGBoost, and NLTK to develop and deploy ML models.
Visualize insights using Matplotlib, Seaborn, Plotly, and GGplot.
Apply rigorous model testing and validation techniques (e.g., F1, ROC, RMSE).
Collaborate in agile scrum teams and follow best practices in software development (e.g., code reviews, CI/CD).
Innovate new ML/AI techniques when existing methods fall short.
Leverage LLMs and AI-assisted coding tools like Copilot to rapidly prototype and iterate.

Requirements

MS/PhD in Computer Science, Machine Learning, Information Retrieval, or related fields with 2-4 years of experience.
Strong background in NLP, Search Science, Query Understanding, and ML/DL.
Proficiency in Python, SQL, PySpark, and cloud-based big data systems.
Experience leading ML projects and mentoring junior data scientists.
Familiarity with embedding models like T5, CLIP, BERT, and RoBERTa.
Strong data visualization and model evaluation skills.

Nice to Have

Publications or patents in relevant fields.
Experience with open source frameworks (e.g., scikit-learn, TensorFlow, Torch).
Background in creating inclusive digital experiences and knowledge of 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' 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' experience in an analytics related field.
Option 3: 5 years' experience in an analytics or related field.

Benefits & Perks

Competitive pay with performance-based bonus awards.
Health benefits including medical, vision, and dental coverage.
Financial benefits such as 401(k), stock purchase, and company-paid life insurance.
Paid time off including PTO, sick leave, parental leave, family care leave, bereavement, jury duty, and voting.
Other benefits like short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement.
Live Better U education benefit program covering tuition, books, and fees.

Working at Walmart

Walmart values creating accessible products and services, fostering an inclusive culture, and continuously innovating with state-of-the-art AI and ML techniques to improve customer experience and operational excellence.

Apply Now