Китайский брат выложил в открытый доступ замену для NumPy, которая выполняет вычисления на GPU.
Называется CuPy. Во многих случаях достаточно заменить одну строку:

import numpy as np

на

import cupy as cp

и тот же код может работать на CUDA до 100 раз быстрее.

Что умеет:
→ Совместим с существующим кодом на NumPy и SciPy.
→ Не требует переписывать программу или изучать новый синтаксис.
→ Поддерживает не только CUDA, но и AMD ROCm.

Проект полностью с открытым исходным кодом.

👉 @PythonPortal