AI/ML Engineer
BambooHRSalary not specified
Full Time
Mid-level (3 to 4 years)
Not SpecifiedCompensation
Senior (5 to 8 years)Experience Level
Full TimeJob Type
UnknownVisa
Advertising Technology, Retail, Consumer ElectronicsIndustries
Requirements
- 5 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
- Preferred experiences
- 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
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
Skills
GenAI
LLM
PyTorch
TensorFlow
ONNX
TVM
Arm NN
Qualcomm SNPE
CI/CD
Federated Learning
Machine Learning
Data Pipelines
Deep Learning
Walmart
Global retail chain offering affordable products
About Walmart
Walmart operates a chain of hypermarkets, discount department stores, and grocery stores, providing a wide variety of products at low prices to everyday consumers, families, and communities. The company combines physical retail locations with an e-commerce platform, allowing customers to shop for groceries, apparel, electronics, and household items both in-store and online. Walmart's efficient supply chain management and large-scale purchasing enable it to maintain affordability, setting it apart from competitors. The company also offers financial services and health and wellness services, including pharmacies and vision centers. Walmart's goal is to save people money and support communities, as demonstrated by its efforts during the COVID-19 pandemic and its commitment to social responsibility, including programs for veterans and military families.
Bentonville, ArkansasHeadquarters
1962Year Founded
$5,515MTotal Funding
IPOCompany Stage
Financial Services, Consumer GoodsIndustries
10,001+Employees
Benefits
PTO: Paid vacation, sick time, personal time and holiday time
10% discount on regularly priced general merchandise and fresh fruits and vegetables
6% 401(k) match to all employees, including hourly workers, after one year
Roth IRA available
Associate Stock Purchase Plan
maximum and eligible preventive care covered at 100%
Health reimbursement plans
Risks
Rising U.S. labor costs could pressure profit margins.
Amazon's grocery delivery expansion threatens Walmart's market share.
Fintech investments expose Walmart to financial regulatory risks.
Differentiation
Walmart's omnichannel strategy integrates physical and online retail seamlessly.
The company offers a wide range of products at consistently low prices.
Walmart's commitment to community support enhances its brand reputation.
Upsides
Walmart's investment in fintech expands its financial services offerings.
Robotics integration in distribution centers boosts operational efficiency.
Culturally themed product lines attract diverse customer segments.
Related Jobs
RemoteRemoteEdge Engineer
RoboflowSalary not specified
- Full Time
- Senior (5 to 8 years), Expert & Leadership (9+ years)
RemoteGenerative AI Annotation Operations Engineer
Sustainable TalentSalary not specified
- Full Time
- Mid-level (3 to 4 years), Senior (5 to 8 years)
RemoteStaff Machine Learning & Computer Vision Engineer
Dandy$232,500 - $258,800/year
- Full Time
- Senior (5 to 8 years)
RemoteMachine Learning Engineer
SweedSalary not specified
- Full Time
- Mid-level (3 to 4 years)
RemoteAI Platform Engineer
0x LabsSalary not specified
RemoteTechnical Project Manager - Deep Learning Frameworks
OpenTeamsSalary not specified
- Full Time
- Senior (5 to 8 years), Expert & Leadership (9+ years)
RemoteResearch Engineer, Agents
AnthropicSalary not specified
- Full Time
- Junior (1 to 2 years)