Machine Learning Engineer

We are Rokt, a hyper-growth ecommerce leader. Rokt is the global leader in ecommerce, unlocking real-time relevance in the moment that matters most. Rokt’s AI Brain and ecommerce Network powers billions of transactions connecting hundreds of millions of customers, and is trusted to do this by the world’s leading companies.

We are a team of builders helping smart businesses find innovative ways to meet customer needs and generate incremental revenue. Leading companies drive 10-50% of additional revenue—and often all their profits—from the extra products or services they sell. This economic edge unleashes a world of possibilities for growth and innovation.

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 Machine Learning Engineer

A fixed annual salary of $130,000 - $185,000 (including superannuation,) an employee equity plan grant, and world-class benefits. 

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 AI. As an ML Engineer on our Research & Modelling team, you'll translate cutting-edge academic insights into scalable ML products that shape user experiences across more than 6 billion transactions and 400 million customers globally. 

Working alongside world-class engineers and product teams, you’ll design, train, and productionize proprietary models that solve real business challenges—ranging from click-through rate prediction and recommendation systems to creative content generation using large language models (LLMs).

If you're driven by bold ideas, inspired by solving hard problems, and excited to push the boundaries of what's possible with machine learning, we’d love to hear from you.

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 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.

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.