Staff Engineer, Machine Learning

$230,000 - $260,000/Yr

Pubmatic - Redwood City, CA

Full-time - Senior

Redwood City, CA

Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services

About the position

PubMatic is seeking a Sr Principal Machine Learning Engineer to join our team in Redwood City, CA, on a hybrid schedule. This position reports to the SVP of Addressability & Marketplace and is designed for a proven 'doer' who will develop, implement, and extend data-intensive machine learning software for real-time auctioning, ad inventory estimation, and audience segmentations. The successful candidate will work closely with our Big Data, Ad Serving, and Product Managers to apply machine learning techniques to create proofs of concept (POCs) and lead data scientists in implementing these POCs into production and scaling up the solutions. In this role, you will be responsible for designing and implementing core components of our algorithms, modeling the vast amounts of data generated daily by PubMatic. You will develop and implement data-intensive machine learning software for various AdTech applications, ensuring excellence in delivery to both internal and external customers. There is also an opportunity for people leadership within the team, should that interest you. The ideal candidate will have a strong background in machine learning, with a PhD in a STEM field and at least 3 years of hands-on industry experience in designing and building large-scale ML algorithms and ETL processes. You will need to demonstrate solid experience with programming languages such as Python or R, including familiarity with ML libraries and deep learning frameworks. A strong foundation in statistics, probability, and vector calculus is essential, along with advanced SQL or NoSQL skills. Excellent communication skills and a natural curiosity for learning about ML developments are also crucial for success in this role.

Responsibilities

• Design and implement core components of machine learning algorithms.
• Develop and implement data-intensive machine learning software for real-time auctioning, ad inventory estimation, and audience segmentations.
• Collaborate with data scientists, product managers, and software engineers to support new machine learning products.
• Ensure excellence in delivery to internal and external customers.
• Lead a team of data scientists in implementing proofs of concept into production.

Requirements

• PhD in a STEM field required.
• 3+ years of hands-on industry work experience designing and building large-scale ML algorithms and ETL processes.
• 5+ years total analytical work experience, including academic research.
• Solid experience with Python or R, including ML libraries (SKLearn, NumPy, caret, e1071) and parallelization techniques.
• Experience with at least one deep learning framework (TensorFlow, PyTorch, Caffe, Theano, Keras, etc.).
• Strong foundation in graduate statistics and probability, including inference, hypothesis testing, and A/B testing.
• Proficiency in vector calculus and optimization techniques.
• Experience with shallow and deep machine learning algorithms.
• Advanced skills in SQL or NoSQL databases, including complex queries and stored procedures.
• Familiarity with cloud compute engines (AWS, Azure, GCP) and ML on GPU clusters.

Nice-to-haves

• Experience in programmatic advertising and real-time bidding (RTB).
• Knowledge of deep reinforcement learning concepts.
• Proficiency with Spark, Hadoop, Kafka, and Hive.
• Experience with Scala, Java, or C/C++.
• Record of STEM publications in top journals or conferences.
• High rank in Kaggle competitions.

Benefits

• Paid leave programs
• Paid holidays
• Healthcare, dental, and vision insurance
• Disability and life insurance
• Commuter benefits
• Physical and financial wellness programs
• Unlimited DTO in the US
• Reimbursement for mobile expenses
• Fully stocked pantries
• Catered lunches 5 days a week