(USA) Staff, Software Engineer

Walmart • Sunnyvale, CA

2d • Onsite

About The Position

Walmart's Advertising Technology group enables the connection between supplier brands and retail shoppers at unprecedented scale. We are a highly motivated group of engineers and data scientists, working in an agile group to solve sophisticated and high impact problems. We serve billions of ad requests every month with our high-performance ad servers. We are building a first‑party, privacy‑centric advertising platform that unites retail data, smart‑TV inventory, and a growing CTV publisher network. Our mission is to deliver relevant, measurable ad experiences while giving consumers full control of their data. What you'll do... We are building the next generation of AI powered experiences on smart TVs think on device large language models, real time content understanding, privacy preserving audience insights, and interactive generative graphics that transform millions of households to discover, shop, and play. You will be one of the first engineers bridging deep learning research with production code on consumer electronics hardware. About the team: Walmart's Advertising Technology group enables the connection between supplier brands and retail shoppers at unprecedented scale. We are a highly motivated group of engineers and data scientists, working in an agile group to solve sophisticated and high impact problems. We serve billions of ads requests every month with our high-performance ad servers. There are millions of customers shop on Walmart websites and in stores daily, and advertising helps advertisers bring the best products to our customers. If you want to influence millions of customers on their shopping journeys, we have a role for you. The AdTech M&R data team is responsible for delivering reporting and measurement for Advertisers to analyze and optimize campaigns. We are a team of data developers and machine learning developers whose strengths are: (1) building scalable data pipelines (2) using machine learning techniques and data science (3) making sense of broadly defined problems through data analysis.

Requirements

• 8 years building and shipping ML or DL products (or 2 with a PhD degree).
• Proven track record in generative models (LLMs, diffusion, transformers, VAEs) and modern CV/NLP stacks.
• C++ / Rust and Python expertise; comfortable refactoring kernels for efficiency on limited memory devices.
• Hands on experience with edge inference (TensorRT, CoreML, TVM, ONNX, TFLite, WebGPU, or similar).
• Familiarity with CTV / ACR pipelines (frame grabbers, fingerprinting, video embeddings) or embedded multimedia systems.
• Strong grasp of distributed training (DDP/Horovod) and MLOps (Kubeflow, Airflow, MLFlow, Feature stores).
• Knowledge of data privacy frameworks (FL, DPSGD, HE, SMPC) and ability to translate compliance constraints into code.
• Bachelor or higher in CS, EE, Math, or related field.
• 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

• Background in ad tech or retail media optimization (incrementally real time ROAS auction dynamics).
• Experience optimizing models for ARM Cortex‚ NEON, NPU, or VVC ASICs inside smart‚ TV SoCs.
• Contributions to OSS projects in quantization, model compression, or video ML (e.g., bits and bytes, MLC‚AI, PyTorch‚ Video).
• Familiarity with incrementality measurement, attribution, and real time auction dynamics (e.g., bid shading, floor price optimization).
• Working knowledge of data privacy frameworks (FL, DP SGD, HE, SMPC) and ID resolutions (UID2, RampID, SharedID).
• Publications/patents in generative media, multimodal learning, or ACR.
• 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

• Design & train GenAI models targeting CTV use cases: on device LLM quantization, multimodal video text encoders, and diffusion based visual experience generators.
• Deploy to edge environments optimize PyTorch/TensorFlow models with ONNX RT/TVM, Arm NN, or Qualcomm SNPE; own CI/CD pipelines that push updates to millions of TVs.
• Integrate privacy enhancing tech such as federated learning, encrypted feature extraction, and PSI to align with global data regulations and Walmart stack.
• Prototype novel user features (e.g., conversational search, generative ad creatives, adaptive picture modes) in collaboration with product, design, and ad tech teams.
• Publish & share file patents, present at CVPR/NeurIPS, mentor junior ML engineers, and contribute to open source edge AI tooling.

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.
• 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.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level