Staff Data Scientist: Personalization & Recs at Scale

Socket.dev; New York (NY); On-site; USD 132,000 - 264,000; Full time

Benefits offered by this job

Health benefits

401(k) plan

Stock purchase program

Paid time off

Job summary

Walmart is seeking a Staff Data Scientist in New York to lead the development and deployment of machine learning solutions powering personalization and ad targeting across the Walmart platform. You will translate business needs into scalable ML systems, drive rigorous evaluation, and guide experiments to deliver measurable impact.

The role requires deep learning, PyTorch, PySpark, and production ML experience, with collaboration across Product and Engineering.

Qualifications

• Design, train, and deploy ML models for real-world personalization and ad targeting.
• Proven ability to translate product questions into scalable ML solutions.
• Strong foundation in statistics, optimization, and model evaluation.

Responsibilities

• Lead ML projects from ideation to deployment and monitoring.
• Collaborate with Product and Engineering teams to align on priorities.
• Maintain rigorous offline and online evaluation frameworks for models.

Skills

Machine learning

Deep learning

Python

PyTorch

Experimentation

Education

Bachelor's degree

Master's degree

Experience

Tools

Spark

PySpark

TensorFlow

MLflow

Job description

Walmart is seeking a Staff Data Scientist in New York to lead the development and deployment of machine learning solutions powering personalization and ad targeting across the Walmart platform. You will translate business needs into scalable ML systems, drive rigorous evaluation, and guide experiments to deliver measurable impact.

The role requires deep learning, PyTorch, PySpark, and production ML experience, with collaboration across Product and Engineering.