Senior ML Engineer

Responsibilities

• Develop and optimize low-level kernels and runtime components for AI inference
• Improve performance of inference engines GPU platforms
• Profile and debug system-level and hardware-level performance issues
• Integrate support for new hardware architectures (Hopper, Blackwell, Rubin)
• Collaborate with ML and backend teams to optimize end-to-end execution

Requirements

• Strong proficiency in C++ or expertise in GPU programming
• Focus on low-level high-performance coding and memory management
• Experience in GPU programming or systems-level software development
• Experience with profiling and debugging tools for CPUs and GPUs
• Solid understanding of CPU/GPU architecture and memory hierarchy

Preferred Qualifications

• Experience with GPU computing programming: CUDA, ROCm, CUTLASS, Cute, ThunderKittens, Triton, Pallas, Mosaic GPU
• Familiarity with ML inference runtimes like TensorRT or TVM
• Knowledge of Linux internals, drivers, or compiler toolchains
• Experience with tools like perf, VTune, Nsight, or ROCm profiler
• Familiarity with popular inference engines such as vLLM, sglang, or TGI

Benefits

• Competitive compensation
• Career growth and learning opportunities
• Flexibility and ownership
• Collaborative and innovative culture
• Opportunity to work on impactful AI projects
• International environment and talented teams

About the Company

Nebius is building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment. Headquartered in Amsterdam, Nebius owns the hard problems across compute, storage, networking, and applied AI.