Мы ищем Python-разработчика, который будет заниматься оптимизацией/доработкой пайплайнов генерации изображений и видео на базе open-source AI-решений. Работа связана с ускорением inference-процессов, улучшением качества генерации и оптимизацией GPU-вычислений.
Что нужно делать:
- Выделение и рефакторинг pipeline-процессов генерации из open-source в отдельные оптимизированные методы.
- Оптимизировать генерацию изображений и видео.
- Использовать профайлеры для поиска утечек памяти, задержек и узких мест.
- Улучшать производительность GPU-вычислений.
- Дорабатывать и оптимизировать существующий код для повышения качества генерации изображений и видео.
Требования:
- Уверенный Python.
- Опыт работы со стеком: CUDA + cuDNN, NumPy, OpenCV, PyTorch.
Будет плюсом:
- Опыт работы с TensorRT.
- Опыт оптимизации ML/AI пайплайнов.
- Понимание работы GPU, memory management и профилирования.
Мы предлагаем:
- Удаленную работу;
- Гибкий график;
- З/П: 2.500$.
