Nvmath-Python: Nvidia Math Libraries for the Python Ecosystem(github.com)
github.com
Nvmath-Python: Nvidia Math Libraries for the Python Ecosystem
https://github.com/NVIDIA/nvmath-python
5 comments
[deleted]
Closer to arithmetic libraries than math libraries.
Nice, custom prologues and epilogues for FFTs are a welcome feature.
That stuff was in FFTW and cuFFT for many many years. It's one of the features I always pointed to as missing in rocFFT but they've finally caught up.
Yes, it has been in cuFFT for a long time. I thought it's new to python cuda libraries, but no. Turns out it has also been a thing in CuPy for a long time.