Разработчик ML/LLM

Компания "Рекламное агентство ВИВАТ"

Мы ищем разработчика, который поможет нам автоматизировать создание качественных изображений. Ваша задача — построить систему пакетной обработки/генерации изображений.

Что предстоит делать:

• развернуть и поддерживать локальную AI‑инфраструктуру для генерации изображений;
• настраивать и оптимизировать пайплайны на базе ComfyUI / Stable Diffusion / ControlNet / SDXL / Flux / SD3 для пакетной обработки изображений;
• обучать собственные чекпоинты и дорабатывать существующие модели под специфику продукта — чтобы сохранять форму объекта, получать чистый фон и единый стиль изображений;
• настраивать масштабирование (multi-worker / multi-GPU);
• экспериментировать с разными моделями генерации (SDXL, Flux, SD3 и др.), чтобы найти оптимальное соотношение качества и скорости;
• развивать DevOps‑pipeline для ML/LLM‑сервисов: настраивать CI/CD, управлять версиями моделей и конфигураций, обеспечивать безопасные релизы, проводить A/B‑тесты и быстрый rollback;
• настроить систему мониторинга нагрузки на GPU, времени генерации и ошибок (например, с помощью Prometheus + Grafana);
• организовать систему очередей задач генерации и балансировку нагрузки между узлами;
• взаимодействовать с бэкенд для интеграции ML‑решений в API‑сервисы (работа с FastAPI, очередями и кэшированием);
• использовать OpenCV и PIL/Pillow для предобработки входных изображений (вырезание фона, выравнивание, нормализация) и постобработки сгенерированных;
• оптимизировать скорость генерации с помощью TensorRT для ускорения инференса на GPU;
• организовать хранение исходных и сгенерированных изображений,
  а также чекпоинтов моделей в S3‑совместимых хранилищах.

Чего мы ждём от вас:

• уверенные знания Python(asyncio, aiohttp/FastAPI), PyTorch;
• глубокий опыт работы с ComfyUI на продакшн‑уровне;
• хорошее понимание микросервисной архитектуры и event‑driven подхода;
• понимание принципов работы diffusion‑моделей и методов fine‑tuning;
• опыт обучения собственных чекпоинтов для моделей SDXL / SD 1.5 / Flux / SD3 или аналогичных;
• опыт самостоятельного обучения LoRA, DreamBooth, Textual Inversion;
• умение добиваться консистентности между разными генерациями (сохранять ключевые признаки объекта);
• опыт работы с инструментами управления композицией — ControlNet, IP‑Adapter, StyleAdapter;
• опыт использования OpenCV или PIL/Pillow для обработки изображений;
• знакомство с TensorRT или другими инструментами оптимизации инференса для ускорения генерации;
• опыт работы с S3‑совместимыми хранилищами;
• понимание принципов построения REST API и интеграции с бэкендом;
• опыт работы с PostgreSQL (оптимизация сложных запросов, проектирование схем);
• опыт развёртывания и поддержки локальной AI‑инфраструктуры, включая:
  • работу с GPU‑серверами (NVIDIA, CUDA) и оптимизацию использования ресурсов;
  • работу с Docker и Docker Compose;
  • навыки оркестрации контейнеров (Kubernetes — будет плюсом);
  • понимание принципов мониторинга ML‑систем (Prometheus, Grafana или аналоги);
  • навыки работы с Redis (кэширование, сессии, очереди);
  • опыт построения отказоустойчивых систем и организации резервного копирования.

Что мы предлагаем:

• работа с высоконагруженной системой генерации изображений;
• возможность экспериментировать с самыми свежими моделями и архитектурами;
• минимум бюрократии — вы сможете сосредоточиться на решении интересных задач;
• шанс внести вклад в создание заметного и инновационного продукта.

Открыть контакты работодателя

Зарегистрируйтесь или войдите, чтобы открыть контакты работодателя