Staff, Data Scientist

Job description

Position Summary... What you'll do... We are looking for a computer vision scientist to solve some real retail problems/challenges. Utilize Computer Vision algorithms, Generative AI methods, deep learning architectures, VLMs and state-of-the-art techniques to build and deploy advanced retail solutions. About Team: At Walmart, we prioritize innovation and data security. Our team is dedicated to maintaining a secure operating environment and preserving the trust of our customers, associates, and stakeholders. We combine a range of services and expertise to prevent fraud, detect threats, and manage digital risk and access. Our focus is on mitigating attack risks, securing cloud transformation, and fostering a culture of security and reliability within our team What you'll do: Design, train, and deploy deep learning models for object detection, image-based personalization, including product discovery, visual recommendations, and generative image content. Build and scale foundation model pipelines that leverage multimodal signals - including text, images, and customer embeddings - to power highly personalized visual experiences. Lead innovation efforts on image generative modeling, including diffusion models, CNNs, or text-to-image synthesis aligned with retail context. Implement efficient image understanding models for visual similarity, visual search, or classification across Walmart's vast product catalog. Design model workflow that integrates with MLOps to optimize model performance and latency, especially for customer-facing features with visual components. Contribute to cutting-edge research and internal knowledge by publishing work or presenting findings on vision-language models, multimodal personalization, and generative AI for e-commerce. What you'll bring: MS or (preferably) PhD in Computer Science, Machine Learning, Computer Vision, Applied Mathematics, or a related technical field, with 2-4 years of industry or research experience. Demonstrated expertise in deep learning for computer vision, including CNNs, vision transformers (ViTs), and diffusion-based generative models. Experience building and deploying image generation, manipulation, or understanding systems in production - such as visual search, personalized image recommendations. Proficiency in Python and strong familiarity with deep learning frameworks such as PyTorch, TensorFlow, and image m…