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.

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.
Preferred qualifications: 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.

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.
Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills

machine learning algorithmsdeep learningmulti-modal LLMsdiffusion models3D representation learningspatial modelingkeypoint modelingPythonPyTorchTensorFlow

Soft skills

cross-functional collaborationcommunicationproblem-solvinginnovationoptimization

Certifications

Master's degree in Computer SciencePhD in Machine Learning