Principal, Data Scientist - Gen AI

Walmart•Sunnyvale, CA

1d•$143,000 - $286,000•Onsite

About The Position

Position Summary... What you'll do... Principal Data Scientist (Computer Vision) Join Walmart and your work could help over 275 million global customers live better every week. Yes, we are the Fortune #1 company. But you’ll quickly find we’re a company who wants you to feel comfortable bringing your whole self to work. A career at Walmart is where the world’s most complex challenges meet a kinder way of life. Our mission spreads far beyond the walls of our stores. From day one, you’ll be empowered and equipped to do the best work of your life. careers.walmart.com About Walmart Global Tech Org Imagine working in an environment where one line of code can make life easier for hundreds of millions of people and put a smile on their face. That is what we do at Walmart Global Tech. We are a team of 15,000+ software engineers, data scientists and service professionals within Walmart, the world’s largest retailer, delivering innovations that improve how our customers shop and empower our 2.2 million associates. To others, innovation looks like an app, service, or some code, but Walmart has always been about people. People are why we innovate, and people power our innovations. Being human led is our true disruption. About Emerging Tech Team The Emerging Tech team is passionate about solving customer and associate problems with the newest technologies. The team is responsible for creating breakthrough capabilities, delivering frictionless experiences, and making these technologies easily available to thousands of Walmart developers and 2.2 million associates. The applications and services built on these capabilities are used by hundreds of millions of customers daily. We are building new platforms to bring physical and digital world together. What you will be part of At Walmart’s Emerging Tech Extended Reality team, we own some of the most challenging, fascinating, and impactful work in the fields of Computer vision, Machine learning and Deep Learning for next generation Augmented Reality and Virtual Reality experiences. We are looking for Applied Scientists/researchers/Computer vision engineers with algorithms, programming and/or systems background.

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.

Nice To Haves

Master’s degree or PhD in Engineering, Computer Science, or a related technical field.
3 year of experience in a technical leadership role for an applied research project with an industry application use-case.
Experience with 3D Real-time rendering of scenes, objects using NeRF and other more recent neural approaches.
Experience writing architectures with CNNs, vision transformers and pre-training from scratch/fine-tuning.
Experience with deep learning for synthetic data generation using GANs, diffusion models, multimodal architectures like CLIP.
Experience with Few-shot learning, transfer learning, unsupervised and semi-supervised methods, contrastive learning, active learning, weakly supervised auto labeling, semantic search index.
Experience with GCS, BigQuery, Nvidia Toolkits or equivalent.
Demonstrated knowledge of the following programming Languages: Java, Python, Bash, Spark (SQL, Streaming), Hadoop Map-Reduce experience.
Build Manager (Maven), Distributed Version Control (GIT), Continuous Integration (Jenkins) experience.
Bonus Experience in Pytorch 3D, Nvidia Kaolin is a plus.
Experience in building Augmented Reality applications using Unity or other AR SDK’s
Experience with Mixed Reality/Virtual Reality(Holo Lens, Gear, Occulus, Vive, Rift.)
Experience in using 3D Modelling tools such as Maya, Blender and other photogrammetry tooling.
Experience in Game Design, 3D modelling, and 3D Computer Graphics is a plus
Data science, machine learning, optimization models, PhD in Machine Learning, Computer Science, Information Technology, Operations Research, Statistics, Applied Mathematics, Econometrics, Publications or active peer reviewer in related journals or conference, Successful completion of one or more assessments in Python, Spark, Scala, or R, Using open source frameworks (for example, scikit learn, tensorflow, torch), We value candidates with a background in creating inclusive digital experiences, demonstrating knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly. The ideal candidate would have knowledge of accessibility best practices and join us as we continue to create accessible products and services following Walmart’s accessibility standards and guidelines for supporting an inclusive culture.

Responsibilities

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.
Read and implement research papers, strive for improvements in all aspects of the Computer Vision ML pipeline from better data generation techniques, more performant algorithms to highly efficient model deployments.

Benefits

At Walmart, we offer competitive pay as well as performance-based bonus awards and other great benefits for a happier mind, body, and wallet.
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.
You will also receive PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes.
The amount you receive depends on your job classification and length of employment.
It will meet or exceed the requirements of paid sick leave laws, where applicable.
For information about PTO, see https://one.walmart.com/notices.
Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.
Programs range from high school completion to bachelor's degrees, including English Language Learning and short-form certificates.
Tuition, books, and fees are completely paid for by Walmart.
Eligibility requirements apply to some benefits and may depend on your job classification and length of employment.
Benefits are subject to change and may be subject to a specific plan or program terms.
For information about benefits and eligibility, see One.Walmart.
The annual salary range for this position is $143,000.00 - $286,000.00 Additional compensation includes annual or quarterly performance bonuses.
Additional compensation for certain positions may also include : - Stock