Apply on Employer Site

Walmart Canada · 4 hours ago

(USA) Staff, Software Engineer | MLE

(USA) Crossman Respect Building CA SUNNYVALE Home Office

Full-time

Hybrid

Senior Level, Lead/Staff

$143K/yr - $286K/yr

4+ years exp

Walmart Inc. is a leading retailer that focuses on innovative technology solutions. They are seeking a Staff Software Engineer for their Emerging Tech Extended Reality team to develop advanced multi-modal GenAI models, scale data pipelines, and drive production integration for AI applications.

DeliveryRetailShopping

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

Qualification

Machine LearningDeep LearningComputer VisionPythonData PipelinesMulti-modal LLMs3D ModelingDockerCollaborationProblem Solving

Required

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

Preferred

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

Knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly

Knowledge of accessibility best practices

Benefits

401(k) match

Stock purchase plan

Paid maternity and parental leave

PTO

Multiple health plans

Medical, vision and dental coverage

401(k)

Company-paid life insurance

Short-term and long-term disability

Company discounts

Military Leave Pay

Adoption and surrogacy expense reimbursement

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.