Warner Bros. Discovery is seeking a Machine Learning Engineer to join our AI/ML organization. You will build reusable components and services for video understanding, summary, and classification, supporting production workflows through robust infrastructure and backend services.
What You'll Do
- Build and maintain pipelines for model fine-tuning and retraining, including LoRA-based workflows and Large Language Models.
- Integrate and maintain vector search services and semantic similarity infrastructure.
- Design scalable model serving solutions for open-source and foundation models.
- Develop systems for experiment tracking, model versioning, and evaluation.
- Monitor production models for drift and performance degradation.
- Manage compute cost and resource optimization across distributed training jobs.
- Integrate Human-in-the-Loop (HITL) workflows and offline labeling into training pipelines.
- Support model deployment for varied model architectures, including Vision-Language Models, Convolutional Neural Nets, and Embedding Generation models.
- Stand up and maintain Feature Store and data versioning infrastructure.
- Architect and implement RAG pipelines for video metadata, summarization, and Q&A.
- Build evaluation frameworks to assess LLM performance, hallucination frequency, and structured response accuracy.
What We're Looking For
- 5+ years of experience in machine learning engineering, with end-to-end ML workflow expertise.
- Strong background in model retraining, fine-tuning, and evaluation techniques.
- Experience deploying and managing open-source model servers (e.g., Triton, TorchServe, Ray Serve).
- Proficient in managing cost-effective distributed computing environments (e.g., Kubernetes, Ray, SageMaker).
- Familiar with experiment tracking tools (e.g., MLflow, Weights & Biases) and model versioning strategies.
- Deep understanding of ML domains including NLP, RecSys, and reinforcement learning.
- Familiarity with labeling tools, HITL workflows, and offline data curation strategies.
- Comfort working in Agile development environments and collaborating across global teams.
Nice to Have
- Experience with real-time inference systems and streaming data pipelines is a plus.
Technical Stack
- Triton, TorchServe, Ray Serve, Kubernetes, Ray, SageMaker, MLflow, Weights & Biases
Team & Environment
Part of the AI/ML organization, focusing on supporting applications of AI to video.
Benefits & Compensation
- Salary: $132,300.00 - $245,700.00 per year
- Health insurance coverage
- Employee wellness program
- Life and disability insurance
- Retirement savings plan
- Paid holidays, sick time, and vacation
- Annual bonuses
- Short- and long-term incentives
- Program-specific awards
Warner Bros. Discovery embraces the opportunity to build a workforce that reflects a wide array of perspectives, backgrounds and experiences. Being an equal opportunity employer means that we take seriously our responsibility to consider qualified candidates on the basis of merit, without regard to race, color, religion, national origin, gender, sexual orientation, gender identity or expression, age, mental or physical disability, and genetic information, marital status, citizenship status, military status, protected veteran status or any other category protected by law.
