nvmath-python: Desatando el poder de las bibliotecas matemáticas de NVIDIA en Python

nvmath-python lleva la potencia de las bibliotecas matemáticas de NVIDIA al ecosistema de Python, ofreciendo APIs intuitivas para un acceso completo a las funciones en varios espacios de ejecución. Se integra perfectamente con los frameworks de matrices/tensores Python existentes, centrándose en las funcionalidades que faltan en ellos. La biblioteca expone todos los parámetros de la biblioteca cuBLASLt subyacente, algunos de los cuales no están disponibles en otros wrappers. Además, permite prologs y epilogs personalizados para funciones FFT, compilándolos a LTO-IR para un rendimiento optimizado. Los ejemplos muestran la multiplicación de matrices y las operaciones FFT, mostrando sus capacidades.