Machine Learning Engineer II ( Growth & ML), Hyderabad
Warner Bros. Discovery1 day ago
Hyderābād, IndiaMid Level
Responsibilities
- Build and maintain end-to-end ML pipelines for training, evaluation, and batch inference across identity resolution, audience segmentation, and content affinity use cases.
- Implement and experiment with supervised, unsupervised, ranking, and deep-learning models using Python and relevant ML frameworks.
- Engineer features from viewership, engagement, subscription, and behavioral signals while addressing data leakage, collinearity, and training-serving skew.
- Run offline experiments, evaluate models using appropriate metrics, and document findings in MLflow.
- Develop and maintain Databricks data and feature pipelines using PySpark, Delta, and Workflows for feature-store and model-training workflows.
- Write tested, production-quality Python and support model registration, versioning, deployment, and monitoring of batch inference jobs.
- Use AI-assisted development and data-exploration tools, including Cursor, GitHub Copilot, Amazon Q, Databricks Genie, Snowflake Cortex, and MCP tooling.
- Collaborate with Senior and Staff MLEs, Data Engineering, Feature Engineering, and Analytics teams on system design, data quality, and pipeline reliability.
- Document models, pipelines, and experiments for team knowledge sharing.
Requirements
- 2–4 years of industry experience in machine learning, data science, or ML engineering, or 1–2 years with a relevant M.S.
- Strong Python proficiency with experience in pandas, NumPy, scikit-learn, and at least one deep-learning framework such as PyTorch or TensorFlow.
- Hands-on experience with Spark, PySpark, or equivalent large-scale data processing.
- Proficiency in SQL and familiarity with cloud data warehouses or lakehouses such as Snowflake or Databricks.
- Experience with experiment-tracking tools such as MLflow or Weights & Biases.
- Understanding of classification, regression, ranking, embeddings, model evaluation, data structures, algorithms, and clean code.
- Bachelor’s degree in Computer Science, Statistics, Engineering, or a related quantitative field, or equivalent practical experience.
- Preferred experience with Databricks notebooks, Delta Lake, Workflows or DLT, Unity Catalog, and MLflow.
- Preferred exposure to recommendation systems, audience segmentation, identity resolution, forecasting, AWS services, or feature stores such as Databricks Feature Store or Feast.
- Preferred exposure to Cursor, GitHub Copilot, Amazon Q, Databricks Genie, Snowflake Cortex, or MCP.
- Ability to use AI tools to improve productivity across the ML lifecycle and communicate clearly in writing and verbally.
Benefits
- A great place to work.
- Equal opportunity employer.
- Fast-track growth opportunities.
Tech Stack
Apache SparkAWSDatabricksLightGBMMLflowNumPyPandasPythonPyTorchscikit-learnSnowflakeSQLTensorFlowXGBoost
Categories
About Warner Bros. Discovery
Warner Bros. Discovery is a global media company that produces and distributes film, television, streaming, sports, news, and games for consumers and distribution partners. Its portfolio includes Warner Bros. Pictures, HBO and Max, CNN, Discovery Channel, HGTV, and DC, with revenue from subscriptions, advertising, licensing, and theatrical releases. Formed in 2022 through the merger of WarnerMedia and Discovery, it is headquartered in New York City and trades on Nasdaq as WBD.