PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
☆1,093Mar 9, 2026Updated last week
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,730Mar 9, 2026Updated last week
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆917Mar 9, 2026Updated last week
- Graph Neural Network Library for PyTorch☆23,565Updated this week
- Benchmark datasets, data loaders, and evaluators for graph machine learning☆2,077May 6, 2025Updated 10 months ago
- Low-Level Graph Neural Network Operators for PyG☆224Updated this week
- Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch☆184Mar 7, 2026Updated 2 weeks ago
- Python package built to ease deep learning on graph, on top of existing DL frameworks.☆14,245Jul 31, 2025Updated 7 months ago
- [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.☆1,447Feb 24, 2025Updated last year
- Fast Block Sparse Matrices for Pytorch☆549Jan 21, 2021Updated 5 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,437Aug 21, 2025Updated 7 months ago
- Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors☆2,884Mar 5, 2024Updated 2 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,162Feb 6, 2026Updated last month
- Implementation of "GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings" in PyTorch☆166Aug 30, 2022Updated 3 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,430Feb 20, 2026Updated last month
- ☆113Jul 3, 2021Updated 4 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,475May 2, 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
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,936Updated this week
- Strategies for Pre-training Graph Neural Networks☆1,054Jul 29, 2023Updated 2 years ago
- links to conference publications in graph-based deep learning☆5,037Feb 7, 2026Updated last month
- Submanifold sparse convolutional networks☆2,142Jan 9, 2024Updated 2 years ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆974Updated this week
- Profiling and inspecting memory in pytorch☆1,078Sep 5, 2025Updated 6 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,167Mar 22, 2024Updated last year
- Block-sparse primitives for PyTorch☆158Apr 5, 2021Updated 4 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,628Mar 25, 2022Updated 3 years ago
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,371Apr 4, 2025Updated 11 months ago
- Repository for benchmarking graph neural networks (JMLR 2023)☆2,650Jun 22, 2023Updated 2 years ago
- Generate embeddings from large-scale graph-structured data.☆3,459Mar 3, 2024Updated 2 years ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆302Aug 17, 2023Updated 2 years ago
- PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)☆2,961Sep 18, 2025Updated 6 months ago
- Long Range Arena for Benchmarking Efficient Transformers☆786Dec 16, 2023Updated 2 years ago
- Constrained optimization toolkit for PyTorch☆708Jul 29, 2025Updated 7 months ago
- Development repository for the Triton language and compiler☆18,656Mar 14, 2026Updated last week
- CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)☆1,818Feb 1, 2024Updated 2 years ago
- Graph Diffusion Convolution, as proposed in "Diffusion Improves Graph Learning" (NeurIPS 2019)☆274Apr 26, 2023Updated 2 years ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,250Mar 14, 2026Updated last week
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,064Jun 8, 2023Updated 2 years ago
- A library of GPU kernels for sparse matrix operations.☆283Nov 24, 2020Updated 5 years ago