Rokt Logo

Rokt

Senior Machine Learning Engineer

Posted 7 Days Ago
In-Office
Austin, TX
Senior level
In-Office
Austin, TX
Senior level
Design, build, and productionize machine learning models and data pipelines for recommendation, bidding, and forecasting. Collaborate with product and engineering teams, evaluate and monitor model performance, maintain high-quality tested code, prototype research ideas, and run experiments. Hybrid schedule: four days in-office, one remote.
The summary above was generated by AI

Rokt is an ecommerce technology company with the mission of making every transaction more relevant. The Rokt Ecommerce Network leverages proprietary machine learning recommendation systems, powering billions of transactions for hundreds of millions of customers, and is trusted to do this by companies like Live Nation, Fanatics, Macy's, AMC Theatres, PayPal, Uber, Hulu, Staples, Albertsons and HelloFresh.

We are hiring Senior Machine Learning Engineers

We are hiring engineers with significant expertise in both machine learning and software engineering. 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 modelling, forecasting, etc.

Target total compensation ranges from $335k - $400k, comprised of a fixed annual salary of $210k - $260k, plus employee equity plan grant. In addition, you will receive world-class employee benefits.

Responsibilities

  • Collaborate closely with product managers and other engineers to understand business priorities, frame machine learning problems, and architect machine learning solutions for smart bidding, lookalike modelling, forecasting, and related ranking and prediction tasks.
  • Build and productionise machine learning models, including model-specific data pipelines, feature engineering within the team's feature store, and integration with the team's orchestration and serving infrastructure.
  • Evaluate model performance through offline metrics, and monitor deployed models for drift, leading retraining or rollback decisions as needed.
  • Contribute to and maintain the high quality of the code base with tests that provide a high level of functional coverage as well as non-functional aspects such as load testing, unit testing, and integration testing.
  • Keep track of emerging tech and trends, research the state-of-the-art deep learning models, prototype new modelling ideas, and conduct offline and online experiments

Requirements

  • Willingness to work 4 day in-office, 1 day remote weekly schedule.
  • PhD or Master's in Computer Science, Statistics, Mathematics, or related field with specialization in ML, AI, or Information Retrieval (or equivalent experience)
  • Extensive knowledge in and experience with some of the following areas: Bayesian methods, recommender systems, multi-task modelling, meta-learning, click-through rate modelling or conversion rate modelling
  • 3+ years of industry experience building production-grade machine learning systems, spanning model training, tuning, deployment, serving, and monitoring
  • Experience with Kubeflow (or similar), TensorFlow, and a feature store in a production environment is a massive plus
  • Bonus points if you are familiar with any of the following architectures or have experience with the models mentioned: DCNV2, MMOE, Deep & Wide, ESMM, xDeepFM, and GDCN

Benefits

  • Equity in a profitable, fast-growing company approaching $1 Billion in revenue.
  • Dollar-for-dollar 401K matching plan (up to 4% of fixed annual remuneration)
  • Fully funded health insurance (Dental, Optical, and Medical)
  • Generous allowances for wellness, technology, mobile, and transit.
  • Daily catered lunch, stocked pantry & fridges
  • Extra leave (bonus annual leave, sabbatical leave etc.)

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, which will be sent out to your application email. Successful candidates will be contacted to discuss the next steps.

Similar Jobs at Rokt

4 Hours Ago
In-Office
Expert/Leader
Expert/Leader
Artificial Intelligence • Digital Media • eCommerce • Marketing Tech • Software • Automation
Lead applied research and end-to-end development of production ML systems for a high-volume e-commerce marketplace. Prototype and deploy models, optimize auction and decisioning logic, improve performance, robustness, and interpretability, and collaborate with product and engineering. Provide technical leadership for a small specialist team, set engineering standards, manage delivery, and mitigate project risks.
Top Skills: Bayesian ModellingContextual BanditsDcnDistributed TrainingEmbeddingsGpu OptimisationGraph Neural NetworksJit CompilationKnowledge DistillationMixed PrecisionMmoePlePolicy OptimizationProbabilistic MethodsQuantisationRepresentation LearningTransformers
4 Hours Ago
In-Office
Expert/Leader
Expert/Leader
Artificial Intelligence • Digital Media • eCommerce • Marketing Tech • Software • Automation
Lead architecture, design, and delivery of high-throughput, low-latency distributed services. Drive performance optimizations, incident RCA, system design reviews, and adoption of AI/LLM tooling. Provide cross-team technical leadership, mentor engineers, and collaborate with product to manage roadmaps and delivery risks.
Top Skills: AgentsC#CassandraCi/CdCloud PlatformsDockerGoIcebergJavaKafkaKubernetesLlmsObservabilityScalaSparkTrinoTypescript
7 Days Ago
In-Office
Senior level
Senior level
Artificial Intelligence • Digital Media • eCommerce • Marketing Tech • Software • Automation
Design, build, deploy, and operate ultra-high-throughput, scalable backend services. Troubleshoot production incidents, optimize performance and cost, own architecture, mentor engineers, and leverage AI/LLM tooling. Participate in on-call rotation and collaborate cross-functionally to meet business objectives.
Top Skills: Ai ToolingC#CassandraCi/CdCloud PlatformsDockerGoIcebergJavaKafkaKubernetesLlmsScalaSparkTrinoTypescript

What you need to know about the Ottawa Tech Scene

The capital city of Canada and the nation's fourth-largest urban area, Ottawa has proven a rapidly growing global tech hub. With over 1,800 tech companies, many of which are leaders in their sectors, the city's tech talent now makes up more than 13 percent of its total workforce. This growth is driven not only by the big players like UL Solutions and Dropbox, but also by a thriving startup ecosystem, as new businesses emerge to follow in the footsteps of those that came before them.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account