Staff, Data Scientist

What you'll do

• Design and build embedding-based classification models to align external product signals with Walmart's merchandising hierarchy.
• Develop GMV distribution normalization and projection models for reconciling internal and external sales data across categories, time, and geography.
• Construct causal impact models to quantify market-share movement from merchandising actions using advanced statistical methods.
• Engineer ML systems on Databricks (on GCP) for production deployment, including PySpark, Delta, and MLflow integration.
• Establish MLOps foundations for continuous monitoring and scaling of the ROM platform by new team members.
• Own end-to-end lifecycle of models from problem framing to incident response in a production environment.

What we're looking for

• Extensive hands-on experience shipping ML systems from notebook to production at scale and maintaining them through monitoring and retraining.
• Deep expertise in developing and scaling ML on Databricks using PySpark, Delta, MLflow, Unity Catalog, and Model Serving.
• Strong proficiency with GCP services including BigQuery, Vertex AI, Cloud Run, and Composer/Airflow for seamless integration with Databricks.
• Proven skills in vector embeddings, supervised classification at scale, forecasting, distribution modeling, and causal inference for retail data.
• Experience as a founding or solo data scientist on a program, establishing MLOps foundations from scratch and ensuring engineering-level rigor.