Machine Learning Engineer (Image Processing)

Ми шукаємо досвідченого 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 та обробки зображень
• Гнучкий графік роботи
• Мінімум мітингів, максимум технічної роботи
• Заробітну плату, яка обговорюється індивідуально