(USA) Staff, Software Engineer
WalmartAbout the role
Position Summary...
What you'll do...
We’re 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.
What You’ll Do 
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. 
What you'll bring:
- 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, DP‑SGD, HE, SMPC) and ability to translate compliance constraints into code. 
- Bachelor’s or higher in CS, EE, Math, or related field. 
Preferred experiences:
- Background in ad‑tech or retail media optimization (incrementally, real‑time ROAS, auction dynamics). 
- Experience optimizing models for ARM Cortex‑A, 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). 
Apply for this role
Generate a tailored application kit with a matched cover letter, interview prep, and CV highlights — in under 60 seconds.
Apply Now →Generate Application KitFree account required — sign up in 30s