huggingface / pytorch_block_sparseLinks
Fast Block Sparse Matrices for Pytorch
☆549Updated 4 years ago
Alternatives and similar repositories for pytorch_block_sparse
Users that are interested in pytorch_block_sparse are comparing it to the libraries listed below
Sorting:
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Library for 8-bit optimizers and quantization routines.☆780Updated 3 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆326Updated 2 years ago
- ☆382Updated last year
- Block-sparse primitives for PyTorch☆160Updated 4 years ago
- Sparse learning library and sparse momentum resources.☆383Updated 3 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆376Updated 4 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆330Updated 3 years ago
- A GPU performance profiling tool for PyTorch models☆505Updated 4 years ago
- Long Range Arena for Benchmarking Efficient Transformers☆764Updated last year
- Profiling and inspecting memory in pytorch☆1,071Updated 3 weeks ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,115Updated 3 years ago
- Implementation for the Lookahead Optimizer.☆243Updated 3 years ago
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆438Updated last year
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆572Updated 3 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆481Updated 3 years ago
- FastFormers - highly efficient transformer models for NLU☆707Updated 6 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,437Updated last month
- Efficient, check-pointed data loading for deep learning with massive data sets.☆209Updated 2 years ago
- Understanding the Difficulty of Training Transformers☆330Updated 3 years ago
- A repository in preparation for open-sourcing lottery ticket hypothesis code.☆632Updated 3 years ago
- Prune a model while finetuning or training.☆405Updated 3 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- Gradient based Hyperparameter Tuning library in PyTorch☆290Updated 5 years ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆269Updated 4 years ago
- Implementation of a Transformer, but completely in Triton☆275Updated 3 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆301Updated 3 months ago
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆486Updated 4 years ago
- PyTorch functions and utilities to make your life easier☆194Updated 4 years ago
- Fully featured implementation of Routing Transformer☆298Updated 3 years ago