Research Engineer
Москва
Работа в исследовательской команде робототехники над предобучением и дообучением мультимодальных VLA/WAM-моделей для понимания сцен, предсказания динамики и генерации действий; задачи включают эксперименты, подготовку данных и верификацию моделей на робототехнических платформах для повышения качества и надежности решений.
ГрейдSenior
Стек
PythonPyTorchGPU-кластерыраспределённое обучение на нескольких GPUtransformersdiffusionflow matchingGit
Задачи
- •разработка и предобучение мультимодальных VLA/WAM-моделей
- •подготовка и смешение данных: изображения, видео, текст, демонстрации и траектории роботов
- •дообучение и адаптация моделей под новые задачи и платформы
- •планирование и проведение воспроизводимых экспериментов, анализ результатов и ошибок
- •оптимизация пайплайнов обучения и загрузки данных на GPU-кластерах
- •интеграция и проверка моделей совместно с командами робототехники и инфраструктуры
Обязательно
- •опыт разработки и обучения современных DL-моделей (CV, video, multimodal, LLM/VLM или robot learning)
- •уверенное владение Python и PyTorch
- •понимание современных архитектур, прежде всего transformers, и методов обучения на последовательностях
- •практический опыт pretraining или finetuning моделей: подготовка данных, выбор функций потерь, настройка обучения и оценка качества
- •опыт работы с большими датасетами и эффективными пайплайнами загрузки данных
- •опыт обучения моделей на нескольких GPU и понимание распределённого обучения
- •умение планировать эксперименты и делать выводы по их результатам
Будет плюсом
- •опыт с VLA, world models или action-conditioned video models
- •опыт imitation learning, behavior cloning или diffusion policies
- •опыт работы с diffusion, flow matching или авторегрессионной генерацией видео
- •опыт мультимодального предобучения и self-supervised learning
- •опыт переноса моделей между робототехническими платформами и запусков на реальных роботах
- •опыт масштабирования обучения на нескольких узлах (FSDP/DeepSpeed/аналогичные инструменты)
- •публикации или открытые проекты в ML/CV/робототехнике