Senior ML Engineer в команде робота доставки

Что за роль

В команде робота доставки Яндекса открыта позиция Senior ML Engineer. Функция сосредоточена на разработке и масштабировании интерактивных моделей мира для роботов, которые ежедневно доставляют тысячи заказов в сложной городской среде. Роль предполагает работу с передовыми технологиями и крупными вычислительными кластерами.

Что предстоит делать

• Проектировать и обучать 3D/видеотокенизаторы и бэкбоны на базе Diffusion Transformers.
• Строить пайплайны для распределённого обучения тяжёлых foundation-моделей.
• Обучать RL и IL+RL политики внутри латентной симуляции World Model.
• Интегрировать вспомогательные лоссы для задач 3D-детекции и трекинга.
• Создавать надёжный safety-контур для реального времени на edge-девайсах.

Что потребуется

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

Что важно знать

Яндекс активно растёт и планирует масштабировать флот роботов до 20 000 к 2028 году. С нами легко развиваться: если вам нужно подтянуть язык, мы организуем обучение и оплатим 50% стоимости. Полный список бонусов доступен на сайте.