Senior ML Researcher/Engineer — разработка и скейлинг World Models@ Яндекс

О позиции

В команду робота доставки Яндекса требуется Senior ML Engineer/Researcher, который будет заниматься разработкой быстрой интерактивной модели мира и масштабным обучением MBRL-агентов. Роботы доставки Яндекса ежедневно доставляют тысячи заказов, маневрируя в сложной городской среде. Мы активно растём и планируем масштабировать флот до 20 000 роботов к 2028 году.

Чем вы будете заниматься

• Разработка и скейлинг World Models: проектирование и обучение массивных 3D/видеотокенизаторов и бэкбонов на базе Diffusion Transformers (DiT), Flow Matching и других технологий.
• Distributed Training: создание пайплайнов для распределённого обучения тяжёлых foundation-моделей на вычислительном кластере.
• Model-Based RL (MBRL) & Planning: обучение чистого RL и политик IL + RL внутри замороженной латентной симуляции World Model.
• Representation Shaping: интеграция вспомогательных лоссов для perception-задач 3D-детекции, сегментации, трекинга.
• Safety & Inference: построение надёжного safety-контур поверх выходов модели и подготовка конструкции к реалтайм-инференсу на edge-девайсах робота.

Требования

• Экспертное владение JAX и PyTorch.
• Навыки масштабного распределённого обучения.
• Глубокая математическая и ML-база.
• Способность писать, генерировать и проверять быстрый оптимизированный код.

Будет плюсом, если вы

• Работали с Vision Foundation Models, генеративными видео- и image-моделями, синтезом лидарных облаков.
• Имеете опыт в Reinforcement Learning.
• Обладаете опытом продвинутой квантизации тяжёлых трансформеров или диффузионных моделей для edge-девайсов.
• Оптимизировали инференс «на борту» робота.
• Имеете бэкграунд в Autonomous Driving, Motion Planning или Robotics.

Что мы предлагаем

• Возможности для профессионального роста и развития.
• Обучение языку с оплатой 50% стоимости.
• Полный список бонусов доступен на нашем сайте.