logo inner

Senior Machine Learning Engineer

RoktOnsite

Description


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.

The Role


As a Senior 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 productionise proprietary machine learning models to solve different business challenges including smart bidding, lookalike modelling, forecasting, etc.Target total compensation for this role ranges from $300,000 - $325,000 which includes a fixed annual salary between $200,000-$225,000 (including superannuation) and an employee equity plan grant component.

In addition, you will receive world-class employee benefits.

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


  • Masters or PhD in Machine Learning
  • 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 in building production-grade machine learning systems with all aspects of model training, tuning, deploying, serving and monitoring
  • Experience with Kubeflow (or similar), Tensorflow and 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 in this benchmark: DCNV2, MMOE, Deep & Wide, ESMM, xDeepFM, and GDCN.

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:

  • All employees have access to our LevelUp! program, providing an uncapped training allowance and opportunities for coaching, courses, and mentoring to support career growth and development.
  • 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!
  • 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 understand 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.If this sounds like a role you’d enjoy, apply here, and you’ll hear from our recruiting team.

Life at Rokt

Rokt connects the world's biggest brands and e-commerce sites with customers in a buying state of mind. We reach them at the exact moment they're most receptivethe moment of purchaseto introduce them to something new, something exciting. By saying hi the moment it matters most, we create relationships that last a lifetime. We turn people into your people, and customers into repeat customers. Rokt helps advertisers acquire customers at scale. Every month, Rokt opens the door to more than 100 million transactions and matches your messages with the right person to deliver 10x higher engagement than traditional digital advertising. Rokt's platform covers a network of partners consisting of some of the world's most trusted brands. We boost revenue on our partners sites 50x and help them execute marketing efforts during transactions in a more engaging and effective way for customers. Rokt makes powerful connections with consumers at the very moment theyre most receptive to marketing messages and offers, when they've just made an online transaction. Rokt-created links with consumers achieve 100x higher engagement levels than traditional digital advertising. For more information, please visit Rokt.com.
Thrive Here & What We Value1. Missiondriven community of curious explorers2. Bias for action and willingness to take risks3. Value diversity, transparency, and smart humble people who enjoy building a disruptive business together4. Accelerate careers with roadmaps to leadership and an annual $5000 training allowance5. Become a shareholder through equity in the company6. Extra leave (bonus annual leave, sabbatical leave etc.)7. Work with the greatest talent in town8. See the world! We have offices in New York, Seattle, Sydney, Tokyo, and London

Related Sub

This job belongs to these sub. Explore related roles here:
Machine learning jobs
Your tracker settings

We use cookies and similar methods to recognize visitors and remember their preferences. We also use them to measure ad campaign effectiveness, target ads and analyze site traffic. To learn more about these methods, including how to disable them, view our Cookie Policy or Privacy Policy.

By tapping `Accept`, you consent to the use of these methods by us and third parties. You can always change your tracker preferences by visiting our Cookie Policy.

logo innerThatStartupJob
Discover the best startup and their job positions, all in one place.
Copyright © 2025