Nvmath-Python: Nvidia Math Libraries for the Python Ecosystem

Nvmath-Python: Nvidia Math Libraries for the Python Ecosystem(github.com)

67 points by gballan 287 days ago | 4 comments

amelius 284 days ago |

Closer to arithmetic libraries than math libraries.

KeplerBoy 284 days ago |

Nice, custom prologues and epilogues for FFTs are a welcome feature.

physicsguy 284 days ago | |

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.

KeplerBoy 284 days ago | | |

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.