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

Что за роль

Вам предстоит стать частью core-команды, занимающейся разработкой и масштабированием моделей для роботов доставки Яндекса. Основное внимание будет уделено созданию интерактивной модели мира и обучению MBRL-агентов.
Что предстоит делать:
• Разработка и скейлинг World Models, проектирование и обучение 3D/видеотокенизаторов на базе Diffusion Transformers.
• Построение пайплайнов для распределённого обучения тяжёлых foundation-моделей на вычислительном кластере.
• Обучение чистого RL и политик IL + RL внутри замороженной латентной симуляции World Model.
• Интеграция вспомогательных лоссов для perception-задач 3D-детекции, сегментации, трекинга.
• Построение надёжного safety-контуры для реалтайм-инференса на edge-девайсах.

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

Мы ищем кандидата с экспертизой в JAX и PyTorch, навыками масштабного распределённого обучения и глубокой математической базой. В команде вы сможете развиваться и получать поддержку для обучения языкам и другим навыкам.

О компании

Компания активно растёт и планирует масштабировать флот до 20 000 роботов к 2028 году.