Senior ML Engineer (Image Generation / ComfyUI / Stable Diffusion) (удаленная работа)

Вакансия: Senior ML Engineer (Image Generation / ComfyUI / Stable Diffusion)

Описание вакансии

О проекте:
Мы создаём систему интерактивных ИИ-аватаров, которые общаются с
пользователями и умеют генерировать изображения в реальном времени. Цель - сделать устойчивых, "живых" персонажей с уникальным визуальным стилем,
индивидуальностью и эмоциями. Ищем сильного ML / CV инженера, который
поможет вывести визуальную часть на новый уровень.

Задачи:
Настройка и оптимизация пайплайнов ComfyUI / Stable Diffusion / ControlNet / SDXL/Flux/SD3;
Обучение собственных чекпоинтов и доработка существующих моделей под
нужды продукта;
Тюнинг LoRA, Textual Inversion, DreamBooth, работа с кастомными датасетами
Создание консистентных аватаров по описанию или референсам (персонажи
сохраняют внешний вид и стиль между сценами);
Эксперименты с различными моделями генерации изображений (SDXL, Flux, SD3 и др.);
Разработка и поддержка пайплайна fine-tuning / inference (оптимизация VRAM,
скорость, качество);
Взаимодействие с продакт- и инженерными командами для интеграции ML в
продукт (API, UI и т.д.).

Требования:
Глубокий опыт с ComfyUI или Automatic1111 на продакшн-уровне;
Понимание принципов diffusion-моделей и fine-tuning;
Опыт обучения собственных чекпоинтов (SDXL / SD 1.5 / Flux / SD3 или аналогичных моделей);
Опыт обучения LoRA, DreamBooth, Textual Inversion с нуля;
Умение добиваться персонажной консистентности (лица, стиль, позы, эмоции);
Опыт работы с ControlNet, IP-Adapter, StyleAdapter;
Уверенные знания Python, PyTorch;
Опыт оптимизации inference (TensorRT, xFormers, SDXL Turbo и т.д.) будет плюсом;
Бонусом: опыт интеграции ML в продакшн-продукты.

Технологии:
ComfyUI, Stable Diffusion / SDXL / SD3 / Flux, LoRA, DreamBooth, Textual Inversion, ControlNet, PyTorch, Python, CUDA, TensorRT, Diffusers, HuggingFace.

Формат:
Удалённо / гибрид;
Полная занятость или проектная работа;
Оплата выше рынка (обсуждается индивидуально, в зависимости от опыта).

Что мы предлагаем:
Работа с живым продакшн-продуктом, где ИИ-аватары реально взаимодействуют с пользователями;
Возможность экспериментировать с самыми свежими моделями и архитектурами;
Минимум бюрократии максимум R&D;
Тесное взаимодействие с продакт- и инженерными командами;
Отличный шанс сделать заметный и инновационный продукт.