nvmath-python brings the power and performance of NVIDIA math libraries to the Python ecosystem with intuitive, pythonic APIs. The ultimate goal is to provide users full access to all of the available library features in a variety of execution spaces.
pip install nvmath-python[cu12]Please refer to the installation instructions for different ways of installing nvmath-python, including building from source.