Senior Machine Learning Engineer

Warner Bros Discovery Warner Bros Discovery · Media · San Francisco, CA +2 · Technology

Senior Machine Learning Engineer on the Data & Audience Platform team, focusing on building foundational AI/ML intelligence for identity, audience, advertising, and personalization across Warner Bros. Discovery brands. The role involves owning the end-to-end development of production ML systems, including data sourcing, feature engineering, model training, evaluation, deployment, and monitoring. Key responsibilities include technical leadership, architectural decisions, designing scalable feature and inference pipelines, and developing various model types. The role also emphasizes MLOps best practices and contributing to agentic AI development workflows.

What you'd actually do

  1. Lead end-to-end development of production ML systems: data sourcing, feature engineering, model training, evaluation, deployment, and monitoring.
  2. Own one or more flagship ML products — e.g., probabilistic identity resolution (matching unauthenticated device IDs and 1P cookies to households/persons with calibrated confidence), single-title affinity (two- tower retrieval), lookalike modeling, or forecasting — and drive their technical direction.
  3. Make and document key architectural decisions across a workstream (feature-store design, training/serving patterns, evaluation frameworks); provide deep trade-off analysis on scalability, latency, reliability, and cost.
  4. Design scalable feature and inference pipelines on Databricks (PySpark, Delta, Workflows/DLT, Unity Catalog) integrated with Snowflake and activation systems (Mosaic, FreeWheel, GAM), with documented feature contracts, backfill paths, and freshness SLAs.
  5. Establish and evangelize patterns that other engineers adopt; anticipate risks and failure modes before they surface.

Skills

Required

  • design and delivery of production ML systems end to end
  • technical leadership
  • architectural decisions
  • MLOps best practices
  • model versioning
  • champion/challenger promotion
  • automated retraining triggers
  • drift detection
  • production monitoring
  • Databricks (PySpark, Delta, Workflows/DLT, Unity Catalog)
  • Snowflake
  • AWS
  • gradient boosting (XGBoost/LightGBM)
  • embedding/two-tower retrieval
  • neural ranking
  • probability calibration
  • probabilistic/graph-based matching
  • offline and online experiments
  • evaluation frameworks (precision/recall, AUC-ROC, NDCG, decile lift, calibration curves)
  • causal-inference techniques (propensity scoring, uplift/incrementality modeling)
  • lookalike modeling
  • Data Clean Rooms (Snowflake DCR)

Nice to have

  • agentic AI development workflows

What the JD emphasized

  • own the design and delivery of production ML systems end to end
  • take on cross-cutting technical leadership
  • setting patterns
  • driving key architectural decisions
  • raising the bar for the broader ML organization
  • own one or more flagship ML products
  • Make and document key architectural decisions
  • Design scalable feature and inference pipelines
  • Establish and evangelize patterns

Other signals

  • production ML systems
  • ML infrastructure
  • feature stores
  • training and serving pipelines
  • MLOps
  • agentic AI development workflows