Senior/Staff Machine Learning Engineer - Remote

Overview

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.

In Short

• LLM team is currently in search of senior and staff-level ML engineers.
• Work on agentic models for software engineering tasks.
• Conduct experiments to train large language models.
• Explore methods of guided generation and search.
• Mine web-scale data for relevant examples.
• Ensure robust infrastructure for running models.

Requirements

• Understanding of theoretical foundations of machine learning.
• Expertise in deep learning for language processing.
• Experience with training large models on multiple nodes.
• Understanding of performance aspects of neural network training.
• Strong software engineering skills, primarily in Python.
• Experience with modern deep learning frameworks like JAX.
• Proficiency in CI/CD, version control, and unit testing.
• Strong communication and leadership abilities.

Benefits

• Competitive salary and comprehensive benefits package.
• Opportunities for professional growth within Nebius.
• Hybrid working arrangements.
• A dynamic and collaborative work environment.