High-performance GPU SpMV kernels in Python/Numba CUDA. Achieves 96.6 GB/s (53% of cuSPARSE) with 5,432x speedup over CPU. Includes optimization analysis.
-
Updated
Dec 27, 2025 - Python
FFFF
High-performance GPU SpMV kernels in Python/Numba CUDA. Achieves 96.6 GB/s (53% of cuSPARSE) with 5,432x speedup over CPU. Includes optimization analysis.
Triton CSR SpMV performance study comparing SciPy CPU and PyTorch CUDA across workload size and sparse row density.
Sparse matrix operations for AWS Trainium via NKI (cuSPARSE-equivalent) — CSR/COO formats, SpMV and SpMM via gather-matmul-scatter, Schwarz integral screening for quantum chemistry.
Add a description, image, and links to the spmv topic page so that developers can more easily learn about it.
To associate your repository with the spmv topic, visit your repo's landing page and select "manage topics."