Python Engineer - AI/GPU Backend for FinTech

Revolut; Sweden; On-site; SEK 850,000 - 1,300,000; Full time

Benefits offered by this job

Remote work

Job summary

Revolut is seeking a Python Engineer to join the AI team and build the core infrastructure behind our central platform. You will craft scalable backend services and tools, enabling efficient GPU‑based model training and inference for billion‑user scale systems.

You’ll work with PyTorch, CUDA, NCCL, and Kubernetes to optimise performance, memory, and throughput across multi‑GPU environments while maintaining high reliability and observability in production.

Qualifications

• Proven track record designing and operating scalable backend systems in production (Linux, Kubernetes, containerised workloads).
• Experience running and optimising ML workloads (PyTorch or equivalent) on NVIDIA GPUs.
• Solid understanding of GPU execution concepts: memory hierarchy, CPU/GPU synchronisation, host-to-device transfers, CUDA stream behaviour.
• Familiarity with multi-GPU communication (NCCL), data/tensor parallelism, and Kubernetes resource allocation.
• Degree in a STEM subject with CS fundamentals, testing, observability, and ownership.

Responsibilities

• Designing, building, and maintaining scalable backend services and tooling that support the AI lifecycle and GPU-based workloads
• Profiling and optimising model training and inference workloads (latency, throughput, stability) using PyTorch and CUDA-enabled libraries
• Improving GPU compute utilisation, memory consumption, and bandwidth across shared, multi-tenant, and multi-node environments
• Implementing mixed-precision execution, quantisation, efficient batching, gradient accumulation, and memory-efficient attention
• Creating repeatable benchmarking practices, automated regression tests, and guiding teams on frameworks and GPU types

Skills

Python

Linux

Kubernetes

PyTorch

GPU programming

CUDA

System design

ML workloads

Education

STEM degree

Tools

PyTorch

CUDA

NCCL

NVIDIA CUDA

Job description

Revolut is seeking a Python Engineer to join the AI team and build the core infrastructure behind our central platform. You will craft scalable backend services and tools, enabling efficient GPU‑based model training and inference for billion‑user scale systems.

You’ll work with PyTorch, CUDA, NCCL, and Kubernetes to optimise performance, memory, and throughput across multi‑GPU environments while maintaining high reliability and observability in production.