Staff Machine Learning Engineer

Rokt is a hyper-growth ecommerce leader that enables companies to unlock value by making each transaction relevant at the moment that matters most, when customers are buying.
$235,000 - $300,000
Machine Learning
Staff Software Engineer
Hybrid
501 - 1,000 Employees
5+ years of experience
AI · E-Commerce

Description For Staff Machine Learning Engineer

We are Rokt, a hyper-growth ecommerce leader. We enable companies to unlock value by making each transaction relevant at the moment that matters most, when customers are buying. Together, Rokt's AI-based relevance Platform and scaled ecommerce network powers billions of transactions. In December 2022, Rokt's valuation increased to $2.4 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.

As a Staff Machine Learning Engineer, you will be working with our engineering and product teams to design, build and productionize proprietary machine learning models to solve different business challenges including smart bidding, lookalike modeling, forecasting, and more. You will collaborate closely with product managers and other engineers to understand business priorities, frame machine learning problems, and architect machine learning solutions. You'll build and productionize machine learning models, contribute to the high-quality code base, keep track of emerging tech and trends, and share your knowledge through brown bags and tech talks.

Key responsibilities include:

  • Collaborating with product managers and engineers to understand business priorities and frame ML problems
  • Building and productionizing ML models, including data preparation/processing pipelines
  • Maintaining high-quality code base with comprehensive testing
  • Researching state-of-the-art deep learning models and conducting experiments
  • Sharing knowledge and evangelizing engineering best practices

Requirements:

  • Bachelor's degree in Computer Science or equivalent (PhD in ML/DL is a plus)
  • 5+ years developing production-grade ML systems, 3+ years in software engineering
  • Experience with Bayesian methods, Reinforcement learning, Deep learning, and Recommendation systems
  • Experience with Kubernetes, Kubeflow, TFX, and Feature Store is a plus

Rokt offers a transparent career ladder with clear paths and competitive compensation. Join us in our mission to unlock the full potential in ecommerce and beyond!

Last updated 4 months ago

Responsibilities For Staff Machine Learning Engineer

  • Collaborate with product managers and engineers to understand business priorities and frame ML problems
  • Build and productionize ML models, including data preparation/processing pipelines
  • Contribute to high-quality code base with comprehensive testing
  • Research state-of-the-art deep learning models and conduct experiments
  • Share knowledge through brown bags, tech talks, and evangelizing best practices

Requirements For Staff Machine Learning Engineer

Python
Java
Kubernetes
  • Bachelor's degree in Computer Science or equivalent (PhD in ML/DL is a plus)
  • 5+ years developing production-grade ML systems
  • 3+ years in software engineering with Python, Golang, Java, or similar
  • Experience with Bayesian methods, Reinforcement learning, Deep learning, and Recommendation systems
  • Excel in dynamic, fast-paced settings
  • Experience with Kubernetes, Kubeflow, TFX, and Feature Store is a plus

Benefits For Staff Machine Learning Engineer

401k
Equity
Medical Insurance
Education Budget
  • Equity in the company
  • Annual $5000 training allowance
  • Catered lunch and healthy snacks
  • 4% dollar-for-dollar 401K matching plan
  • Fully funded premium health insurance for entire family
  • Dog-friendly office
  • Extra leave (bonus annual leave, sabbatical leave)
  • Offices in New York, Seattle, Sydney, Tokyo and London
  • Flexible work arrangements

Interested in this job?

Jobs Related To Rokt Staff Machine Learning Engineer

Customer Engineering Manager, AI, Google Cloud (English, Korean)

Lead a team of Customer Engineers at Google Cloud, focusing on AI solutions and cloud transformation for Korean market customers.

Senior Software Engineering Manager, Machine Learning, Labs

Lead machine learning initiatives at Google Labs as a Senior Software Engineering Manager, overseeing technical vision and team development in New York.

Microarchitecture RTL Lead, Silicon AI/ML, TPU, Google Cloud

Lead role in developing cutting-edge TPU technology for AI/ML hardware acceleration at Google, combining technical expertise with team leadership.

Engineering Manager, Gemini Code Assist

Lead the development of GenAI-based developer tools at Google, managing a team working with cutting-edge LLMs and machine learning solutions.

Engineering Manager, Core Machine Learning, Google Cloud

Lead Google Cloud's Core Machine Learning team as Engineering Manager, overseeing development of enterprise-grade AI solutions with C++ and machine learning expertise.