Software Engineer (ML Platform and Operations)

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 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.

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 Software Engineer for the ML Operations and Platform Team

Total Compensation between: $290,000 - $324,000, which includes a fixed annual salary between $200,000-$225,000 (including superannuation), employee equity plan grant & world class benefits.

As a Software Engineer on the ML Platform and Operations team, you will play a crucial role in improving our machine learning architecture and operational processes. You will be responsible for designing, implementing, and optimizing the machine learning infrastructure that supports all our machine learning initiatives. You will collaborate with data scientists and engineers to support efficient data access, model management, and deployment.

About the Role:

• Collaborate with data scientists to understand model requirements and translate them into production-ready solutions.
• Contribute to the advancement of Rokt’s machine learning infrastructure by working on projects such as the feature store, model serving platform, and model registry, among others.
• Manage and enhance the machine learning pipelines for model training, evaluation, and deployment, aiming to simplify the creation and deployment process while improving cost efficiency and runtime.
• Design and implement initiatives to improve reliability and latency for the real-time serving of features and models.
• Implement best practices for version control, CI/CD, and automated testing of machine learning models.
• Manage cloud resources and services related to machine learning, including data storage and compute instances.
• Stay updated on emerging technologies and tools in the machine learning ecosystem, and recommend improvements to existing systems.
• Document processes, architectures, and workflows to ensure knowledge sharing and team alignment.

Requirements

About You:

• Bachelor’s or Master’s degree in Computer Science, a similar technical field of study, or equivalent practical experience.
• 2+ years of industry experience in software engineering roles, with development experience in programming languages such as Golang, Python, Java, and/or other programming languages.
• Experience with big data frameworks (Apache Spark, Hadoop, Kafka) for processing large datasets.
• Experience with CI/CD tools and version control systems (e.g., Git).
• Familiarity with cloud platforms (AWS, GCP, Azure) and containerization technologies (Docker, Kubernetes).
• Familiarity with machine learning concepts and frameworks (TensorFlow, PyTorch, Scikit-learn) and MLOps tools (MLflow, Kubeflow).
• Familiarity with building low-latency, distributed backend APIs.
• Strong problem-solving skills and the ability to work collaboratively in a team environment.

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:

• Accelerate your career. We offer roadmaps to leadership and an annual $10,000 training allowance
• 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.