rusty1s/pytorch_sparse
A PyTorch extension library providing optimized sparse matrix operations with automatic differentiation support.

Velocity · 7d
+0.4
★ / day
Trend
→steady
star history
This package provides a collection of optimized sparse matrix operations with autograd support, including coalesce, transpose, sparse-dense matrix multiplication, and sparse-sparse matrix multiplication. All operations support both CPU and GPU execution and work with varying data types. The library simplifies sparse tensor manipulation by accepting index and value tensors directly rather than requiring torch.sparse_coo_tensor construction.