WHY WORK AT NEBIUS

Nebius is leading a new era in cloud computing to serve the global AI economy. We create the tools and resources our customers need to solve real-world challenges and transform industries, without massive infrastructure costs or the need to build large in-house AI/ML teams. Our employees work at the cutting edge of AI cloud infrastructure alongside some of the most experienced and innovative leaders and engineers in the field.

THE ROLE

We are seeking a highly skilled ML/AI Engineer to join our team to lead and support benchmarking of GPU platforms for machine learning and AI workloads. You will play a critical role in evaluating the performance of GPU-based hardware for various deep learning and AI frameworks, enabling data-driven decisions for platform optimisation and next-generation hardware development.

YOUR RESPONSIBILITIES WILL INCLUDE:

• Work closely with hardware, development teams to profile and analyse GPU performance at the system and kernel level.
• Evaluate and compare GPU performance across different platforms, architectures, and software stacks (e.g., CUDA, ROCm).
• Debug and optimise ML workloads to run efficiently on GPU hardware, identifying and resolving performance bottlenecks.
• Perform acceptance testing for new GPU clusters, ensuring hardware and software meet performance, stability, and compatibility requirements for AI workloads.
• Perform experiments across diverse GPU system configurations to assess the impact of varying interconnect strategies and system-level optimisations on performance and scalability.
• Develop tools and dashboards to visualise performance metrics, bottlenecks, and trends.
• Contribute to internal tooling, frameworks, and best practices