Machine Learning Engineer • Senior

Who you are

You have strong proficiency in C++ and expertise in GPU programming, particularly focused on low-level optimizations for AI inference. Your experience includes developing and optimizing kernels and runtime components, ensuring high performance across various GPU platforms. You are skilled at profiling and debugging system-level and hardware-level performance issues, and you have a keen understanding of integrating support for new hardware architectures like Hopper, Blackwell, and Rubin.

You thrive in collaborative environments, working closely with ML and backend teams to optimize end-to-end execution. Your background includes a solid understanding of AI and machine learning principles, and you are passionate about pushing the boundaries of AI cloud infrastructure. You are excited about the opportunity to contribute to a cutting-edge team that is transforming industries through innovative AI solutions.

What you'll do

As a Senior ML Engineer at Nebius AI, you will be responsible for developing and optimizing low-level kernels and runtime components for our AI inference platform. You will work on improving the performance of inference engines across our extensive GPU platforms, ensuring that our solutions are fast, reliable, and scalable. Your role will involve profiling and debugging system-level and hardware-level performance issues, allowing you to identify and implement optimizations that enhance overall system efficiency.

You will integrate support for new hardware architectures, collaborating with cross-functional teams to ensure seamless execution of AI models. Your contributions will directly impact the deployment of various foundation models, including text, vision, audio, and emerging multimodal architectures. You will be part of a dynamic team that values initiative and innovation, and you will have opportunities for professional growth within the organization.

What we offer

At Nebius AI, we provide a competitive salary and a comprehensive benefits package, along with flexible working arrangements. You will be part of a collaborative work environment that encourages innovation and values your contributions. As we continue to grow and expand our products, you will have the chance to work at the forefront of AI cloud infrastructure, making a real impact in the industry. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds.