ptillet / torch-blocksparse
Block-sparse primitives for PyTorch
☆154Updated 3 years ago
Alternatives and similar repositories for torch-blocksparse:
Users that are interested in torch-blocksparse are comparing it to the libraries listed below
- ☆200Updated last year
- Butterfly matrix multiplication in PyTorch☆165Updated last year
- Research and development for optimizing transformers☆125Updated 3 years ago
- Implementation of a Transformer, but completely in Triton☆253Updated 2 years ago
- Low Precision Arithmetic Simulation in PyTorch☆271Updated 8 months ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆49Updated 6 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- A library of GPU kernels for sparse matrix operations.☆252Updated 4 years ago
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 2 years ago
- ☆158Updated 7 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- ☆136Updated last year
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- A GPU performance profiling tool for PyTorch models☆500Updated 3 years ago
- ☆97Updated 5 months ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated this week
- Pytorch library for factorized L0-based pruning.☆45Updated last year
- ☆157Updated last year
- [ACL'20] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing☆332Updated 6 months ago
- Practical low-rank gradient compression for distributed optimization: https://arxiv.org/abs/1905.13727☆145Updated 3 months ago
- Structured matrices for compressing neural networks☆66Updated last year
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 10 months ago
- pytorch-profiler☆50Updated last year
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Block Sparse movement pruning☆78Updated 4 years ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆135Updated 8 months ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago