Taro Logo

Lead Machine Learning Engineer / Staff Applied Scientist

Rokt is a hyper-growth ecommerce leader that enables companies to increase value through AI & ML-powered solutions, connecting over 400 million customers to world-leading companies.
Machine Learning
Staff Software Engineer
In-Person
501 - 1,000 Employees
10+ years of experience
AI · E-Commerce

Job Description

Rokt, a $3.5B valued ecommerce technology leader, is seeking a Lead Machine Learning Engineer/Staff Applied Scientist to drive their ML initiatives. This role combines hands-on technical leadership with people management, focusing on developing and deploying sophisticated ML systems that power over 6.5 billion transactions annually. The position offers the opportunity to work with cutting-edge ML technologies, lead a high-caliber team, and contribute to a rapidly growing company that's revolutionizing ecommerce through AI-driven solutions. The role demands deep expertise in machine learning, with a focus on areas like reinforcement learning, multi-task learning, and deep learning architectures. Based in Sydney, this position offers comprehensive benefits including equity, training allowance, and flexible work arrangements. The ideal candidate will bring 10+ years of industry experience and a strong academic background in computer science or related fields.

Last updated 3 months ago

Responsibilities For Lead Machine Learning Engineer / Staff Applied Scientist

  • Own the research, design, development, testing, deployment and maintenance of machine learning systems and services
  • 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
  • Lead and coach a small team of high-calibre ML Engineers
  • Set standards for technical excellence
  • Work cross-functionally to plan and manage delivery expectations
  • Monitor progress and mitigate risks and bottlenecks

Requirements For Lead Machine Learning Engineer / Staff Applied Scientist

Python
  • PhD or Master's in Computer Science, Statistics, Mathematics, or related field with ML specialization
  • 10+ years of industry experience building production-grade ML systems
  • Deep expertise in Reinforcement Learning, Multi-Task Learning, Deep Learning Architectures, or Bayesian Methods
  • Proven ability to drive impact across multiple teams as a hands-on technical leader
  • Experience with models like DCNV2, MMOE, Deep & Wide, ESMM, xDeepFM, and GDCN

Benefits For Lead Machine Learning Engineer / Staff Applied Scientist

Education Budget
Equity
  • Annual $5000 training allowance
  • Company equity for all employees
  • Catered lunch and healthy snacks
  • Gym membership
  • Extra leave including bonus annual leave and sabbatical leave
  • Flexible work arrangements
  • Global office locations

Related Jobs

Machine Learning Engineer - Senior / Principal/ Senior Principal

Senior/Principal Machine Learning Engineer position at Oracle Health AI, focusing on healthcare transformation through advanced AI/ML technologies and LLM implementation.

Senior / Principal Machine Learning Engineer - Oracle Health and AI

Senior/Principal ML Engineer role at Oracle Health and AI, focusing on developing generative AI solutions for healthcare applications.

Staff Deep Learning Engineer

Staff Deep Learning Engineer position at Shield AI, focusing on developing and implementing cutting-edge AI perception capabilities for autonomous aircraft systems.

Staff Engineer

Staff Engineer position at ShiftCare, leading technical excellence and architectural decisions for healthcare SaaS platform using Ruby on Rails. Remote work in Australia/NZ.

Staff Software Engineer - Unpublished R&D Product

Staff Software Engineer position at Riot Games working on an unpublished R&D product, focusing on backend systems and game development in Sydney.