Ми шукаємо досвідченого ML-інженера, який працює з моделями комп’ютерного зору (CV) та вміє адаптувати їх під реальні задачі вебплатформи. Вашою основною зоною відповідальності буде інтеграція сучасних моделей сегментації, matting та inpainting, дообучення моделей, а також розгортання inference-пайплайнів з GPU-оптимізацією.
Технічний стек:
- PyTorch, TorchScript, ONNX
- OpenCV, CUDA
- FastAPI або Flask для розробки REST API
- TensorRT для оптимізації inference
- Досвід з моделями: LaMa, ZITS, MODNet, RVM або подібними

Основні задачі:
- Робота з уже існуючими відкритими моделями (inpainting, matting, segmentation)
- Дообучення моделей на власних датасетах з урахуванням специфіки платформи
- Видалення водяних знаків (LaMa, ZITS), розмиття фону (MODNet, RVM)
- Побудова пайплайнів inference (TorchScript/ONNX > TensorRT)
- Розробка REST API для взаємодії з моделями
- Деплой сервісів на сервер із NVIDIA GPU

Буде плюсом:
- Розуміння архітектури нейромереж, принципів дообучення та transfer learning
- Досвід оптимізації inference для продакшену
- Вміння писати чистий, структурований код та документацію

Ми пропонуємо:
- Роботу над цікавими продуктами в галузі AI та обробки зображень
- Гнучкий графік роботи
- Мінімум мітингів, максимум технічної роботи
- Заробітну плату, яка обговорюється індивідуально

Контактна інформація →