Toloka

ML researcher

In short

Senior ML Researcher role at Toloka AI, a company building data and evaluation infrastructure for GenAI models. The role leads LLM post-training strategy, RL/alignment pipeline architecture (GRPO, PPO/DPO, RLAIF), evaluation systems, and model optimization research, with mentoring and cross-functional leadership duties. Requires 6+ years in ML engineering/applied research including 3+ years leading LLM post-training or alignment at scale, deep expertise in RLHF/RLAIF and distributed training (PyTorch, Deepspeed, Megatron, FSDP, vLLM/TensorRT-LLM), and fluent C1 English. Fully remote position.

About Toloka
Toloka AI creates data that powers leading GenAI models and innovations. It combines experts, crowdsourcing, and a technology platform to teach AI models to reason and evaluate their efficacy and safety.
Responsibilities
•Define the multi-quarter strategy and architecture for Toloka's post-training stack, keeping fine-tuning, RL, and evaluation capabilities ahead of industry trends
•Architect advanced post-training and RL pipelines, including GRPO, PPO/DPO, reward modeling, Process Reward Models, and RLAIF, for platform-wide LLM alignment and agent behavior
•Design and calibrate automated evaluation ecosystems, including human-aligned LLM-as-a-judge frameworks, dynamic benchmarking suites, and regression control setups
•Lead research on model distillation, quantization, context compression, and speculative decoding to optimize latency-cost trade-offs at platform scale
•Architect autonomous guiding agents and tool-use workflows, establishing frameworks for multi-step reasoning, self-correction, and evaluation-driven feedback loops
•Raise the team's engineering and research standards through architectural design reviews, hands-on mentoring of Senior ML Researchers, and reproducible ML R&D best practices
•Partner with Product and Engineering directors to translate client challenges into scalable platform architecture
•Author research write-ups and blog posts, and deliver external tech talks
Requirements
•6+ years in ML engineering or applied research, including 3+ years leading LLM post-training, fine-tuning, or alignment initiatives at scale
•Deep theoretical and practical mastery of LLM alignment: SFT, LoRA/PEFT, RLHF/RLAIF (GRPO, DPO, PPO), reward modeling, and reasoning-oriented post-training
•Proven experience building evaluation pipelines from scratch, addressing LLM-as-a-judge biases, and calibrating automated evaluations against human gold standards
•Strong proficiency in Python, PyTorch, distributed training frameworks (Deepspeed, Megatron, FSDP), and high-performance inference engines (vLLM, TensorRT-LLM)
•Ability to turn ambiguous technical challenges into robust platform features while balancing advanced research, production reliability, and cost constraints
•Proven experience as a technical leader, mentoring senior researchers or engineers and influencing cross-functional roadmaps
•Fluent spoken and written English (C1), with the ability to explain complex technical concepts to internal teams, external clients, and the research community
Conditions
•Competitive compensation package including base salary, bonus, and ESOP
•Paid PTO and benefits vary by location
•IT setup and home office allowances

Stop applying.
Get referred.

$10 in free credits

In just 60 seconds, fully on autopilot — Eraldt's AI agents read your CV, map the roles you fit, find live vacancies and the people who can refer you, write personal emails, and get you referred.

This isn't an application: your CV doesn't go to the company, and we don't promise a referral for this exact role.

Discover who else lands interviews on autopilot