Nebius

Senior Machine Learning Engineer, Model Training and Reinforcement Learning

San Francisco, CA, US$195,200-$262,200Posted 1 month ago

Job Description

About Nebius

Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity of building large in-house AI/ML infrastructure.

Built by engineers, for engineers. From large-scale GPU orchestration to inference optimization, we own the hard problems across compute, storage, networking and applied AI.

Listed on Nasdaq (NBIS) and headquartered in Amsterdam, we have a global footprint with R&D hubs across Europe, the UK, North America and Israel. Our team of 1,500+ includes hundreds of engineers with deep expertise across hardware, software and AI R&D.

### The role

Nebius Token Factory is building an AI training and model post-training capability for frontier model improvement. This role owns the infrastructure that makes large-scale training and RL experiments possible, reliable, reproducible, and efficient. The work sits at the intersection of distributed systems, GPU performance, model training frameworks, RL pipelines, and production engineering.

A Senior Machine Learning Engineer owns substantial ML work end to end. They can translate an ambiguous capability goal into concrete experiments, implement and debug training and RL recipes, build the supporting data and systems, and deliver measurable improvements in model quality, experiment throughput, and reliability. They are deeply hands-on and can independently debug both model-behavior failures and distributed training failures.

Your responsibilities

:

  • Design and run model-training and post-training experiments, including SFT, continued pretraining, preference optimization (DPO/IPO/KTO), and RL methods such as RLHF/RLAIF, PPO, and GRPO.
  • Build reward functions, judge models, verifiers, task environments, and evaluation sets for reasoning, coding, tool use, and

Apply for this role

Keep looking

Similar Remote Jobs That Pay Well