Senior Machine Learning Engineer

Job Description

Job Description

Rokt is an ecommerce technology company with the mission of making every transaction more relevant. The Rokt Ecommerce Network leverages proprietary machine learning recommendation systems, powering billions of transactions for hundreds of millions of customers, and is trusted to do this by companies like Live Nation, Fanatics, Macy's, AMC Theatres, PayPal, Uber, Hulu, Staples, Albertsons and HelloFresh.

We are hiring Senior Machine Learning Engineers

We are hiring engineers with significant expertise in both machine learning and software engineering. You will be working with our engineering and product teams to design, build and productionize proprietary machine learning models to solve different business challenges including smart bidding, lookalike modelling, forecasting, etc.

Target total compensation ranges from $335k - $400k, comprised of a fixed annual salary of $210k - $260k, plus employee equity plan grant. In addition, you will receive world-class employee benefits.

Responsibilities

• Collaborate closely with product managers and other engineers to understand business priorities, frame machine learning problems, and architect machine learning solutions for smart bidding, lookalike modelling, forecasting, and related ranking and prediction tasks.

• Build and productionise machine learning models, including model-specific data pipelines, feature engineering within the team's feature store, and integration with the team's orchestration and serving infrastructure.
• Evaluate model performance through offline metrics, and monitor deployed models for drift, leading retraining or rollback decisions as needed.
• Contribute to and maintain the high quality of the code base with tests that provide a high level of functional coverage as well as non-functional aspects such as load testing, unit testing, and integration testing.
• Keep track of emerging tech and trends, research the state-of-the-art deep learning models, prototype new modelling ideas, and conduct offline and online experiments

Requirements

• Willingness to work 4 day in-office, 1 day remote weekly schedule.

• PhD or Master's in Computer Science, Statistics, Mathematics, or related field with specialization in ML, AI, or Information Retrieval (or equivalent experience)
• Extensive knowledge in and experience with some of the following areas: Bayesian methods, recommender systems, multi-task modelling, meta-learning, click-through rate modelling or conversion rate modelling
• 3+ years of industry experience building production-grade machine learning systems, spanning model training, tuning, deployment, serving, and monitoring
• Experience with Kubeflow (or similar), TensorFlow, and a feature store in a production environment is a massive plus
• Bonus points if you are familiar with any of the following architectures or have experience with the models mentioned: DCNV2, MMOE, Deep & Wide, ESMM, xDeepFM, and GDCN

Benefits

• Equity in a profitable, fast-growing company approaching $1 Billion in revenue.
• Dollar-for-dollar 401K matching plan (up to 4% of fixed annual remuneration)
• Fully funded health insurance (Dental, Optical, and Medical)
• Generous allowances for wellness, technology, mobile, and transit.
• Daily catered lunch, stocked pantry & fridges
• Extra leave (bonus annual leave, sabbatical leave etc.)

Equal employment opportunities are available to all applicants without regard to race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.

If this sounds like a role you’d enjoy, apply here, and you’ll hear from our recruiting team.

Note: The first stage of the recruitment process for this role is to complete a 15-minute online aptitude test, which will be sent out to your application email. Successful candidates will be contacted to discuss the next steps.