Walmart

Principal Data Scientist – Gen AI

Walmart

full-time

Posted on:

Location Type: Office

Location: SunnyvaleCaliforniaUnited States

Visit company website

Explore more

AI Apply
Apply

Salary

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

Job Level

About the role

  • Process large scale Walmart datasets using distributed cloud computing technologies to create and maintain 3D assets for the Walmart catalog.
  • Use computer vision techniques to create advanced deep learning models for 2D/3D image generation and synthesis, image classification, scene segmentation, object detection, keypoint detection.
  • Build and maintain ML/DL model pipelines and services with a focus on latency, throughput, and scalability.
  • End to end ownership and collaboration with product, engineering, vendors, and annotation partners to drive production ML application integrations and maintenance.
  • Collaborate with other internal Walmart Data Science teams to enable high quality user experiences powered by AI/ML around computer vision.

Requirements

  • Master's degree in Computer Science with specialization in computer vision and/or machine learning or equivalent practical experience.
  • 5 years of experience with machine learning algorithms and tools and 3 year of experience in deep learning.
  • Experience in launching Machine Learning models in products and building production pipelines/services.
  • Experience with TensorFlow, Tensorflow 3D, Pytorch, Pytorch3D, JAX, numpy, C++, Python, Unix, Docker, CPU and GPU architectures.
  • Experience with publishing at conferences/journals in Machine learning/Computer Vision.
  • Master’s degree or PhD in Engineering, Computer Science, or a related technical field is preferred.
  • 3 years of experience in a technical leadership role for an applied research project with an industry application use-case is preferred.
Benefits
  • Health benefits include medical, vision and dental coverage.
  • Financial benefits include 401(k), stock purchase and company-paid life insurance.
  • Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
  • Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.
  • Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.
Applicant Tracking System Keywords

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

Hard Skills & Tools
machine learningdeep learningcomputer visionimage classificationscene segmentationobject detectionkeypoint detectionmodel pipelinesscalabilitylatency
Soft Skills
collaborationend to end ownershiptechnical leadership
Certifications
Master's degree in Computer SciencePhD in Engineering