All roles at PubMatic

Senior Machine Learning Platform Engineer

PuneFull-timeHybridSenior levelPosted open since mar 2026

Senior DevOps and cloud role at PubMatic in Pune. Needs a B.Tech or B.E.. Hybrid.

About the role

As a Senior Machine Learning Platform Engineer , you will design and scale the infrastructure and frameworks that enable machine learning development, experimentation, and production across a global ecosystem handling trillions of ad impressions. You will collaborate closely with ML Engineers, Data Scientists, and Product stakeholders to accelerate experimentation, maximize efficiency, and translate AI solutions from concept to production.

What you'll do

  • Experimentation & Observability: Develop frameworks for experiment tracking, automated reporting, and observability to monitor model health, drift, and anomalies
  • Work with industry-standard ML efficiency tools to optimize training workloads, accelerate experiments, and monitor performance at scale
  • Work with petabyte-scale datasets and billions of transactions, powering global adtech
  • Apply AI/ML to deal troubleshooting, competitive intelligence, benchmarking, forecasting, and actionable insights
  • Build advanced frameworks such as RAG systems, reinforcement learning strategies, and embedding platforms

What you need

  • Machine learning
  • CI/CD
  • Terraform
  • Docker
  • Kubernetes
  • Python

Good to have

  • Grafana
  • MLOps
  • Spark
  • Hadoop
  • Kafka
  • Snowflake
Apply on PubMatic's site