Lead Machine Learning Engineer

3 weeks ago


Council of the City of Sydney, Australia Rokt Full time

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.

ML Engineering @ Rokt

Rokt’s engineering team builds best-in-class ecommerce technology. Our bespoke platform handles millions of transactions per day and considers billions of data points, bringing Machine Learning Engineers the opportunity to research, design, build, train, and deploy models onto real-time systems running at internet scale.

Leading a field-topping team of Applied Scientists, you will undertake deep research and experimentation, harnessing a diversity of bleeding-edge ML theories, concepts, and models to improve sophisticated ML-driven systems powering 6.5 billion+ transactions per year.

As a Lead Machine Learning Engineer / Senior Staff Applied Scientist, you bring deep expertise in machine learning and proven track record driving impact as a hands-on technical leader. You will lead a small, high-calibre team of Applied Scientists and collaborate with other engineering leaders to tackle complex projects that increase the relevance of Rokt’s offerings to customers.

You will be a seasoned expert who thrives when solving complex ML problems at scale - leading the cutting-edge ML research and production deployment of high-impact models which advance: recommender systems, conversion optimisation, user experience enhancement, and multi-task learning architectures. Ultimately, your models will help power over 6.5 billion+ targeted recommendations per year.

Responsibilities

Responsibilities - Technical

- Own the research, design, development, testing, deployment and maintenance of machine learning systems and services at Rokt.
- Conduct applied ML research, prototype new modeling approaches, and rigorously test innovations.
- Evaluate and improve model performance, ensuring robustness, scalability, and interpretability.
- Translate complex business needs into practical ML solutions, collaborating with product and engineering teams.
- Stay ahead of emerging ML trends, contributing to knowledge sharing through tech talks, brown bags, and best practice evangelism.

Responsibilities - People Leadership

- Act as technical, project, and people leader to a small team of high-calibre MLEs
- Coach and develop a small team of mid-senior ML Engineers
- Provide leadership on complex technical issues
- Set standards for technical excellence - from coding to architectural best practices
- Work cross-functionally to plan and manage delivery expectations of roadmap items
- Work cross-functionally with engineering and product leadership to plan and manage delivery expectations of roadmap items
- Actively monitor progress and intervene where required to mitigate risks and bottlenecks, managing expectations within and outside the team - e.g., risks, unexpected issues

Requirements

- PhD or Master’s in Computer Science, Statistics, Mathematics, or related field with specialization in ML, AI, or Information Retrieval.
- 10+ years of industry experience building production-grade ML systems.
- Deep expertise in at least two of the following:

- Reinforcement Learning (Bandits, Policy Optimization)
- Multi-Task Learning & Mixture of Experts (MMOE, PLE, DCN)
- Deep Learning Architectures (Transformers, Graph Neural Networks)
- Bayesian Methods & Probabilistic Modeling
- ML for Ads, E-commerce, or Two-Sided Marketplaces (big plus)

- Proven ability to drive impact across multiple teams and organisation as a hands-on technical leader
- 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

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

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

#J-18808-Ljbffr



  • Council of the City of Sydney, Australia Rokt Full time

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


  • Council of the City of Sydney, Australia Singtel Group Full time

    Overview At Optus, we are fuelled by optimism, unwavering customer focus, and a team of amazing people. And that’s where we come in…. “the amazing people” part! We don’t sit back and wait for the future to happen, we are out there crafting our own path through new technology, innovation, and investment. We are truly a challenger brand, with...


  • Council of the City of Sydney, Australia Atlassian Full time

    Atlassians can choose where they work – whether in an office, from home, or a combination of the two. That way, Atlassians have more control over supporting their family, personal goals, and other priorities. We can hire people in any country where we have a legal entity. Interviews and onboarding are conducted virtually, a part of being a...


  • City of Melbourne, Australia black.ai Full time

    Who We Are Nomad Atomics is on a mission to make the broad uptake of quantum sensing a reality and simultaneously push the limits of our field beyond what we think is possible. We are building the world's most advanced fit‑for‑purpose quantum sensors to allow us to see the world like never before. Who You Are You are a voracious learner, a problem...


  • City of Melbourne, Australia Myob Group Limited Full time

    We’re a leading business management solution with a core purpose: helping more businesses in Australia and New Zealand start, survive and succeed. At MYOB, we believe what’s good for one business is good for all business—and for all of us. Whether you support them, work for them, or dream of building your own, when businesses run smoothly, everybody...


  • Council of the City of Sydney, Australia Atlassian Full time

    Principal Machine Learning System Engineer Atlassians can choose where they work – whether in an office, from home, or a combination of the two. That way, Atlassians have more control over supporting their family, personal goals, and other priorities. We can hire people in any country where we have a legal entity. About Central AI Org Our organization...


  • Sydney, New South Wales, Australia ProFound People Full time $120,000 - $180,000 per year

    Machine Learning Engineer | Sydney | Python, Computer Vision, LinuxOne of our top client's is in a rapid period of growth right now, after an outstanding amount of investment and continuous demand for their products.As a Machine Learning Engineer, you can expect to lead end-to-end machine learning solutions, getting involved in the design and deployment of...


  • City of Melbourne, Australia News Corporation Full time

    Senior Engineering Manager - Machine Learning - News Corp + **Make an impact leading exciting squads utilising new AI technologies!** + **Work across multiple leading property brands impacting millions of users!** + **Permanent full time role based in Richmond** With bold and ambitious goals, REA Group...


  • City of Melbourne, Australia REA Group Ltd. Full time

    Staff Machine Learning Engineer Location: Richmond, Melbourne VIC. Type: Full time. Requisition: R0013991 What you’ll do - Design scalable, multi-modal AI solutions (text, image, audio, video) for our web and app platforms. - Utilize Generative AI, LLMs, and VLMs, combining them with traditional ML techniques to deliver impactful solutions. - Build and...


  • Council of the City of Sydney, Australia Westpac Group Full time

    Quality Engineering Lead (Automation & AI) Sydney, NSW, Australia Job Description Create your best future and join Westpac as a Quality Engineering Lead – Automation and AI UNITE is our business-led, tech-enabled transformation, and a key priority for the Group. UNITE is about delivering 'one best way' to make our processes, systems and tech simpler so...