Senior ML Engineer — разработка и масштабирование моделей для роботов доставки@ Яндекс

О позиции

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

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

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

Требования

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

Будет плюсом

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

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

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