huggingface / pytorch_block_sparseLinks
Fast Block Sparse Matrices for Pytorch
☆545Updated 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:
- Library for 8-bit optimizers and quantization routines.☆716Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- ☆376Updated last year
- Block-sparse primitives for PyTorch☆155Updated 4 years ago
- A GPU performance profiling tool for PyTorch models☆503Updated 3 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆321Updated 2 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆374Updated 4 years ago
- Experimental ground for optimizing memory of pytorch models☆365Updated 7 years ago
- Implementation for the Lookahead Optimizer.☆240Updated 3 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,112Updated 3 years ago
- PyTorch layer-by-layer model profiler☆607Updated 4 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 2 years ago
- Profiling and inspecting memory in pytorch☆1,059Updated 9 months ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆479Updated 2 years ago
- Long Range Arena for Benchmarking Efficient Transformers☆757Updated last year
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆438Updated 9 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆299Updated 3 years ago
- FastFormers - highly efficient transformer models for NLU☆705Updated 2 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,429Updated this week
- Sparse learning library and sparse momentum resources.☆381Updated 2 years ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆208Updated last year
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Automatic differentiation with weighted finite-state transducers.☆455Updated 3 years ago
- PyTorch functions and utilities to make your life easier☆195Updated 4 years ago
- Understanding the Difficulty of Training Transformers☆329Updated 3 years ago
- Implementation of a Transformer, but completely in Triton☆265Updated 3 years ago
- Estimates the size of a PyTorch model in memory☆357Updated 4 years ago
- Tutorial for building a custom CUDA function for Pytorch☆515Updated 6 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆329Updated last year