Senior ML Engineer (AI Research)

Nebius

3 months ago

Amsterdam, Netherlands +2 more

Senior / Staff+

Responsibilities

Conduct experiments to efficiently train large language models on interaction traces.
Explore guided generation and search methods in trajectory space.
Mine relevant data at web scale for model post-training.
Experiment with various reinforcement learning configurations in verifiable domains.
Train AI agents on tasks with non-verifiable reward signals.

Profound understanding of machine learning and reinforcement learning foundations.
Deep expertise in modern deep learning for language processing and generation.
Substantial experience training large models on multiple computational nodes.
Strong software engineering skills, primarily in Python.
Deep experience with modern deep learning frameworks, particularly JAX.
Strong communication and leadership abilities.
Experience designing and analyzing machine learning experiments with statistical rigor.
Ability to formulate research questions and document findings clearly.

Competitive salary and comprehensive benefits package.
Opportunities for professional growth within Nebius.
Flexible working arrangements.
Dynamic and collaborative work environment that values initiative and innovation.

Python