← all repositories

rusty1s/pytorch_sparse

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

1.1k stars Python ML Frameworks
pytorch_sparse
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.