Machine Learning Engineer

Rokt is a hyper-growth ecommerce leader, unlocking real-time relevance in the moments that matter most. Our AI Brain and ecommerce Network power 7.5 billion+ transactions p/y and are trusted by thousands of leading companies globally (Expedia, Best Buy, Lowe’s, Walgreens, Ticketmaster, Venmo, Apple, Disney+, Uber Eats, Capital One) and in Australia (The Iconic, David Jones, Dan Murphy's, Dominos, Foxtel, Amazon, Afterpay, PayPal)

We are a team of builders helping smart businesses innovate, meet customer needs, and generate incremental revenue. Rokt’s economic edge unleashes a world of possibilities for growth and innovation.

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.

We are looking for a Machine Learning Engineer

Target total compensation for this role is $245,000 - $260,000, comprised of a fixed annual salary of $170,000 - $185,000 (including superannuation), and a stake in Rokt through our employer equity grant plan. In addition, you will receive a world-class benefits package. Equity grants are issued in good faith, subject to company policies, board approval, and individual eligibility.

About the role: 

If you're nearing the completion of your PhD in Machine Learning, have recently graduated, or bring equivalent hands-on experience, this is your opportunity to launch a high-impact career in applied Machine Learning.

You would be joining the Rokt Brain Relevancy team in Sydney. Our focus is delivering the most relevant experience for every user in the transaction moment, harnessing advanced ML and real-time data to optimise every offer and interaction to maximise value for customers, partners, and advertisers. We build adaptive and transparent models that continuously learn and improve.

As an ML Engineer, you'll translate cutting-edge academic insights into scalable ML products that shape user experiences across more than 7.5 billion transactions and 400 million customers globally. 

Focus areas for the team include personalisation & relevance models, selection & scoring algorithms, exploration vs. exploitation tradeoffs, cold start & smart bidding models, predictive & agentic models and statistical modelling, & data enrichment.

Requirements

What You’ll Do

• Frame & Solve ML Problems: Partner with colleagues to understand priorities, define machine learning problems, and design solutions that drive business outcomes.
• Build & Operationalize ML Systems: Develop and productionize end-to-end ML pipelines—including data processing, orchestration, modeling, and continuous deployment—ensuring reliability and scalability.
• Research & Experiment: Stay current with emerging ML technologies. Prototype new models and validate them through experimentation, both offline and online (e.g., A/B testing).
• Accelerate Development with AI: Leverage AI-driven tools and automation to implement clean, maintainable and well-tested code - allowing you to focus on creative problem-solving and quality.
• Collaborate & Innovate: Work closely with diverse teams—to build cohesive ML solutions. Share knowledge through talks, brown bags, and mentoring
• Drive Business Impact: Build capabilities that create personalized experiences and unlock millions in incremental revenue for the world’s leading ecommerce brands.

Who You Are

• AI-Driven & Curious: You're an AI enthusiast who stays ahead of the curve—whether through a recent PhD in Machine Learning or hands-on experience with advanced architectures like MOEs, MMOE, or xDeepFM
• ML Domain Expertise: You have deep knowledge in one or more of: deep learning, Bayesian methods, reinforcement learning, econometrics, NLP or gradient boosting techniques.
• Technically Versatile: You bring a solid grasp of software engineering, system design, and cloud architecture, with bonus points for production experience in Kubernetes, TFX, Kubeflow, or Feature Stores.
• Creative Problem Solver: You break down complex challenges using first principles thinking, creativity, and a strong analytical foundation across both ML and engineering domains.
• Entrepreneurial Mindset: You take ownership of outcomes, move fast with confidence, and aren’t afraid to navigate ambiguity—figuring things out independently when needed.
• Collaborative Team Player: You communicate clearly and work well with cross-functional teams. You value feedback, share ideas openly, and help others succeed.
• Driven & Results-Oriented: You set the bar high, consistently deliver quality, and care deeply about the real-world impact of your work—on both customers and business outcomes.

Benefits

Why Join Rokt

• Build the Future of AI in Ecommerce: Be at the forefront of AI-driven transformation in a company that’s pioneering how brands engage customers in the “moment that matters” during online transactions.
• Hyper-Growth = Fast Progression: Rokt is a rapidly growing tech leader, which means huge opportunities for your career advancement, learning, and taking on bigger responsibilities quickly.
• Culture of Builders: Work with a smart, humble, and bold team that shares a “builder” DNA – we love to innovate, take risks, and turn ambitious ideas into tangible results. We win as a team and learn from every experiment.
• Ownership & Impact: Every Rokt’star (employee) has a voice and real equity in the company. You’ll have autonomy to make decisions, drive projects, and see the direct impact of your work on millions of users.
• World-Class Benefits & Support: Join a people-first culture with transparent career paths, continuous development (LevelUp training, mentorship), and great perks (equity grants, catered lunches, global offices, and more) that empower you to do your best work.

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! 
• 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 minimum 4 days per week). 

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.

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