nvmath-python:PythonでNVIDIAの数学ライブラリを活用

2025-09-22
nvmath-python:PythonでNVIDIAの数学ライブラリを活用

nvmath-pythonは、NVIDIAの数学ライブラリの機能をPythonエコシステムにもたらします。直感的なPythonic APIを提供し、様々な実行空間で提供されるNVIDIAライブラリのすべての機能に完全にアクセスできます。nvmath-pythonは既存のPython配列/テンソルフレームワークとシームレスに連携し、それらのフレームワークにはない機能を提供することに重点を置いています。また、cuBLASLtライブラリのパラメータすべてにアクセスでき、他のNVIDIA C-APIライブラリのラッパーでは利用できないパラメータも含まれます。さらに、FFT関数のカスタムプロローグとエピローグを作成し、それらをLTO-IRにコンパイルすることもできます。