AI Platform Engineer (Serverless Training and Inference)

Revolut Ltd; Abu Dhabi; On-site; AED 300,000 - 500,000; Full time

Job summary

Revolut Ltd in the UAE is seeking a Senior AI Platform Engineer to build serverless model training and inference platforms on GPU infrastructure. You’ll design self-service tooling and orchestration for scalable ML workloads, enabling data scientists and engineers to move fast.

Ideal candidates have strong Python backend experience, Kubernetes proficiency, and hands-on cloud infra skills, plus exposure to Flyte or Ray for distributed execution.

Qualifications

• Proven Python backend experience at scale.
• Experience with ML platforms for training or inference on GPUs.
• Strong Kubernetes and cloud infra expertise.
• Familiar with Flyte or Ray for distributed execution.

Responsibilities

• Design self-service platforms for model training, distributed training, and inference on shared GPU infrastructure.
• Develop APIs, SDKs, and workflows that automate workload submission, infrastructure provisioning, and execution.
• Build orchestration and distributed execution capabilities using Flyte, Ray, or equivalent technologies.
• Implement workload scheduling, queueing, resource allocation, and autoscaling across Kubernetes-based GPU environments.
• Build reliable workload lifecycle management, including checkpointing, retries, failure recovery, and cancellation.
• Establish monitoring, multi-tenant isolation, and cost visibility to improve platform reliability and GPU utilisation.

Skills

Python backend

Kubernetes

Cloud infrastructure

Distributed systems

Observability

Fault tolerance

Education

STEM degree

Tools

Flyte

Ray

Job description

About Revolut

People deserve more from their money. More visibility, more control, and more freedom. Since 2015, Revolut has been on a mission to deliver just that.

Our powerhouse of products — including spending, saving, investing, exchanging, travelling, and more — help our 80+ million customers get more from their money every day.

As we continue our lightning-fast growth, 2 things are essential to our success: our people and our culture. In recognition of our outstanding employee experience, we've been certified as a Great Place to Work™. So far, we have 15,000+ people working around the world, from our offices and remotely, to help us achieve our mission. And we're looking for more brilliant people. People who love building great products, redefining success, and turning the complexity of a chaotic world into the simplicity of a beautiful solution.

About The Role

Our Technology team builds the systems and experiences that keep Revolut moving. From the infrastructure behind our innovative app to the features used by millions of people around the world, they bring sharp thinking, speed, and a focus on meaningful impact to everything they do.

We’re looking for a Senior AI Platform Engineer to focus on serverless model training and inference.

As part of the AI team that's driving innovation and powering products, you'll be developing the core infrastructure behind our central platform that empowers our data scientists and engineers to solve tomorrow's challenges today.

Up to shape what's next in finance? Let's get in touch.

What You'll Be Doing

• Designing and building self-service platforms for model training, distributed training, and inference on shared GPU infrastructure
• Developing APIs, SDKs, and workflows that automate workload submission, infrastructure provisioning, and execution
• Building orchestration and distributed execution capabilities using Flyte, Ray, or equivalent technologies
• Implementing workload scheduling, queueing, resource allocation, and autoscaling across Kubernetes-based GPU environments
• Building reliable workload lifecycle management, including checkpointing, retries, failure recovery, and cancellation
• Establishing monitoring, multi-tenant isolation, and cost visibility to improve platform reliability and GPU utilisation

What You'll Need

• Impressive Python engineering skills, and a track record of designing, building, and operating scalable backend platforms in production
• Experience building or extending ML platforms for training or inference on GPU infrastructure
• Expertise with Kubernetes, containerised workloads, and cloud infrastructure
• Familiarity with workflow orchestration and distributed execution frameworks, such as Flyte, Ray, or equivalent technologies
• An understanding of distributed training, GPU resource allocation, workload scheduling, and autoscaling
• Experience designing reliable distributed systems with fault tolerance, observability, and operational ownership
• A degree in a STEM subject or equivalent practical experience, with solid computer science fundamentals and software engineering practices

Building a global financial super app isn’t enough. Our Revoluters are a priority, and that’s why in 2021 we launched our inaugural D&I Framework, designed to help us thrive and grow everyday. We're not just doing this because it's the right thing to do. We’re doing it because we know that seeking out diverse talent and creating an inclusive workplace is the way to create exceptional, innovative products and services for our customers. That’s why we encourage applications from people with diverse backgrounds and experiences to join this multicultural, hard-working team.