Une implémentation de tableaux multi-dimensionnels compatibles avec ceux de NumPy, fonctionnant avec puces CUDA. Cela permet de transférer les calculs du CPU au GPU pour un gain de performance d’un facteur allant jusqu’à 10.000% lors de la parralélisation.
Étant compatible avec NumPy, CuPy propose :
- Des tableaux multidimensionnels
- Des fonctions de broadcast sophistiquées
- Des outils pour intégrer du code C/C++ et Fortan
- De l’algèbre linéaire, des transformées de Fourier et de la manipulation de nombres pseudo-aléatoires.