nvmath-python: Unleashing NVIDIA's Math Libraries in Python
2025-09-22
nvmath-python brings the power of NVIDIA's math libraries to the Python ecosystem, offering intuitive APIs for full access to features across various execution spaces. It seamlessly integrates with existing Python array/tensor frameworks, focusing on functionalities missing from them. The library exposes all parameters of the underlying cuBLASLt library, some unavailable in other wrappers. Furthermore, it allows custom prologs and epilogs for FFT functions, compiling them to LTO-IR for optimized performance. Examples demonstrate matrix multiplication and FFT operations, showcasing its capabilities.
Development
Math Libraries