(USA) Staff, Software Engineer | MLE

Walmart•Sunnyvale, CA

1d•Onsite

About The Position

Position Summary... What you'll do... . 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 a specialization in Computer Vision, Machine Learning, or equivalent practical experience.
3+ years of experience with machine learning algorithms and tools, with at least 1 year dedicated to deep learning.
Proven experience launching ML models in production environments and building scalable pipelines/services.
Hands-on experience with multi-modal LLMs, diffusion models, 3D representation learning, or spatial/keypoint modeling.
Proficiency in Python, along with frameworks such as PyTorch, PyTorch3D, TensorFlow, TensorFlow 3D, JAX, or NumPy.
Experience with Unix, Docker, and optimizing workflows for CPU/GPU architectures.
Option 1: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 4 years’ experience in software engineering or related area.
Option 2: 6 years’ experience in software engineering or related area.

Nice To Haves

PhD in Machine Learning, Computer Science, or a related technical field.
Experience in digital commerce, AR/VR, or rich media personalization.
Deep expertise in 3D Deep Learning (e.g., PyTorch3D, NVIDIA Kaolin) and 3D Modeling tools (e.g., Maya, Blender, photogrammetry).
Experience building Augmented Reality applications using Unity or other AR SDKs.
Deep experience with scalable ML infrastructure and content delivery systems.
Understanding of human-in-the-loop evaluation, perceptual quality metrics, and responsible AI frameworks.
Recognized contributions to the field through publications, patents, open-source projects, or public talks.
Master’s degree in Computer Science, Computer Engineering, Computer Information Systems, Software Engineering, or related area and 2 years' experience in software engineering or related area
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

Develop Multi-Modal GenAI Models: Design and build models for content generation (imagery, video, 3D assets) and content intelligence (segmentation, object detection, spatial awareness, and keypoint detection).
Scale Data Pipelines: Process large-scale datasets using distributed cloud computing technologies to create and maintain high-fidelity 2D and 3D assets.
Ensure Model Excellence: Establish robust experimentation frameworks to evaluate model performance across realism, accuracy, latency, and downstream usability.
Drive Production Integration: Collaborate end-to-end with product, engineering, and annotation partners to deploy scalable AI applications and manage their lifecycle.
Innovate & Optimize: Translate state-of-the-art research papers into production-grade solutions, improving every aspect of the pipeline—from data generation techniques to efficient model inference.
Cross-Functional Collaboration: Partner with internal Data Science teams to enable high-quality, AI-powered user experiences in the computer vision space.

Benefits

Beyond our great compensation package, you can receive incentive awards for your performance.
Other great perks include 401(k) match, stock purchase plan, paid maternity and parental leave, PTO, multiple health plans, and much more.
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.
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.