nvmath-python：为Python生态系统带来NVIDIA数学库的强大功能

nvmath-python 将 NVIDIA 数学库的强大功能带入 Python 生态系统。该软件包旨在提供直观的 Python 式 API，让用户能够全面访问 NVIDIA 库在各种执行空间中提供的所有功能。nvmath-python 与现有的 Python 数组/张量框架无缝协作，专注于提供这些框架中缺少的功能。它允许访问底层 NVIDIA cuBLASLt 库的所有参数，其中一些参数在 NVIDIA C-API 库的其他封装中是不可用的。此外，它还支持自定义 FFT 函数的 prolog 和 epilog，并将其编译为 LTO-IR。