Senior Machine Learning Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Rokt · 1 day ago

Senior Machine Learning Engineer

Rokt is a hyper-growth ecommerce leader that leverages AI to enhance customer experiences and drive revenue. The Senior Machine Learning Engineer will work with engineering and product teams to design, build, and productionize machine learning models to address various business challenges.

B2BBig DataData IntegrationE-CommerceMarketingMarketing Automation
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Build and productionise machine learning models including data preparation/processing pipelines, machine learning orchestrations, improvements of services performance and reliability and etc
Contribute 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 with load testing, unit testing, integration testing, etc
Collaborate closely with product managers and other engineers to understand business priorities, frame machine learning problems, and architect machine learning solutions
Share your knowledge by giving brown bags, tech talks, and evangelising appropriate tech and engineering best practices

Qualification

Machine LearningAWSTensorFlowProduction-grade systemsBayesian methodsRecommender systemsMulti-task modellingMeta-learningSoft skills

Required

Masters or PhD in Machine Learning
3+ years of industry experience in building production-grade machine learning systems with all aspects of model training, tuning, deploying, serving and monitoring
Good Knowledge in AWS, Kubeflow (or similar), Tensorflow and Feature Store in a production environment
Good 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

Preferred

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 and ESMM

Benefits

Become a shareholder. Every Rokt'star gets equity in the company
Enjoy catered lunch every day and healthy snacks in the office.
Access generous retirement plans like a 4% dollar-for-dollar 401K matching plan and great health benefits for you and your dependents.
Dog-friendly office
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

Company

Rokt is an ecommerce marketing technology that enhances the value of transactions through personalized experiences. It is a sub-organization of Rokt.

H1B Sponsorship

Rokt has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (31)
2024 (20)
2023 (13)
2022 (12)
2021 (10)
2020 (1)

Funding

Current Stage
Late Stage
Total Funding
$822M
Key Investors
Tiger Global ManagementTDM Growth PartnersMA Growth Ventures
2025-01-16Secondary Market· $335M
2022-12-05Secondary Market
2021-12-16Series E· $325M

Leadership Team

leader-logo
Bruce Buchanan
CEO
linkedin
leader-logo
Patrick Connorton
General Counsel & Corporate Secretary
linkedin
Company data provided by crunchbase