Senior ML Engineer (Kandinsky)
Москва
Команда разрабатывает генеративные модели и масштабируемые ML‑пайплайны для работы с визуальным контентом. Вы будете исследовать и обучать модели, оптимизировать инференс и внедрять решения для автоматической разметки и параллельной обработки данных
ГрейдSenior
Стек
PythonPyTorchDeepSpeedFSDPдиффузионные моделитрансформеры (DiT)VAETritontorch.compileквантизацияLLM/VLMреляционные БДвекторные БДасинхронная обработкаAPI
Задачи
- •Разрабатывать и обучать генеративные модели для изображений и видео
- •Исследовать и проверять новые архитектуры и методы генерации
- •Дистиллировать модели и оптимизировать скорость инференса и использование памяти
- •Строить масштабируемые пайплайны для разметки, фильтрации и параллельной обработки данных
- •Адаптировать латентные представления и архитектуры под новые задачи
Обязательно
- •Опыт обучения генеративных или CV‑моделей и работы с большими датасетами
- •Знание Python
- •Практический опыт с PyTorch и распределённым обучением (FSDP/DeepSpeed)
- •Хорошая математическая база (линейная алгебра, теория вероятностей, статистика, оптимизация)
- •Умение оценивать эксперименты и воспроизводить исследования
- •Английский для чтения статей и документации
- •Опыт работы с AI‑агентами для ускорения исследований
Будет плюсом
- •Опыт оптимизации с Triton и torch.compile
- •Знание квантизации и оптимизации инференса
- •Опыт backend‑разработки, работы с API и асинхронной обработкой
- •Опыт с реляционными и векторными БД
- •Интеграция и работа с LLM/VLM
- •Научные публикации и вклад в open source