
CuPy je open source matematická knihovna pro Python, která používá array kompatibilní s NumPy a je akcelerována na GPU NVidia pomocí CUDA (odtud název). Minulý týden vyšla knihovna CuPy ve verzi 9, která přidává podporu pro GPU AMD pomocí ROCm 4.0. Použití s AMD je stále experimentální, zájemci se mohou podívat do návodu.
Mezi další novinky patří JIT API ke psaní kernelů CUDA přímo v Pythonu. Nově je vyžadována CUDA alespoň 9.2 a Python alespoň 3.6. Zdrojové kódy CuPy jsou na GitHubu.
(zdroj: phoronix)