ЯндексML-разработчик в команду претрейна VLA для гуманоидных роботов
офис, гибрид, Москва
Команда занимается претрейном Vision-Language-Action моделей для управления гуманоидными и манипуляционными роботами. Вы будете разрабатывать архитектуры, готовить мультимодальные датасеты и масштабировать распределённое обучение, чтобы улучшать перенос навыков и стабильность запусков
ГрейдSenior
Стек
PythonPyTorchDeepSpeedFSDPJAXCUDA
Задачи
- •Разработка и обучение VLA/мультимодальных моделей для манипуляций
- •Подготовка и согласование мультимодальных робототехнических датасетов
- •Автоматизация контроля качества и очистки данных
- •Масштабирование и оптимизация многоузлового распределённого обучения
- •Проведение исследований, абляций и воспроизводимых экспериментов
- •Валидация моделей на реальных роботах и анализ ошибок
Обязательно
- •Опыт разработки и обучения трансформерных/генеративных или мультимодальных моделей
- •Глубокие знания DL, устройства трансформеров и методов оптимизации
- •Уверенное программирование на Python и работа с PyTorch или JAX
- •Опыт распределённого обучения с FSDP, DeepSpeed или аналогами
- •Опыт работы с большими датасетами и пайплайнами подготовки данных
- •Умение ставить эксперименты, выбирать метрики и обеспечивать воспроизводимость
Будет плюсом
- •Опыт с VLA, Imitation Learning, Behavior Cloning или обучением политик
- •Обучение диффузионных или flow-matching моделей, генерация видео/изображений
- •Оптимизация многоузлового обучения: профилирование GPU, mixed precision, activation checkpointing
- •Разработка пайплайнов автоматического контроля качества и инструментов визуализации
- •Внедрение и оценка обученных политик на реальных роботах
- •Исследовательские публикации или вклад в опенсорс с воспроизводимыми результатами
