nvmath-python: Desbloqueando as bibliotecas matemáticas da NVIDIA em Python
2025-09-22
nvmath-python traz o poder das bibliotecas matemáticas da NVIDIA para o ecossistema Python, oferecendo APIs intuitivas para acesso completo aos recursos em vários espaços de execução. Ele se integra perfeitamente às estruturas de matriz/tensor Python existentes, focando em funcionalidades ausentes nelas. A biblioteca expõe todos os parâmetros da biblioteca cuBLASLt subjacente, alguns indisponíveis em outros wrappers. Além disso, permite prologs e epilogs personalizados para funções FFT, compilando-os para LTO-IR para desempenho otimizado. Exemplos demonstram multiplicação de matrizes e operações FFT, mostrando suas capacidades.
Desenvolvimento
Bibliotecas Matemáticas