Machine Learning Engineer II ( Data & Audience Platform Team), Hyderabad
Core
Build and maintain production ML pipelines for identity resolution, audience segmentation, and content affinity modeling to drive advertising yield and personalization.
Role type
Machine Learning Engineer II (hands-on IC)
Builds
Production ML systems, data pipelines, and feature stores for WBD's Data & Audience Platform
Domain
Media & Entertainment / AI & Machine Learning
Deliverable
production ML models
Required skills
Python, PySpark, SQL, scikit-learn, PyTorch, MLflow, model evaluation, data structures
Preferred skills
Databricks (Delta Lake, Unity Catalog), AWS SageMaker, recommendation systems, agentic AI tools
Technologies
Databricks, Snowflake, AWS, PySpark, MLflow, XGBoost, LightGBM, PyTorch
Responsibilities
Build end-to-end ML pipelines for training and batch inference; Engineer features from viewership and behavioral signals; Develop data pipelines on Databricks feeding feature stores; Deploy and monitor batch inference jobs; Use AI-assisted development tools for coding and data exploration.
Seniority
Mid-level, hands-on IC