Senior ML Engineer (Token Factory)

Nebius

15 days ago

Amsterdam, Netherlands +6 more

Senior

Responsibilities

Enhance fine-tuning methodologies for cutting-edge LLMs focusing on model quality and training efficiency.
Research and implement advanced inference optimization techniques.
Re-implement state-of-the-art open-source LLM architectures in JAX.

Profound understanding of theoretical foundations of machine learning and reinforcement learning.
Deep expertise in modern deep learning for language processing and generation.
Substantial experience with training large models on multiple computational nodes.
Understanding of performance aspects of large neural network training.
Strong software engineering skills, primarily in Python.
Deep experience with modern deep learning frameworks, specifically JAX.
Proficiency in contemporary software engineering approaches, including CI/CD and version control.
Strong communication and leadership abilities.

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

Python