nvmath-python : Libérer la puissance des bibliothèques mathématiques NVIDIA en Python

nvmath-python apporte la puissance des bibliothèques mathématiques de NVIDIA à l'écosystème Python, offrant des API intuitives pour un accès complet aux fonctionnalités dans divers espaces d'exécution. Il s'intègre parfaitement aux frameworks de tableaux/tenseurs Python existants, en se concentrant sur les fonctionnalités qui leur manquent. La bibliothèque expose tous les paramètres de la bibliothèque cuBLASLt sous-jacente, dont certains ne sont pas disponibles dans d'autres wrappers. De plus, elle permet des prologs et des epilogs personnalisés pour les fonctions FFT, les compilant en LTO-IR pour des performances optimisées. Des exemples montrent la multiplication matricielle et les opérations FFT, illustrant ses capacités.