Hasty Briefsbeta

Nvmath-Python: Nvidia Math Libraries for the Python Ecosystem

15 hours ago
https://github.com/NVIDIA/nvmath-python
Copy Link
  • #Python
  • #NVIDIA
  • #CUDA
  • nvmath-python integrates NVIDIA math libraries into Python, offering intuitive APIs.
  • Supports various execution spaces and works with Python array/tensor frameworks.
  • Provides access to all parameters of NVIDIA cuBLASLt library, some unavailable elsewhere.
  • Example usage includes matrix multiplication with fine-tuning options like mixed-precision compute types.
  • Allows for device-side API calls, enabling NVIDIA library functions within custom device kernels.
  • Supports custom prologs and epilogs for FFT functions, compiles them to LTO-IR.
  • Example demonstrates unitary scaling for FFT outputs via a custom epilog.
  • Subject to Apache 2.0 license and currently in Beta, welcoming feedback for improvements.
AboutLogin