Software Engineer (ML Platform and Operations)

We are Rokt, a hyper-growth ecommerce leader. We enable companies to unlock value by making each transaction relevant at the moment that matters most, when customers are buying. Together, Rokt's AI-based relevance Platform and scaled ecommerce Network powers billions of transactions. In December 2022, Rokt’s valuation increased to $2.4 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

Compensation: $160,000-$190,000 including superannuation, plus 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.

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.