microsoft / infinibatch
Efficient, check-pointed data loading for deep learning with massive data sets.
☆205Updated last year
Alternatives and similar repositories for infinibatch:
Users that are interested in infinibatch are comparing it to the libraries listed below
- Understanding the Difficulty of Training Transformers☆327Updated 2 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆100Updated 4 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆310Updated last year
- Profile the GPU memory usage of every line in a Pytorch code☆82Updated 6 years ago
- FairSeq repo with Apollo optimizer☆108Updated last year
- Implementation of a Transformer, but completely in Triton☆251Updated 2 years ago
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆429Updated 2 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 4 years ago
- LM Pretraining with PyTorch/TPU☆134Updated 5 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆130Updated 7 months ago
- Fast Block Sparse Matrices for Pytorch☆547Updated 3 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!☆111Updated last year
- Implementation of https://arxiv.org/abs/1904.00962☆372Updated 4 years ago
- Torch Distributed Experimental☆115Updated 5 months ago
- ☆87Updated 2 years ago
- ☆316Updated 3 years ago
- Configure Python functions explicitly and safely☆126Updated last month
- Prune a model while finetuning or training.☆397Updated 2 years ago
- Code for the paper "Are Sixteen Heads Really Better than One?"☆170Updated 4 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 3 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset☆123Updated 5 years ago
- XtremeDistil framework for distilling/compressing massive multilingual neural network models to tiny and efficient models for AI at scale☆154Updated last year
- Library for 8-bit optimizers and quantization routines.☆717Updated 2 years ago
- Neural Text Generation with Unlikelihood Training☆310Updated 3 years ago
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- Tracking the progress in non-autoregressive generation (translation, transcription, etc.)☆306Updated last year
- ☆213Updated 4 years ago
- ☆335Updated 9 months ago