PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
☆1,105Jun 3, 2026Updated 2 months ago
Alternatives and similar repositories for pytorch_sparse
Users that are interested in pytorch_sparse are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- PyTorch Extension Library of Optimized Scatter Operations☆1,745Jun 3, 2026Updated 2 months ago
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆928Jun 5, 2026Updated 2 months ago
- Graph Neural Network Library for PyTorch☆24,046Updated this week
- Benchmark datasets, data loaders, and evaluators for graph machine learning☆2,094May 6, 2025Updated last year
- Low-Level Graph Neural Network Operators for PyG☆238Updated this week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch☆183Jun 5, 2026Updated 2 months ago
- Python package built to ease deep learning on graph, on top of existing DL frameworks.☆14,282Jul 31, 2025Updated last year
- [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.☆1,472Feb 24, 2025Updated last year
- Fast Block Sparse Matrices for Pytorch☆551Jan 21, 2021Updated 5 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,434Aug 21, 2025Updated last year
- Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors☆2,956Mar 5, 2024Updated 2 years ago
- Implementation of "GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings" in PyTorch☆167Aug 30, 2022Updated 4 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,189Jul 17, 2026Updated last month
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,582Updated this week
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆115Jul 3, 2021Updated 5 years ago
- [ICLR 2020; IPDPS 2019] Fast and accurate minibatch training for deep GNNs and large graphs (GraphSAINT: Graph Sampling Based Inductive L…☆509Aug 12, 2022Updated 4 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,482May 2, 2025Updated last year
- Strategies for Pre-training Graph Neural Networks☆1,070Jul 29, 2023Updated 3 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,996Updated this week
- links to conference publications in graph-based deep learning☆5,096Jun 7, 2026Updated 2 months ago
- Submanifold sparse convolutional networks☆2,145Jan 9, 2024Updated 2 years ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆990Mar 19, 2026Updated 5 months ago
- Profiling and inspecting memory in pytorch☆1,077Aug 12, 2026Updated 2 weeks ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- torch-optimizer -- collection of optimizers for Pytorch☆3,171Mar 22, 2024Updated 2 years ago
- Block-sparse primitives for PyTorch☆158Apr 5, 2021Updated 5 years ago
- Repository for benchmarking graph neural networks (JMLR 2023)☆2,670Jun 22, 2023Updated 3 years ago
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,477Apr 4, 2025Updated last year
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,626Mar 25, 2022Updated 4 years ago
- Generate embeddings from large-scale graph-structured data.☆3,454Mar 3, 2024Updated 2 years ago
- Long Range Arena for Benchmarking Efficient Transformers☆788Dec 16, 2023Updated 2 years ago
- PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)☆2,992May 30, 2026Updated 3 months ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆305Aug 17, 2023Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Constrained optimization toolkit for PyTorch☆710Jun 7, 2026Updated 2 months ago
- CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)☆1,821Feb 1, 2024Updated 2 years ago
- Development repository for the Triton language and compiler☆20,044Updated this week
- Graph Diffusion Convolution, as proposed in "Diffusion Improves Graph Learning" (NeurIPS 2019)☆276Apr 26, 2023Updated 3 years ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,264Updated this week
- A library of GPU kernels for sparse matrix operations.☆289Nov 24, 2020Updated 5 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,068Jun 8, 2023Updated 3 years ago