ML Engineer (Python)

ML-інженер (Python, генеративні моделі, Diffusion / LoRA / ComfyUI)

Шукаємо Python ML-інженера до команди на ранньому етапі розробки AI-платформи для генерації мультимодального контенту зображень і відео на основі сучасних дифузійних моделей.

Проєкт стартує з нуля, MVP ще в планах, користувачів поки немає.

💡 Що ми створюємо:

Мета побудувати систему генерації персоналізованого візуального контенту.

Фокус інтеграція та кастомізація моделей на кшталт SDXL, LoRA, ControlNet, з можливістю донавчання на користувацьких даних та динамічного підключення ваг.

На старті доведеться спроєктувати архітектуру, підібрати інструменти, протестувати підходи та зібрати перший робочий прототип.

🧠 Завдання:

• Розробка архітектури ML-бекенду з нуля (Python, FastAPI);

• Інтеграція inference-пайплайнів (ComfyUI, Diffusers, Hugging Face);

• Налаштування генерації зображень: LoRA, inpainting, ControlNet;

• Автоматизація донавчання моделей (DreamBooth / Kohya / Diffusers-LoRA);

• Побудова системи зберігання та завантаження кастомних ваг (LoRA);

• Підбір рішень для масштабування: Docker, Kubernetes, хмарні GPU (наприклад, Nebius);

• Інтеграція допоміжних модулів: OCR, TTS, face recognition (за потреби).

⚙️ Стек (те, з чим уже працюємо або плануємо використовувати):

• Python 3.10+, FastAPI, Celery/RQ, Pydantic;

• PyTorch, Hugging Face Diffusers, PEFT, Accelerate;

• Stable Diffusion (SDXL / SD 1.5 / Flux), LoRA, ControlNet, DreamBooth;

• ComfyUI, Automatic1111 (опційно);

• Docker, Kubernetes, PostgreSQL, Redis;

• GPU-інфраструктура (локальна або хмарна).

🔍 Кого шукаємо:

• Впевнене володіння Python та PyTorch;

• Досвід роботи з дифузійними моделями та фреймворками (Diffusers, ComfyUI, A1111);

• Уміння проєктувати ML-пайплайни для inference та навчання з нуля;

• Здатність швидко розбиратись у R&D-завданнях і пропонувати рішення;

• Бажання будувати продукт з найперших стадій: від ідеї до першого прототипу.