Senior Data Engineer - AI Systems

The AI Systems team owns the core recommendations engine and ML platform that powers billions of AI-driven marketing decisions daily across some of the world's largest consumer brands. As a Senior Data Engineer, you will own the Spark-based data pipelines and data infrastructure at the heart of this system - building, scaling, and optimizing the data layer that feeds our production ML models. You will work alongside ML engineers and scientists in a collaborative environment, contributing data pipelines and products to power our core recommender systems and our DaVinci Personalization product. This is an opportunity to work end-to-end on large-scale data systems that touch millions of customers, on a team working at the intersection of data engineering and machine learning.

This role will be reporting to the Director of Engineering (AI/ML).

Responsibilities:

• Build, maintain, and optimize production data pipelines that power AI-driven personalization at scale across content selection, send-time optimization, subject line personalization, and frequency capping
• Own and scale Spark-based batch pipelines, including cluster configuration, tuning, and performance optimization across GCP Dataproc
• Build and maintain our ML Data Lake, ensuring data quality, accessibility, and efficient storage
• Support the data needs of ML Engineers and Scientists for model development, training, and evaluation
• Identify and resolve performance bottlenecks and scaling limitations in data pipelines and infrastructure
• Collaborate with distributed systems engineers on the platform's architectural evolution, ensuring data layer continuity throughout
• Continuously improve data infrastructure for greater scalability and reliability
• Release features and data products that deliver measurable and tangible business value

Qualifications:

• 5+ years of data engineering experience
• Deep expertise with Apache Spark, including the PySpark DataFrame API and experience solving challenging scaling problems
• Experience with large-scale data processing, cluster configuration, optimization, and tuning (we use GCP Dataproc)
• Strong software development skills in Python (unit testing, git, code review, CI/CD)
• Experience with data storage formats (we use Parquet, Delta Lake)
• Experience with event streaming data (we use Kafka)
• Experience with cloud computing platforms (we use Google Cloud Platform)
• Experience with advanced query optimization
• Familiar with Software Development Lifecycle practices, such as continuous integration/continuous delivery and automated deployment (we use Docker, Kubernetes, and GitHub Actions)
• Ability to collaborate with technical partners - you'll be working closely with ML engineers, scientists, and other teams to determine requirements and make design decisions
• Enjoys working in a fast-paced, goal-driven environment

The base pay range for this position is $165K - $215K USD/year, which can include additional bonus depending on the position ultimately offered, in addition to a full range of medical, financial, and/or other benefits. The base pay offered may vary depending on job-related knowledge, skills, and experience.