ML-программист КШ
Задачи
Ключевые задачи:
- Разработка и обучение мультимодальных RL-агентов для манипуляции: Создание алгоритмов, которые интегрируют визуальное восприятие (RGB-D), языковые инструкции и проприоцептивные данные (усилие, момент) для управления манипулятором.
Фокус на:
- VLA-планирование: Генерация последовательности действий (траектория, хват, сила) на основе текстового/голосового запроса и визуального контекста;
- Адаптивное управление: Использование обратной связи от тактильных и силомоментных датчиков для коррекции действий в реальном времени (упреждение проскальзывания, работа с деформируемыми объектами);
- Интеграция Foundation Models в контур управления: Адаптация, fine-tuning и эффективное развертывание предобученных моделей (VLMs) и языковых моделей (LLMs) для задач семантического восприятия, планирования и генерации целей (reward) для RL-агента;
- Перенос из симуляции в реальность (Sim2Real) для VLA: Разработка и внедрение методов (рандомизация домена, адаптивные политики) для преодоления разрыва между симуляцией и реальным миром, особенно для мультимодальных представлений;
- Полный цикл — от идеи до "железа": Участие во всем процессе: прототипирование в симуляции, оптимизация для реального времени, развертывание на бортовые компьютеры роботов (Jetson, GPU), тестирование и валидация на физических стендах.
Требования
Наш стек технологий:
- Основное: Python 3.x, PyTorch, Transformers (Hugging Face), PyTorch3D;
- RL/Робототехника: NVIDIA Isaac Lab/Gym, SKRL, Stable-Baselines3, ROS 2 (MoveIt 2, ros2_control), CUDA;
- Визуализация и симуляция: NVIDIA Isaac Sim, Blender;
- Инструменты: Docker, Git, ClearML, Linux, ONNX, TensorRT;
- Хардвер: Роботы-манипуляторы (KUKA, Franka, UR), адаптивные захваты, RGB-D камеры (Intel RealSense), тактильные и силомоментные датчики.
Мы ожидаем, что вы:
- Имеете от 2+ лет коммерческого или серьезного исследовательского опыта в ML/DL, с портфолио проектов (GitHub, публикации);
- Обязательно имеете практический опыт в одной из областей и знакомство с другими:
a. Машинное обучение в робототехнике: Разработка и обучение RL/DL моделей для управления роботами (манипуляция, локомоция). Понимание кинематики, динамики;
b. Visual Language Action (VLA) / Мультимодальные модели: Опыт работы с VLMs (CLIP, BLIP, ViLT) или LLMs для планирования в робототехнике (OpenVLA);
- Имеете продвинутые навыки в PyTorch и работе с данными;
- Имеете опыт работы с симуляторами роботов (Isaac Sim, MuJoCo, PyBullet) и понимаете проблемы Sim2Real;
- Умеете писать производственный код, а не только исследовательские ноутбуки. Опыт с C++ будет большим плюсом;
- Готовы к тесной междисциплинарной работе (программирование, ML, робототехника "низкого уровня").
Будет плюсом:
- Опыт дообучения Foundation Models (LLM, VLM) под специфические задачи;
- Знакомство с иерархическим RL, обратным RL или имитационным обучением (IL, BC);
- Опыт оптимизации и развертывания нейросетей на edge-устройствах (ONNX, TensorRT, Triton);
- Глубокое понимание компьютерного зрения для роботов: 6D pose estimation, сегментация, 3D реконструкция.