nvmath-python: Entfesselt die NVIDIA-Mathematikbibliotheken in Python

nvmath-python bringt die Leistungsfähigkeit der NVIDIA-Mathematikbibliotheken in das Python-Ökosystem und bietet intuitive APIs für den vollständigen Zugriff auf Funktionen in verschiedenen Ausführungsräumen. Es integriert sich nahtlos in bestehende Python-Array/Tensor-Frameworks und konzentriert sich auf fehlende Funktionalitäten. Die Bibliothek legt alle Parameter der zugrunde liegenden cuBLASLt-Bibliothek offen, von denen einige in anderen Wrappern nicht verfügbar sind. Darüber hinaus ermöglicht sie benutzerdefinierte Prologe und Epiloge für FFT-Funktionen und kompiliert sie in LTO-IR für optimierte Leistung. Beispiele zeigen Matrixmultiplikation und FFT-Operationen und demonstrieren die Fähigkeiten.