Senior ML Engineer для роботов доставки

Что за роль

Вам предстоит стать частью core-команды, работающей над роботами доставки Яндекса, которые ежедневно справляются с задачами в сложной городской среде. Основное внимание будет уделено разработке интерактивных моделей мира и масштабному обучению MBRL-агентов.

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

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

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

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

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

Работа в команде, которая активно растёт и планирует масштабировать флот до 20 000 роботов к 2028 году, предоставит вам возможности для профессионального роста и обучения. Вы сможете внести значимый вклад в развитие технологий доставки.