ML Engineer

The role is for Nebius AI R&D, a team focused on applied research in AI. We are looking for senior- and staff-level ML engineers to work on research in areas such as Guided search and reinforcement learning for agentic systems, RL for reasoning models, web-scale data collection for training agents, and efficient model distillation. Examples of responsibilities include conducting experiments to train a large language model on traces of interactions with various environments, exploring methods of guided generation and search in the trajectory space, mining relevant data at web scale for model post-training, conducting RL experiments with different configurations in verifiable domains, and training AI agents on tasks with non-verifiable reward signals. We expect you to have a profound understanding of ML and RL theory, deep expertise in modern deep learning for language processing and generation, substantial experience with training large models on multiple computational nodes, strong Python software engineering skills, deep experience with JAX, strong communication and leadership abilities, and ability to document research findings clearly and contribute to technical publications. Nice to have includes experience with deep RL for LLMs, RoPE, ZeRO/FSDP, and a Master/PhD; a track record of building products in start-up-like environments; experience with distributed systems; open-source projects; and excellent English. Benefits & Perks: 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. Nebius is an equal opportunity employer and committed to a diverse, inclusive workplace. Apply to join Nebius and shape the future of AI.