PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
☆1,106Jun 3, 2026Updated 4 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,748Jun 3, 2026Updated 4 months ago
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆931Jun 5, 2026Updated 4 months ago
- Graph Neural Network Library for PyTorch☆24,112Updated this week
- Benchmark datasets, data loaders, and evaluators for graph machine learning☆2,098May 6, 2025Updated last year
- Low-Level Graph Neural Network Operators for PyG☆240Updated this week
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch☆184Jun 5, 2026Updated 4 months ago
- Python package built to ease deep learning on graph, on top of existing DL frameworks.☆14,281Jul 31, 2025Updated last year
- [MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.☆1,474Feb 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,965Mar 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,191Jul 17, 2026Updated 2 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,615Aug 26, 2026Updated last month
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆116Jul 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…☆508Aug 12, 2022Updated 4 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,486May 2, 2025Updated last year
- Strategies for Pre-training Graph Neural Networks☆1,071Jul 29, 2023Updated 3 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆9,005Updated this week
- links to conference publications in graph-based deep learning☆5,101Sep 28, 2026Updated last week
- Submanifold sparse convolutional networks☆2,144Jan 9, 2024Updated 2 years ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆991Mar 19, 2026Updated 6 months ago
- Profiling and inspecting memory in pytorch☆1,079Aug 12, 2026Updated last month
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- 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,485Apr 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,624Mar 25, 2022Updated 4 years ago
- Generate embeddings from large-scale graph-structured data.☆3,452Mar 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 4 months ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆306Aug 17, 2023Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Constrained optimization toolkit for PyTorch☆712Jun 7, 2026Updated 4 months ago
- CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)☆1,820Feb 1, 2024Updated 2 years ago
- Development repository for the Triton language and compiler☆20,337Updated 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,266Sep 16, 2026Updated 3 weeks ago
- 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,067Jun 8, 2023Updated 3 years ago