Principal Machine Learning Engineer - Relevancy

Rokt; New York (NY); On-site; USD 315,000 - 350,000; Full time

Job summary

Rokt is seeking a Principal Machine Learning Engineer with expertise in machine learning and software engineering. The ideal candidate will collaborate across teams to design and build ML models that solve business challenges. This high-level role offers a competitive salary and significant total compensation in a rapidly growing company dedicated to transparency and inclusivity.

Benefits

Equity in the company

Catered lunch every day

Generous retirement plans

Fully funded health insurance for family

Dog-friendly office

Extra leave

Flexible hours

Qualifications

• 10+ years of experience in building production-grade machine learning systems.
• Deep knowledge in AWS, Kubeflow, Tensorflow.
• Experience with Bayesian methods, recommender systems, multi-task modelling.

Responsibilities

• Collaborate closely with product managers and engineers to architect ML solutions.
• Build and productionise machine learning models with data processing pipelines.
• Research and prototype new deep learning models.

Skills

Machine Learning

Software Engineering

AWS

Kubeflow

Tensorflow

Education

Masters or PhD in Machine Learning

Job description

2 days ago Be among the first 25 applicants

Get AI-powered advice on this job and more exclusive features.

This range is provided by Rokt. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$315,000.00/yr - $350,000.00/yr

We are Rokt, a hyper-growth ecommerce leader. We enable companies to increase value by unlocking real-time relevancy in the moment that matters most, when customers are buying. Together, our AI and ML-powered Rokt Brain and ecommerce Rokt Network will power more than 6.5 billion transactions connecting 400 million customers across the world's leading companies. In January 2025, Rokt's valuation increased to $3.5 billion USD, allowing us to expand rapidly across 15 countries.

The Rokt engineering team builds best-in-class ecommerce technology that provides personalized and relevant experiences for customers globally and empowers marketers with sophisticated, AI-driven tooling to better understand consumers. Our bespoke platform handles millions of transactions per day and considers billions of data points which give engineers the opportunity to build technology at scale, collaborate across teams and gain exposure to a wide range of technology.

At Rokt, we practice transparency in career paths and compensation. At Rokt, we believe in transparency, which is why we have a well-defined career ladder with transparent compensation and clear career paths based on competency and ability. Rokt'stars constantly strive to raise the bar, pushing the envelope of what is possible.

We are looking for a Principal ML Engineer - Relevancy

Total Compensation between: $515,000 - $750,000, which includes a fixed annual salary between $315,000-$350,000, employee equity plan grant & world class benefits.

As a Principal Machine Learning Engineer, you are someone who has significant expertise in both machine learning and software engineering. You will be working with our engineering and product teams to design, build and production-ise proprietary machine learning models to solve different business challenges including smart bidding, budget pacing, lookalike modeling, and more.

About The Role

• Collaborate closely with product managers and other engineers to understand business priorities, frame machine learning problems, and architect machine learning solutions.
• Build and productionise machine learning models including data preparation/processing pipelines, machine learning orchestrations, improvements of services performance and reliability and etc.
• Contribute 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 with load testing, unit testing, integration testing, etc.
• Keep track of emerging tech and trends, research the state-of-art deep learning models, prototype new modelling ideas, and conduct offline and online experiments.
• Share your knowledge by giving brown bags, tech talks, and evangelising appropriate tech and engineering best practices.
  
  

Requirements

About you:

• Masters or PhD in Machine Learning
• 10+ years of industry experience in building production-grade machine learning systems with all aspects of model training, tuning, deploying, serving and monitoring
• Deep Knowledge in AWS, Kubeflow (or similar), Tensorflow and Feature Store in a production environment.
• Deep 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
• Bonus points if you are familiar with any of the following architectures or have experience with the models mentioned in this benchmark: DCNV2, MMOE, Deep & Wide and ESMM.
  
  

Benefits

About Rokt'stars:

As a mission-driven, hyper-growth community of curious explorers, our ambition is to unlock real-time relevancy in ecommerce and beyond. Our bias for action means we are not afraid to quickly venture into uncharted territories, take risks or challenge the status quo; in doing so we either win or learn. We work together as one aligned team never letting egos get in the way of brilliant ideas. We value diversity, transparency and smart humble people who enjoy building a disruptive business together. We pride ourselves on being a force for good as we make the world better.

About The Benefits

We leverage best-in-class technology and market-leading innovation in AI and ML, with all of that being underlined by building and maintaining a fantastic and inclusive culture where people can be their authentic selves, and offering a great list of perks and benefits to go with it:

• Become a shareholder. Every Rokt'star gets equity in the company
• Enjoy catered lunch every day and healthy snacks in the office. Plus join the gym on us!
• Access generous retirement plans like a 4% dollar-for-dollar 401K matching plan and get fully funded premium health insurance for your entire family!
• Dog-friendly office
• Extra leave (bonus annual leave, sabbatical leave etc.)
• Work with the greatest talent in town
• See the world! We have offices in New York, Seattle, Sydney, Tokyo and London
  
  

We believe we're better together. We love spending time together and are in the office most days (teams are in the office 4 days per week). We also get that you need to balance your life and your commitments so you have the flexibility to manage your own hours and can spend up to a week of every quarter working from anywhere.

We at Rokt choose to create a company that is as diverse and inclusive as the world we live in by attracting, growing & keeping the best talent. 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.

Seniority level

• Seniority level

  Director

Employment type

• Employment type

  Full-time

Job function

• Industries

  Advertising Services, Technology, Information and Internet, and Software Development

Referrals increase your chances of interviewing at Rokt by 2x

Get notified about new Machine Learning Engineer jobs in New York, NY.

Software Engineer, Machine Learning (Multiple Levels) - Slack

New York, NY $167,300 - $334,600 4 days ago

New York, NY $100,000 - $250,000 2 weeks ago

Machine Learning Engineer I - Windreich Department of Artificial Intelligence & Human Health

New York, NY $109,000 - $163,695 1 day ago

New York, NY $120,000 - $160,000 1 week ago

New York, NY $180,000 - $290,000 2 months ago

New York, NY $180,000 - $210,000 2 days ago

Machine Learning Engineer, Personalization

New York, NY
$136,878.00
-
$195,540.00
2 days ago

New York, NY
$160,000.00
-
$280,000.00
1 year ago

New York, NY
$141,000.00
-
$202,000.00
1 week ago

New York, NY
$187,000.00
-
$270,000.00
2 weeks ago

New York, NY
$140,000.00
-
$200,000.00
8 months ago

New York, NY
$220,000.00
-
$260,000.00
1 day ago

New York, NY
$140,000.00
-
$200,000.00
8 months ago

New York, NY
$150,000.00
-
$200,000.00
1 month ago

Machine Learning Engineer I - Windreich Department of Artificial Intelligence & Human Health

New York, NY
$109,000.00
-
$163,695.00
1 day ago

New York, NY
$180,000.00
-
$240,000.00
2 months ago

Machine Learning Engineer, GenRecs, Personalization

New York, NY
$138,250.00
-
$197,500.00
2 days ago

Machine Learning Engineer, GenAI Applied ML

New York, NY
$176,000.00
-
$220,000.00
2 weeks ago

New York, NY
$175,000.00
-
$265,000.00
3 weeks ago

New York, NY
$110,000.00
-
$250,000.00
2 months ago

New York, NY
$164,000.00
-
$198,000.00
2 weeks ago

New York, NY
$140,000.00
-
$140,000.00
2 weeks ago

Senior Software Engineer I, Machine Learning

Brooklyn, NY
$183,000.00
-
$216,000.00
1 week ago

New York, NY
$128,000.00
-
$160,000.00
2 days ago

New York, NY $190,000 - $285,000 3 days ago

Software Engineer II | Full Stack Developer

New York, NY $130,000 - $180,000 5 months ago

New York, NY $132,000 - $154,000 2 weeks ago

Staff Applied Scientist I, Machine Learning Recommendations

Brooklyn, NY $218,000 - $257,000 2 weeks ago

New York, NY $148,100 - $244,400 3 days ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.