
Principal Data Scientist – Gen AI
Walmart
full-time
Posted on:
Location Type: Office
Location: Sunnyvale • California • United States
Visit company websiteExplore more
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