Senior Machine Learning Platform Engineer

Role Overview: As a Senior Machine Learning Platform Engineer at PubMatic, you will be responsible for designing and scaling infrastructure and frameworks to support machine learning development, experimentation, and production across a global ecosystem handling trillions of ad impressions. You will collaborate with ML Engineers, Data Scientists, and Product stakeholders to accelerate experimentation, maximize efficiency, and translate AI solutions from concept to production. This role offers exposure to petabyte-scale datasets, industry-standard ML efficiency tools, and the opportunity to evaluate and adopt emerging AI/ML technologies. Key Responsibilities: - Design and maintain scalable ML pipelines and platforms for ingestion, feature engineering, training, evaluation, inference, and deployment. - 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. - Provide reusable components, SDKs, and APIs to empower teams to leverage AI insights and ML models effectively. - Drive CI/CD, workflow orchestration, and infrastructure-as-code practices for ML jobs to ensure reliability and reproducibility. - Partner cross-functionally with Product, Data Science, and Engineering teams to align ML infrastructure with business needs. - Stay updated on emerging trends in Generative AI, ML Ops, and Big Data to introduce best practices and next-gen solutions. Qualifications: - Bachelor's degree in engineering (CS / IT) or equivalent degree from well-known Institutes / Universities. - 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. - Familiarity with graph-based systems (Neo4j), RAG approaches, or embedding platforms. - Experience with Spark, Hadoop, Kafka, Snowflake, SparkSQL. - Strong expertise in 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. Additional Information: PubMatic employees have returned to offices via a hybrid work schedule (3 days in office and 2 days working remotely) to maximize collaboration, innovation, and productivity. The benefits package includes paternity/maternity leave, healthcare insurance, broadband reimbursement, kitchen with healthy snacks and drinks, catered lunches, and more. (Note: The company information was not specified in the provided job description.),