Cuda-X için Python kütüphanesi matris ve vektör hesaplamalarını hızlandırır

Saberie

Active member
Nvidia, Cuda-X için yeni bir özel Python matematik kütüphanesi yayınladı. Ek C/C++ kütüphanelerini kullanmak zorunda kalmadan Cuda-X'in temel matematik işlemlerine Python benzeri anında erişim sağlar. Donanımla hızlandırılmış derin öğrenme uygulamaları, kitaplıklar, çerçeveler veya derleyiciler geliştirmek için kullanılır.


Duyuru



Matematiksel işlevler açısından, nvmath-python özellikle doğrusal cebiri ve n boyutlu ayrık Fourier dönüşümlerini destekler. Nvidia, bu hesaplamalar için yerel C kütüphanelerine yakın olması gereken çok iyi bir performans vaat ediyor. Ek yük minimumda tutulur ve örneğin birden fazla çekirdeğin ek bir ana bilgisayar olmadan kombinasyonuna (çekirdek füzyonu) izin verir.

Kütüphane hala beta aşamasındadır, Apache 2 lisansı altındadır ve durum bilgisi olmayan (işlev stili) veya durum bilgisi olan (sınıf stili) API'ler sunar. Hesaplamaların ayrıntılarını anlamak için standart günlükçü kütüphanesine entegre olur. Numba gibi Python derleyicileriyle birlikte nvmath-python'un davranışını uyarlayan cihaz geri çağrıları oluşturmak mümkündür.



Nvidia'ya göre nvmath, Python ekosistemiyle hem CuPy, PyTorch veya RAPIDS gibi diğer GPU odaklı paketlerle hem de SciPy, scikit-learn veya NumPy gibi CPU kitaplıklarıyla sorunsuz bir şekilde çalışıyor. İkincisinin yeni bir ana versiyonu da yakın zamanda piyasaya sürüldü.

Nvmath-python hakkında daha fazla bilgiyi proje web sitesinde, GitHub'da veya demo videoda bulabilirsiniz.


(DSÖ)
 
Üst