Linear Algebra — NVIDIA nvmath-python (original) (raw)
Overview#
The Linear Algebra module nvmath.linalg in nvmath-python leverages various NVIDIA math libraries to support multiple linear algebra computations. As of the initial Beta release, we offer the specialized matrix multiplication API based on the cuBLASLt library.
API Reference#
Generic Linear Algebra APIs (nvmath.linalg)#
Generic APIs will be available in a later release.
Specialized Linear Algebra APIs (nvmath.linalg.advanced)#
Helpers#
The Specialized Linear Algebra helpers module nvmath.linalg.advanced.helpersprovides helper functions to facilitate working with some of the complex features ofnvmath.linalg.advanced module.