inhousefyi
← Back to listings

Senior Machine Learning Platform Engineer

PubMaticPune, India · Posted 5 months ago
Full-time
Apply now

Description

About the Role

At PubMatic, data operates at unmatched scale. 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. This role offers direct exposure to petabyte-scale datasets, industry-standard ML efficiency tools (e.g., Triton inference, GPU/accelerated computing), and the opportunity to evaluate and adopt emerging AI/ML technologies.

You will contribute to the next-generation ML platform for adtech, enabling advanced use cases such as troubleshooting issues in bid stream, competitive intelligence, benchmarking, forecasting, reinforcement learning, and retrieval-augmented generation (RAG), while also establishing foundational capabilities like embeddings and observability frameworks.

What You’ll Do

  • Platform Development: Design and maintain scalable ML pipelines and platforms for ingestion, feature engineering, training, evaluation, inference, and deployment.
  • Big Data & Analytics: Build and optimize large-scale data workflows using distributed systems (Spark, Hadoop, Kafka, Snowflake) to support analytics and model training.
  • 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.
  • AI/ML Enablement: Provide reusable components, SDKs, and APIs that empower teams to leverage AI insights and ML models effectively.
  • Automation: Drive CI/CD, workflow orchestration, and infrastructure-as-code practices for ML jobs, ensuring reliability and reproducibility.
  • Collaboration: Partner cross-functionally with Product, Data Science, and Engineering teams to align ML infrastructure with business needs.
  • Innovation: Stay ahead of emerging trends in Generative AI, ML Ops, and Big Data to introduce best practices and next-gen solutions.
  • Impact & Growth Opportunities:
  • 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.

We’d Love for You to Have

  • Four plus years of hands-on experience with petabyte-scale datasets and distributed systems.
  • Data Analytics: 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.

Similar jobs

WPP MediaKuala Lumpur, Malaysia

About WPP Media WPP is the trusted growth partner for the world’s leading brands. With exceptional talent, trusted data and intelligence, and world-class partnerships – all united by our pioneering agentic marketing plat…

Full-time
CoupangMountain View, California, United States

Est. 175,000 USD

Company Introduction We exist to wow our customers. We know we’re doing the right thing when we hear our customers say, “How did I ever live without Coupang?” Born out of an obsession to make shopping, eating, and living…

Full-time
AI Engineer6 months ago

Robots & Pencils is seeking an AI Engineer to design, build, and deploy production-grade AI systems that deliver measurable business value. This is a hands-on engineering role focused on implementation and delivery.…

Full-timeRemote
Lead AI Engineer2 months ago
WPP MediaManila, Philippines

About WPP Media WPP is the trusted growth partner for the world’s leading brands. With exceptional talent, trusted data and intelligence, and world-class partnerships – all united by our pioneering agentic marketing plat…

Full-time
BottomlineIndia

Why Choose Bottomline? Are you ready to transform the way businesses pay and get paid? Bottomline is a global leader in business payments and cash management, with over 35 years of experience and moving more than $16 tri…

Full-time

Est. 250,000 USD

We’re looking for a Principal Architect to lead the design and delivery of complex, multi-domain systems spanning cloud, data, and AI. This role is ideal for a deeply experienced engineer who owns the hardest architectur…

Full-timeRemote

Est. 250,000 USD

We’re looking for a Principal Architect to lead the design and delivery of complex, multi-domain systems spanning cloud, data, and AI. This role is ideal for a deeply experienced engineer who owns the hardest architectur…

Full-timeRemote
CongaBoston, Massachusetts, United States

Est. 140,000 USD

A career that’s the whole package! At Conga, we’ve built a community where our colleagues can thrive. Here you’ll find opportunities to innovate and support growth through individual and team development, all within an e…

Full-time
AgodaBangkok, Thailand

About Agoda At Agoda, we bridge the world through travel. Our story began in 2005, when two lifelong friends and entrepreneurs, driven by their passion for travel, launched Agoda to make it easier for everyone to explore…

Full-time
NebiusRemote

Est. 140,000 USD

About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to…

Full-timeRemote
VMLLondon, England, United Kingdom

Est. 80,000 GBP

Who We Are VML is a leading creative company that combines brand experience, customer experience, and commerce, creating connected brands to drive growth. VML is celebrated for its innovative and award-winning human-firs…

Full-time
SRS AcquiomRemote

Est. 152,500 USD

About SRS Acquiom SRS Acquiom delivers the smartest way to run a deal™ through a platform and services designed to help deal parties manage complex M&A and loan agency transactions more efficiently. Based in Denver,…

Full-timeRemote
Zeta GlobalRemote

Est. 165,000 USD

WHO WE ARE Zeta Global (NYSE: ZETA) is the AI-Powered Marketing Cloud that leverages advanced artificial intelligence (AI) and trillions of consumer signals to make it easier for marketers to acquire, grow, and retain cu…

Full-timeRemote
West MonroeChicago, Illinois, United States

Est. 144,000 USD

Are you ready to make an impact?West Monroe is seeking a Senior Software Engineer to join the Tech Ex AI Assets Foundry Team, working with the internal Labs product team to develop proprietary software assets and AI capa…

Full-time
AgodaBangkok, Thailand

About Agoda At Agoda, we bridge the world through travel. Our story began in 2005, when two lifelong friends and entrepreneurs, driven by their passion for travel, launched Agoda to make it easier for everyone to explore…

Full-time
Cloud Architect15 days ago
Robots and PencilsHouston, Texas, United States

Est. 141,000 USD

Cloud Architect Location: Houston, TX This is a 4 month assignment with potential to extend Houston, local OR be able to travel to Houston, TX to be at customer site every other week for a 4-5days/week Company Overview R…

Full-time
West MonroeChicago, Illinois, United States

Est. 124,000 USD

Are you ready to make an impact?West Monroe is seeking a Software Engineer to join the AI Assets Foundry Team, working with the internal Labs product team to develop proprietary software assets and AI capabilities that h…

Full-time
AxleRemote

Est. 140,000 USD

(ID: 2026-3432) Axle is a bioscience and information technology company that offers advancements in translational research, biomedical informatics, and data science applications to research centers and healthcare organiz…

Full-timeRemote

APPLICATIONS FROM OUTSIDE COLOMBIA WILL NOT BE CONSIDERED FOR THIS ROLERobots & Pencils is an applied AI engineering firm building the next frontier of business architecture. We design and ship AI co-workers that int…

Full-timeRemote
NICERichardson, Texas, United States

Est. 105,000 USD

At NiCE, we don’t limit our challenges. We challenge our limits. Always. We’re ambitious. We’re game changers. And we play to win. We set the highest standards and execute beyond them. And if you’re like us, we can offer…

Full-time

Est. 120,000 USD

The Opportunity We're looking for a Product Manager who is passionate about understanding customer problems, validating opportunities, and leading cross-functional teams to deliver measurable business outcomes. You'll ow…

Full-timeRemote

Est. 125,000 USD

The Opportunity We're looking for a Product Manager who is passionate about understanding customer problems, validating opportunities, and leading cross-functional teams to deliver measurable business outcomes. You'll ow…

Full-timeRemote
PayoneerHerzliya, Tel Aviv District, Israel

About Payoneer Founded in 2005, Payoneer is the global financial platform that removes friction from doing business across borders, with a mission to connect the world’s underserved businesses to a rising global economy.…

Full-time
Product Manager29 days ago

Est. 125,000 USD

The Opportunity We're looking for a Product Manager who is passionate about understanding customer problems, validating opportunities, and leading cross-functional teams to deliver measurable business outcomes. You'll ow…

Full-timeRemote
NICERichardson, Texas, United States

Est. 95,000 USD

At NiCE, we don’t limit our challenges. We challenge our limits. Always. We’re ambitious. We’re game changers. And we play to win. We set the highest standards and execute beyond them. And if you’re like us, we can offer…

Full-time
West MonroeChicago, Illinois, United States

Est. 125,000 USD

Are you ready to make an impact?The Tech Ex Labs Senior Product Owner will define and deliver the roadmap for the firm's portfolio of proprietary software assets and AI capabilities. This role partners with business lead…

Full-time
NebiusNew York City, New York, United States

Est. 155,000 USD

About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to…

Full-time
Coupang InternalShanghai, China

Please complete the attached Internal Transfer Request Form and submit. Please make sure to apply with your Coupang e-mail address. Company Introduction We exist to wow our customers. We know we’re doing the right thing…

Full-time
NebiusIsrael

About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to…

Full-time
Alarm.comKraków, Lesser Poland Voivodeship, Poland

Est. 120,000 PLN

EBS is a rapidly expanding, entrepreneurial technology company and part of Alarm.com. Alarm.com is the leading cloud-based platform for smart security and the Internet of Things. More than 7.6+ million home and business…

Full-time