Posted in

Principal Applied Scientist/Machine Learning Engineer

Principal Applied Scientist/Machine Learning Engineer

CompanyRokt
LocationSan Francisco, CA, USA
Salary$315000 – $350000
TypeFull-Time
DegreesMaster’s, PhD
Experience LevelSenior, Expert or higher

Requirements

  • PhD or Master’s in Computer Science, Statistics, Mathematics, or related field with specialization in ML, AI, or Information Retrieval.
  • 10+ years of industry experience building production-grade ML systems.
  • Deep expertise in at least two of the following: Reinforcement Learning (Bandits, Policy Optimization), Multi-Task Learning & Mixture of Experts (MMOE, PLE, DCN), Deep Learning Architectures (Transformers, Graph Neural Networks), Bayesian Methods & Probabilistic Modeling, ML for Ads, E-commerce, or Two-Sided Marketplaces (big plus!).
  • Proven ability to drive impact across multiple teams and organizations.

Responsibilities

  • Develop and implement machine learning models to solve high-impact business problems.
  • Design and optimize scalable ML pipelines and infrastructure for production environments.
  • Conduct applied ML research, prototype new modeling approaches, and rigorously test innovations.
  • Evaluate and improve model performance, ensuring robustness, scalability, and interpretability.
  • Translate complex business needs into practical ML solutions, collaborating with product and engineering teams.
  • Stay ahead of emerging ML trends, contributing to knowledge sharing through tech talks, brown bags, and best practice evangelism.

Preferred Qualifications

  • Familiarity with architectures or experience with models mentioned in the benchmark: DCNV2, MMOE, Deep & Wide, ESMM, xDeepFM, and GDCN.