Principal Data Scientist — AI & Graph ML Leader

Safeway; Pleasanton (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Weekly wages

Early wage access

Associate discounts

Health benefits

Time off

Career development

Inclusive environment

Job summary

Albertsons Companies in Pleasanton, CA is seeking a senior AI/ML leader to drive end-to-end deep learning initiatives. You will architect scalable frameworks, govern model development, and mentor teams while partnering with Business and Engineering leaders to deliver measurable AI value.

The role emphasizes distributed training, large-model deployments, and enterprise-grade inference across cloud platforms, with a focus on retail and customer insights.

Qualifications

• PhD strongly preferred in a quantitative field.
• Master's degree with exceptional industry research may be considered.

Responsibilities

• Lead end-to-end deep learning solutions from problem framing to deployment within Databricks Lakehouse.
• Serve as technical authority for Graph Neural Networks and related architectures across high-impact business problems.
• Architect scalable ML platforms with reusable frameworks and governance.

Skills

Deep Learning

GNNs

Graph Representation Learning

Foundation Models/LLMs

PyTorch/TensorFlow

Spark/PySpark

Python

Distributed GPU

Education

PhD in CS/AI/ML

MS with exceptional industry experience

Tools

Databricks Lakehouse

Delta Lake

MLflow

Unity Catalog

Spark

NVIDIA CUDA

Job description

Albertsons Companies in Pleasanton, CA is seeking a senior AI/ML leader to drive end-to-end deep learning initiatives. You will architect scalable frameworks, govern model development, and mentor teams while partnering with Business and Engineering leaders to deliver measurable AI value.

The role emphasizes distributed training, large-model deployments, and enterprise-grade inference across cloud platforms, with a focus on retail and customer insights.