Staff Software Engineer - Python, Computer Vision

Warner Bros Discovery Warner Bros Discovery · Media · Bangalore, Karnātaka, India · Technology

Staff Software Engineer for the Video AI Team, focusing on building a scalable platform for training, deployment, and observability of ML models for video applications like segmentation, annotation, clip generation, and understanding. The role involves designing core platform components, building ML pipelines, deploying and optimizing CV/Generative AI models, and working with cloud-native systems on GCP.

What you'd actually do

  1. Design and lead the development of core components of the machine learning platform with a focus on scalability, reliability, and cloud-agnostic principles.
  2. Build APIs and microservices that power end-to-end machine learning workflows, including model deployment, orchestration, model registry integration, and observability.
  3. Architect and deploy AI/ML pipelines for video ingestion, preprocessing, annotation, training, evaluation, serving, monitoring, and continuous improvement.
  4. Develop cloud-native AI systems on Google Cloud Platform using Vertex AI, Vertex AI Pipelines, Vertex AI Endpoints, GKE, Cloud Run, BigQuery, Cloud Storage, Pub/Sub, and related services.
  5. Own cloud infrastructure and CI/CD automation using Infrastructure-as-Code principles, with hands-on implementation using Terraform, Kubernetes, Docker, and deployment automation tools.

Skills

Required

  • Python
  • GCP
  • Vertex AI
  • Kubernetes
  • Docker
  • Terraform
  • PyTorch
  • TensorFlow
  • Hugging Face
  • Computer Vision
  • Generative AI
  • Deep Learning
  • backend systems
  • distributed services
  • cloud-native platforms
  • API design
  • architecture skills

Nice to have

  • Multimodal AI models
  • Vision Transformers
  • LLM
  • VLM
  • RAG

What the JD emphasized

  • production deployment of machine learning systems
  • Deep Learning, Computer Vision, and Generative AI models
  • evaluating, fine-tuning, training, and deploying open-source Computer Vision and Multimodal AI models for production use cases

Other signals

  • ML platform
  • model deployment
  • video understanding
  • Computer Vision
  • Generative AI