Senior Machine Learning Engineer

We are Rokt, a hyper-growth ecommerce leader. We enable companies to increase value by unlocking real-time relevancy in the moment that matters most, when customers are buying. Together, our AI and ML-powered Rokt Brain and ecommerce Rokt Network will power more than 6.5 billion transactions connecting 400 million customers across the world’s leading companies. In January 2025, Rokt’s valuation increased to $3.5 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 understand consumers better. Our bespoke platform handles millions of transactions per day. It 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.

At Rokt, we practice transparency in career paths and compensation. At Rokt, we believe in transparency, which is why we have a well-defined career ladder with transparent compensation and clear career paths based on competency and ability. Rokt’stars constantly strive to raise the bar, pushing the envelope of what is possible.

We are looking for a Senior Machine Learning Engineer

Target total compensation ranges from ₹50,00,000 - ₹80,00,000, including a fixed annual salary of ₹35,00,000 - ₹55,00,000, an employee equity plan grant, and world-class benefits. 

Equity grants are issued in good faith, subject to company policies, board approval, and individual eligibility.

As a Senior Machine Learning Engineer, you will apply your expertise in modeling, statistics, and programming to design and implement cutting-edge machine learning models that address key business challenges, including user targeting, dynamic ad content generation, entity embeddings, and recommendation systems. Collaborating closely with engineering, product, and data science teams, you will develop scalable, production-ready solutions, optimize model performance, and ensure integration into real-time systems. This role offers the opportunity to significantly impact business outcomes by leveraging AI to personalize experiences and optimize advertising strategies.

About the role: 

• Work on a variety of Machine Learning projects in the Ads recommendation domain: user segmentation and targeting, user embeddings generation, Gen AI driven Ad content generation.
• Leverage the latest Generative AI techniques to dynamically create engaging Ad content and model features.
• Collaborate closely with product managers and other engineers to understand business priorities, frame machine learning problems, architect machine learning solutions.
• Build and productionize machine learning models including data preparation/processing pipelines, machine learning orchestrations, improvements of services performance and reliability and etc.

• Use the latest generative AI coding tools including ChatGPT and Github Copilot.

• Contribute and maintain the high quality of 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.
• Keep track of emerging tech and trends, research the state-of-art deep learning models, prototype new modeling ideas, and conduct offline and online experiments.
• Mentor other team members, facilitate within/across team workshops and lead the agile development.

Requirements

About you:

• Bachelor’s degree in Computer Science, a similar technical field of study or equivalent practical experience. A PhD degree in Machine Learning or Deep Learning is a plus.
• 3+ years of industry experience in building production-grade machine learning systems with all aspects of model training, tuning, deploying, serving and monitoring. Industry Experience in applied ML in Ads is a plus.
• 2+ years of industry experience in software engineering roles and development experience in Python, Golang, Java, and/or other programming languages.
• Concrete experience working with deep learning frameworks, either TensorFlow or Pytorch.
• Experience working on Generative AI initiatives.
• Strong understanding of software engineering best practices
• Be motivated, self-driven in a fast (we truly mean fast) paced environment with a proven track record demonstrating impact across several teams and/or organizations.
• Ability to communicate and collaborate effectively with business stakeholders and manage tasks in a timely manner.
• At Rokt we encourage autonomy; teams have complete ownership of their systems including building, running and monitoring. As such, you may be required to be on-call and respond to systems alerts should they arise.
• Ideas, opinions, and the ability to share them through respectful proposals, presentations, and team-wide discussions, An eagerness to work and learn in the open and share your learnings with your teammates.

Benefits

About Rokt’stars:

As a mission-driven, hyper-growth community of curious explorers, our ambition is to unlock real-time relevancy in ecommerce and beyond. Our bias for action means we are not afraid to quickly venture into uncharted territories, take risks, or challenge the status quo; in doing so we either win or learn. We work together as one aligned team, never letting egos get in the way of brilliant ideas. We value diversity, transparency, and smart humble people who enjoy building a disruptive business together. We pride ourselves on being a force for good as we make the world better. 

About The Benefits:

We leverage best-in-class technology and market-leading innovation in AI and ML, with all of that being underlined by building and maintaining a fantastic and inclusive culture where people can be their authentic selves, and offering a great list of perks and benefits to go with it:

• All employees have access to our LevelUp! program, providing opportunities for coaching, courses, and training to support career growth and development.
• Become a shareholder. Every Rokt’star gets equity in the company
• Enjoy catered lunch every day and healthy snacks in the office. Plus join the gym on us! 
• 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

We believe we’re better together. We love spending time together and are in the office most days (teams are in the office 4 days per week). We also get that you need to balance your life and your commitments so you have the flexibility to manage your own hours and can spend up to a week of every quarter working from anywhere.

We at Rokt choose to create a company that is as diverse and inclusive as the world we live in by attracting, growing & keeping the best talent. 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.