Вышел CuPy — открытая библиотека для вычислений на GPU
CuPy — тот редкий случай, когда open source не пытается заново изобрести Python для GPU. Библиотека совместима с NumPy и SciPy, так что во многих проектах достаточно заменить импорт на import cupy as cp и уже запускать вычисления на ускорителе.
Поддержка CUDA и AMD ROCm делает проект заметно практичнее типичных «только под один стек» решений. Для команд, у которых уже есть код на NumPy/SciPy, это выглядит как нормальный путь ускорения без переписывания половины пайплайна.
Источник: Machine Learning with Python
Все новости: ai.popovs.tech
