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,063Jun 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
- A library of GPU kernels for sparse matrix operations.☆283Nov 24, 2020Updated 5 years ago
- ☆18Oct 15, 2020Updated 5 years ago
- ☆19Dec 10, 2021Updated 4 years ago
- ☆115Aug 26, 2024Updated last year
- Distributed SDDMM Kernel☆12Jul 8, 2022Updated 3 years ago
- Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"☆1,610Aug 12, 2020Updated 5 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆15Aug 30, 2019Updated 6 years ago
- Repository for artifact evaluation of ASPLOS 2023 paper "SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning"☆25Feb 24, 2023Updated 3 years ago
- ☆164Jul 22, 2024Updated last year
- Adaptive floating-point based numerical format for resilient deep learning☆14Apr 11, 2022Updated 3 years ago
- ☆15Mar 31, 2023Updated 2 years ago
- ☆23Oct 24, 2022Updated 3 years ago
- Ahead of Time (AOT) Triton Math Library☆92Updated this week
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,092Aug 12, 2025Updated 6 months ago
- Research and development for optimizing transformers☆131Feb 16, 2021Updated 5 years ago
- Reorder-based post-training quantization for large language model☆199May 17, 2023Updated 2 years ago
- Open single and half precision gemm implementations☆398Apr 2, 2023Updated 2 years ago
- ☆22Feb 18, 2025Updated last year
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,075Apr 17, 2024Updated last year
- A library to analyze PyTorch traces.☆467Feb 4, 2026Updated 3 weeks ago
- A GPipe implementation in PyTorch☆863Jul 25, 2024Updated last year
- ☆222Feb 21, 2023Updated 3 years ago
- A simple semantic search engine for scientific papers.☆28Sep 14, 2023Updated 2 years ago
- This toolbox☆12Jul 2, 2021Updated 4 years ago
- ☆12May 23, 2018Updated 7 years ago
- ☆10Aug 2, 2021Updated 4 years ago
- This simulator models multi core systems, intended primarily for studies on main memory management techniques. It models a trace-based ou…☆12Jan 18, 2016Updated 10 years ago
- ☆12Jul 9, 2021Updated 4 years ago
- ☆11Aug 4, 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
- A GPU performance profiling tool for PyTorch models☆510Jul 13, 2021Updated 4 years ago
- Improving Post Training Neural Quantization: Layer-wise Calibration and Integer Programming☆98Jun 10, 2021Updated 4 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,534Updated this week
- ☆10Mar 2, 2017Updated 9 years ago
- Multiple 1-stencil implementations using nvidia cuda.☆13Dec 2, 2017Updated 8 years ago