Block-sparse primitives for PyTorch
☆158Apr 5, 2021Updated 4 years ago
Alternatives and similar repositories for torch-blocksparse
Users that are interested in torch-blocksparse are comparing it to the libraries listed below
Sorting:
- Fast Block Sparse Matrices for Pytorch☆549Jan 21, 2021Updated 5 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,064Jun 8, 2023Updated 2 years ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆50Mar 1, 2018Updated 8 years ago
- Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.☆23Aug 21, 2020Updated 5 years ago
- ☆115Aug 26, 2024Updated last year
- A library of GPU kernels for sparse matrix operations.☆283Nov 24, 2020Updated 5 years ago
- ☆18Oct 15, 2020Updated 5 years ago
- ☆10Feb 1, 2022Updated 4 years ago
- Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"☆1,611Aug 12, 2020Updated 5 years ago
- Distributed SDDMM Kernel☆12Jul 8, 2022Updated 3 years ago
- The codes for ECCV'22: Learning to Train a Point Cloud Reconstruction Network without Matching☆10Nov 16, 2022Updated 3 years ago
- ☆12May 23, 2018Updated 7 years ago
- ☆166Jul 22, 2024Updated last year
- Implementation and analysis of five different GPU based SPMV algorithms in CUDA☆40Feb 5, 2019Updated 7 years ago
- PyCUDA based PyTorch Extension Made Easy☆27Mar 22, 2024Updated 2 years ago
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,093Mar 9, 2026Updated last week
- The codes for RFNet: Recurrent Forward Network for Dense Point Cloud Completion☆20Jan 17, 2022Updated 4 years ago
- ☆223Feb 21, 2023Updated 3 years ago
- ☆20Oct 11, 2023Updated 2 years ago
- Adaptive floating-point based numerical format for resilient deep learning☆14Apr 11, 2022Updated 3 years ago
- ☆10Aug 2, 2021Updated 4 years ago
- Dinghow's notes about Digital Art, School of Software Engineering, Tongji University☆12Sep 23, 2019Updated 6 years ago
- [VLDB 2024] Source code for FusionQuery: On-demand Fusion Queries over Multi-source Heterogeneous Data☆11Mar 11, 2025Updated last year
- Open single and half precision gemm implementations☆397Apr 2, 2023Updated 2 years ago
- ☆17Oct 22, 2022Updated 3 years ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,078Apr 17, 2024Updated last year
- [ICLR 2023] 'Revisiting Pruning At Initialization Through The Lens of Ramanujan Graph" by Duc Hoang, Shiwei Liu, Radu Marculescu, Atlas W…☆14Aug 4, 2023Updated 2 years ago
- Development repository for the Triton language and compiler☆18,708Updated this week
- ☆32Aug 24, 2022Updated 3 years ago
- ☆17Jul 10, 2022Updated 3 years ago
- Improving Post Training Neural Quantization: Layer-wise Calibration and Integer Programming☆98Jun 10, 2021Updated 4 years ago
- A GPU performance profiling tool for PyTorch models☆510Jul 13, 2021Updated 4 years ago
- A Vector Caching Scheme for Streaming FPGA SpMV Accelerators☆10Sep 7, 2015Updated 10 years ago
- A GPipe implementation in PyTorch☆862Jul 25, 2024Updated last year
- ☆103Dec 11, 2020Updated 5 years ago
- ☆57Dec 8, 2020Updated 5 years ago
- Multiple 1-stencil implementations using nvidia cuda.☆12Dec 2, 2017Updated 8 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆199Apr 27, 2022Updated 3 years ago
- Karras et al. (2022) diffusion models for PyTorch☆14Nov 15, 2022Updated 3 years ago