PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
☆1,092Aug 12, 2025Updated 6 months ago
Alternatives and similar repositories for pytorch_sparse
Users that are interested in pytorch_sparse are comparing it to the libraries listed below
Sorting:
- PyTorch Extension Library of Optimized Scatter Operations☆1,726Jan 21, 2026Updated last month
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆917Aug 12, 2025Updated 6 months ago
- Graph Neural Network Library for PyTorch☆23,498Feb 19, 2026Updated last week
- Benchmark datasets, data loaders, and evaluators for graph machine learning☆2,074May 6, 2025Updated 9 months ago
- Low-Level Graph Neural Network Operators for PyG☆222Updated this week
- Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch☆184Aug 12, 2025Updated 6 months ago
- [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.☆1,443Feb 24, 2025Updated last year
- Python package built to ease deep learning on graph, on top of existing DL frameworks.☆14,238Jul 31, 2025Updated 7 months ago
- Fast Block Sparse Matrices for Pytorch☆549Jan 21, 2021Updated 5 years ago
- Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors☆2,874Mar 5, 2024Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,158Feb 6, 2026Updated 3 weeks ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,926Updated this week
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,472May 2, 2025Updated 9 months ago
- Implementation of "GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings" in PyTorch☆166Aug 30, 2022Updated 3 years ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆971Jun 14, 2025Updated 8 months ago
- Profiling and inspecting memory in pytorch☆1,078Sep 5, 2025Updated 5 months ago
- links to conference publications in graph-based deep learning☆5,028Feb 7, 2026Updated 3 weeks ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,248Feb 22, 2026Updated last week
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,834Updated this week
- Submanifold sparse convolutional networks☆2,141Jan 9, 2024Updated 2 years ago
- Constrained optimization toolkit for PyTorch☆707Jul 29, 2025Updated 7 months ago
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,350Apr 4, 2025Updated 10 months ago
- Strategies for Pre-training Graph Neural Networks☆1,051Jul 29, 2023Updated 2 years ago
- Generate embeddings from large-scale graph-structured data.☆3,411Mar 3, 2024Updated last year
- Fast, differentiable sorting and ranking in PyTorch☆850Jun 9, 2025Updated 8 months ago
- PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)☆2,949Sep 18, 2025Updated 5 months ago
- Differentiable convex optimization layers☆2,060Updated this week
- Repository for benchmarking graph neural networks (JMLR 2023)☆2,648Jun 22, 2023Updated 2 years ago
- ☆112Jul 3, 2021Updated 4 years ago
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- [ICLR 2020; IPDPS 2019] Fast and accurate minibatch training for deep GNNs and large graphs (GraphSAINT: Graph Sampling Based Inductive L…☆507Aug 12, 2022Updated 3 years ago
- Bayesian optimization in PyTorch☆3,470Updated this week
- C++ extensions in PyTorch☆1,184Jan 13, 2026Updated last month
- Pytorch framework for doing deep learning on point clouds.☆2,690Updated this week
- Long Range Arena for Benchmarking Efficient Transformers☆781Dec 16, 2023Updated 2 years ago