Senior ML Engineer — High-Performance AI Inference

Nebius; Amsterdam; On-site; EUR 70,000 - 90,000; Full time

Benefits offered by this job

Competitive compensation

Career growth opportunities

Collaborative culture

International environment

Job summary

Nebius is seeking a skilled professional to join our Token Factory team in Amsterdam, focusing on optimizing high-performance inference platforms for AI cloud services.

The role involves identifying bottlenecks for rapid GPU processing and implementing cutting-edge architectures. Ideal candidates will have a solid understanding of machine learning, experience in software engineering (especially Python), and robust GPU profiling skills.

This is an exciting opportunity to contribute to groundbreaking AI projects in a dynamic and collaborative environment.

Qualifications

• Profound understanding of machine learning and transformer architecture.
• Experience profiling GPU workloads using various tools.
• Strong software engineering skills, particularly with Python.
• Communication and leadership abilities.

Responsibilities

• Identify inference bottlenecks and drive production speedups.
• Implement novel architectures for inference engines.
• Design and productionize low-precision training pipelines.

Skills

Machine learning fundamentals

GPU workload profiling

Software engineering in Python

Deep learning frameworks

Performance optimization techniques

Continuous integration and testing

Tools

Nsight

PyTorch Profiler

CUDA

Job description

Nebius is seeking a skilled professional to join our Token Factory team in Amsterdam, focusing on optimizing high-performance inference platforms for AI cloud services.

The role involves identifying bottlenecks for rapid GPU processing and implementing cutting-edge architectures. Ideal candidates will have a solid understanding of machine learning, experience in software engineering (especially Python), and robust GPU profiling skills.

This is an exciting opportunity to contribute to groundbreaking AI projects in a dynamic and collaborative environment.