Staff, Data Scientist - Machine Learning Engineer | Catalog Trust & Safety Team

Trust and Safety (T&S) team is an integral part of the Catalog Data Science Org and is responsible for maintaining customer trust in the Walmart marketplace. We employ state-of-the-art GenAI and Machine Learning models to identify products that violate Walmart’s marketplace compliance policies. Our end-to-end machine learning pipelines are designed to scale and detect policy violations in millions of items, ensuring a safe marketplace for our customers. Our work has very high visibility as it directly impacts the marketplace growth at Walmart.

As a Staff Data Scientist on the Trust & Safety team, you will collaborate with other Data Scientists and ML Engineers to develop and deploy machine learning models in production systems. You will drive the migration of all models to a new GenAI platform, significantly enhancing the efficiency and accuracy in detecting defective items that violate compliance policies.

• Evaluate and enhance Deep Learning practices within Catalog Trust and Safety (T&S) team at Walmart
• Use GenAI, NLP, Computer Vision and other deep learning techniques to build models for classification, segmentation and detection for T&S use cases
• Collaborate Cross-Functionally with Compliance Business, Product, Operations, and other engineering teams to align on engagement strategies, business requirements, and model KPIs
• Drive innovation in ML Architecture by implementing and optimizing highly scalable model architectures tailored to Trust & Safety

• PhD or Master’s in computer science with 5+ years of experience developing and deploying machine learning models in production systems
• Solid understanding and hands-on experience with both supervised and unsupervised machine learning models
• Proven track record of training, tuning, and deploying both classical machine learning and deep learning models
• Comprehensive knowledge of Deep learning models and techniques including Neural Networks, CNN, RNN, Transformers and others
• Experience in building and deploying large-scale training and serving pipelines for batch and real-time applications
• Proficiency in Python, Spark, SQL, and associated Python packages commonly used for machine learning like Numpy, PyTorch, TensorFlow, scikit-learn etc
• Ability to solve intricate problems with multilayered data sets and multi batch inferencing
• Commitment to uphold ML engineering best practices - enforcing high standard for quality, reliability, and security in deployed machine learning solutions
• Proactive approach in staying abreast with the latest AI/ML technologies and acting as a thought leader for ML Engineering within the organization and broader technical community
• Solid understanding and practical experience with deploying ML pipelines using Docker, Kubernetes, Apache Airflow, Beam, Spark, Kafka etc.
• Practical experience with GCP, Azure, AWS or other cloud environments
• Working knowledge of relational and NoSQL databases
• Strong verbal and written communication skills
• Experience with Version Control Systems, especially Git
• Experience of CI/CD and familiarity with various tools for automating the deployment, scaling, and management of machine learning models

Walmart’s culture is a competitive advantage, and it’s fostered by being together. Working together in person allows us to collaborate, align quickly and innovate with greater speed.  We use our campuses to create purposeful connection rooted in deepening understanding and investing in the development of our associates.

 
Our hubs: Walmart is a global company with offices across the United States and around the world. Our global headquarters is in Bentonville, Arkansas, with primary hubs in the San Francisco Bay area and New York/New Jersey.