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.

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.

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.

Principal Data Scientist – Gen AI

Key skills