Rokt Logo

Rokt

Senior Machine Learning Engineer

Posted 7 Days Ago
In-Office
Austin, TX
210K-400K Annually
Senior level
In-Office
Austin, TX
210K-400K Annually
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

3 Hours Ago
In-Office
290K-470K Annually
Expert/Leader
290K-470K Annually
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
3 Hours Ago
In-Office
290K-470K Annually
Expert/Leader
290K-470K Annually
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
210K-400K Annually
Senior level
210K-400K Annually
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 Boston Tech Scene

Boston is a powerhouse for technology innovation thanks to world-class research universities like MIT and Harvard and a robust pipeline of venture capital investment. Host to the first telephone call and one of the first general-purpose computers ever put into use, Boston is now a hub for biotechnology, robotics and artificial intelligence — though it’s also home to several B2B software giants. So it’s no surprise that the city consistently ranks among the greatest startup ecosystems in the world.

Key Facts About Boston Tech

  • Number of Tech Workers: 269,000; 9.4% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Thermo Fisher Scientific, Toast, Klaviyo, HubSpot, DraftKings
  • Key Industries: Artificial intelligence, biotechnology, robotics, software, aerospace
  • Funding Landscape: $15.7 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Summit Partners, Volition Capital, Bain Capital Ventures, MassVentures, Highland Capital Partners
  • Research Centers and Universities: MIT, Harvard University, Boston College, Tufts University, Boston University, Northeastern University, Smithsonian Astrophysical Observatory, National Bureau of Economic Research, Broad Institute, Lowell Center for Space Science & Technology, National Emerging Infectious Diseases Laboratories

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account