jobs

ML Infrastructure Engineer

About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployme

Company

Nebius

Location

Amsterdam, Netherlands; Remote - Europe; Remote - United States

Status

Open

Posted

2026-05-12T12:58:23+00:00

Nebius is hiring an ML/AI engineer to benchmark GPU platforms for machine-learning and AI workloads, comparing architectures and software stacks including CUDA and ROCm. The work covers profiling at kernel and system level, debugging and optimising training and inference workloads, acceptance testing new GPU clusters, studying interconnect effects on scaling, and building dashboards for performance data. The posting asks for deep knowledge of large neural-network training and inference performance, parallelism techniques, PyTorch, JAX or Megatron-LM, CUDA and NCCL, and containers. Profiling tools and vLLM experience help.