microsoft / infinibatch
Efficient, check-pointed data loading for deep learning with massive data sets.
☆205Updated last year
Related projects ⓘ
Alternatives and complementary repositories for infinibatch
- Understanding the Difficulty of Training Transformers☆328Updated 2 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆309Updated last year
- Code for the paper "Are Sixteen Heads Really Better than One?"☆169Updated 4 years ago
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆433Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆249Updated 2 years ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆100Updated 4 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆130Updated 5 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆296Updated 3 years ago
- ☆315Updated 3 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 3 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆369Updated 3 years ago
- FastFormers - highly efficient transformer models for NLU☆701Updated 10 months ago
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆361Updated 7 months ago
- LM Pretraining with PyTorch/TPU☆132Updated 5 years ago
- Torch Distributed Experimental☆116Updated 3 months ago
- PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset☆123Updated 5 years ago
- Neural Text Generation with Unlikelihood Training☆310Updated 3 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 4 years ago
- Hyperparameter Search for AllenNLP☆134Updated 4 years ago
- PyTorch implementation of L2L execution algorithm☆106Updated last year
- FairSeq repo with Apollo optimizer☆107Updated 11 months ago
- ☆212Updated 4 years ago
- ☆47Updated 4 years ago
- Flexible components pairing 🤗 Transformers with Pytorch Lightning☆611Updated 2 years ago
- XtremeDistil framework for distilling/compressing massive multilingual neural network models to tiny and efficient models for AI at scale☆153Updated 11 months ago
- Prune a model while finetuning or training.☆394Updated 2 years ago
- Block-sparse primitives for PyTorch☆148Updated 3 years ago
- MPNet: Masked and Permuted Pre-training for Language Understanding https://arxiv.org/pdf/2004.09297.pdf☆288Updated 3 years ago