Senior, Data Scientist | Gen AI Vision

Position Summary...

What you'll do...

Senior Data Scientist (Gen AI, 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. Join us and you'll discover why we are a world leader in diversity and inclusion, sustainability, and community involvement. 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 Next Generation Commerce

The Next Generation Commerce org’s vision is to define and deliver the future of commerce—agentic and deeply customer driven. We create AI-powered shopping experiences that adapt in real time to each customer’s shopping journey—whether they’re replenishing, exploring, or shopping with precision—across delivery, pickup, in-store, and beyond. Our mission is clear: serve the shopper by building systems that understand their needs. We obsess over speed, global readiness, and high engineering standards to ensure frictionless, scalable, and continuously improving experiences. As the team that drives breakthrough AI capabilities, we bridge the physical and digital worlds and make advanced technologies accessible to millions—delivering impact at scale for families, associates, and developers alike.

What you will be part of

At Walmart’s Extended reality and Emerging Content 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 experiences powered by multimodal Gen AI.

We are looking for Applied Scientists/researchers/Computer vision engineers with algorithms, programming and/or systems background.

Key responsibilities

· Process and create large scale Walmart datasets using distributed cloud computing technologies to create and maintain 3D, image and video generation assets for the Walmart catalog.

· Use computer vision techniques to create advanced Gen AI deep learning models for multimodal content generation and understanding – 2D/3D image, video generation and synthesis, 3D scene generation, image and video classification, scene segmentation, object detection, keypoint detection.

· Prompt engineering, pre-training, fine-tuning, custom evaluation benchmarks using Gen AI for image, video generation, 3D scene generation models.

· 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 performance algorithms to highly efficient model deployments.

Minimum qualifications:

· Master's degree in Computer Science with specialization in computer vision and/or machine learning or equivalent practical experience.

· 1 year of experience with machine learning algorithms and tools and 1 year of experience in deep learning. Experience in launching Machine Learning models in products and building production pipelines/services.

· Experience with Pytorch, Pytorch3D, JAX, numpy, C++, Python, Unix, Docker, CPU and GPU architectures, Triton/TensorRT frameworks.

· Experience with publishing at conferences/journals in Machine learning/Computer Vision

Preferred qualifications:

· Master’s degree or PhD in Engineering, Computer Science, or a related technical field.

· Experience with 3D Real-time rendering of scenes, objects.

· Experience writing architectures with CNNs, vision transformers and pre-training from scratch/fine-tuning.

· Using LLMs for scene captioning

· Experience with deep learning architectures such as GANs, diffusion models, rectified flow transformers, flow matching models, and LVM/VFM architectures.

· 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, Copilot/Cursor/other coding assistants.

Bonus

· Experience in Pytorch 3D, 3D Modelling tools such as Maya, Blender and other photogrammetry tooling.

· Experience in Game Design, 3D modelling, and 3D Computer Graphics is a plus

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 $117,000.00-$234,000.00

‎ 

Additional compensation includes annual or quarterly performance bonuses.

‎ 

Additional compensation for certain positions may also include:

‎ 

‎ 

- Stock

‎ 

‎ 

Minimum Qualifications...

Outlined below are the required minimum qualifications for this position. If none are listed, there are no minimum 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 years' experience in an analytics related field. Option 3 - 5 years' experience in an analytics or related field.

Preferred Qualifications...

Outlined below are the optional preferred qualifications for this position. If none are listed, there are no preferred qualifications.

Data science, machine learning, optimization models, Master’s degree in Machine Learning, Computer Science, Information Technology, Operations Research, Statistics, Applied Mathematics, Econometrics, 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.

Primary Location...

680 West California Avenue, Sunnyvale, CA 94086-4834, United States of America