Senior Machine Learning Platform Engineer at PubMatic | AdTechTalent
Home / Jobs / Senior Machine Learning Platform Engineer Engineering 5 months ago Hybrid
PubMatic
Senior Machine Learning Platform Engineer
machine learning ml ops big data spark hadoop kafka snowflake sql python scala java docker kubernetes airflow mlflow gpu computing triton inference neo4j rag embedding recommendation systems personalization anomaly detection ci/cd data pipelines experiment tracking analytics bi tools lookers grafana
Ready to apply?
Join PubMatic and work on a role built for experienced AdTech operators.
Full job description Senior Machine Learning Platform Engineer role at PubMatic to design and scale ML infrastructure and frameworks for development, experimentation, and production across global adtech ecosystem. Responsibilities include building scalable ML pipelines, optimizing large-scale data workflows with Spark, Hadoop, Kafka, Snowflake, developing experiment tracking and observability frameworks, providing reusable AI/ML components and APIs, driving CI/CD and infrastructure-as-code for ML jobs, collaborating cross-functionally, and innovating with Generative AI, ML Ops, and Big Data. Requires 4+ years experience with petabyte-scale datasets, proficiency in SQL, Python, analytics/BI tools, exposure to graph-based systems, RAG, embedding platforms, experience with recommendation systems, anomaly detection, BI tools like Looker/Grafana, strong expertise in Spark, Hadoop, Kafka, Snowflake, CI/CD, Docker, Kubernetes, Airflow/MLflow, and programming in Python/Scala/Java. Bachelor’s degree in engineering (CS/IT) or equivalent required. Hybrid work schedule with 3 days in office, 2 remote. Benefits include parental leave, healthcare insurance, broadband reimbursement, snacks, and catered lunches.
What you'll do Design and maintain scalable ML pipelines and platforms for ingestion, feature engineering, training, evaluation, inference, and deployment. Similar jobs
More roles worth a look Related opportunities based on specialty and working model so candidates can keep momentum.
TripleLift
Detroit, United States; New York, US • 4 months ago
$75K – $95K
accounting payroll compensation
TripleLift
Pune, India • 4 months ago
ad ops programmatic campaign management
TripleLift
Build and optimize large-scale data workflows using distributed systems (Spark, Hadoop, Kafka, Snowflake) to support analytics and model training.
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.
Provide reusable components, SDKs, and APIs that empower teams to leverage AI insights and ML models effectively.
Drive CI/CD, workflow orchestration, and infrastructure-as-code practices for ML jobs, ensuring reliability and reproducibility.
Partner cross-functionally with Product, Data Science, and Engineering teams to align ML infrastructure with business needs.
Stay ahead of emerging trends in Generative AI, ML Ops, and Big Data to introduce best practices and next-gen solutions.
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.
Convert business challenges into ML products, pioneering industry-first solutions.
Gain hands-on exposure to GPU/accelerated computing, Triton inference, and modern ML Ops frameworks.
Advance your career with a clear growth path into applied ML engineering and research.
Be part of a culture that values experimentation, thought leadership, and cross-functional collaboration. Requirements Four plus years of hands-on experience with petabyte-scale datasets and distributed systems. Proficiency in SQL, Python (pandas, NumPy), and analytics/BI tools for data exploration and monitoring. Exposure to graph-based systems (Neo4j), RAG approaches, or embedding platforms. Experience building recommendation systems, personalization pipelines, or anomaly detection frameworks. Familiarity with BI tools such as Looker or Grafana. A passion for applied AI/ML and eagerness to bring research ideas into production. Strong expertise with Spark, Hadoop, Kafka, and data warehousing (Snowflake, SparkSQL). Experience with CI/CD, Docker, Kubernetes, Airflow/MLflow, and experiment tracking tools. Skilled in Python/Scala/Java for data-intensive applications. Strong analytical skills and ability to debug complex data/ML pipeline issues. Excellent communication and teamwork skills in cross-functional settings. Bachelor’s degree in engineering (CS / IT) or equivalent degree from well-known Institutes / Universities. Tech stack Spark Hadoop Kafka Snowflake SQL Python pandas NumPy Neo4j Looker Grafana Docker Kubernetes Airflow MLflow Scala Java Triton inference GPU/accelerated computing
Benefits Paternity/maternity leave Healthcare insurance Broadband reimbursement Kitchen loaded with healthy snacks and drinks Catered lunches
Apply now Ready to take the next step in your career? Click the button below to continue to the application process.
Company
PubMatic PubMatic (NASDAQ: PUBM) is the leading AI-powered ad tech company delivering digital advertising performance. Through an intelligent, unified platform that connects buyers, publishers, data partners, and commerce media networks, PubMatic delivers superior performance with greater transparency, control, and efficiency.
Since 2006, PubMatic has pioneered every major advance in programmatic advertising, from enabling the first OpenRTB transactions to embedding AI-driven optimization and privacy-focused innovation across its platform. With omnichannel scale, proven reliability, and a track record of continuous innovation, PubMatic is building a more intelligent, profitable, and sustainable open internet. Built to Connect. Powered to Perform
Industry
Technology, Information and Internet
Website
https://pubmatic.com/
Posted
5 months ago
Category: Engineering
Chicago, United States; Detroit, United States; New York, US; Toronto, Canada • 4 months ago
salesforce apex lightning web components
