microsoft / infinibatchLinks
Efficient, check-pointed data loading for deep learning with massive data sets.
☆208Updated 2 years ago
Alternatives and similar repositories for infinibatch
Users that are interested in infinibatch are comparing it to the libraries listed below
Sorting:
- Understanding the Difficulty of Training Transformers☆329Updated 3 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆132Updated last year
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆313Updated last year
- ☆218Updated 5 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆376Updated 4 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 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
- Training Transformer-XL on 128 GPUs☆140Updated 5 years ago
- A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!☆112Updated 2 years ago
- LM Pretraining with PyTorch/TPU☆134Updated 5 years ago
- Fast Block Sparse Matrices for Pytorch☆548Updated 4 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- Code for the paper "Are Sixteen Heads Really Better than One?"☆172Updated 5 years ago
- FairSeq repo with Apollo optimizer☆114Updated last year
- Slicing a PyTorch Tensor Into Parallel Shards☆299Updated last month
- Cascaded Text Generation with Markov Transformers☆129Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆270Updated 3 years ago
- FastFormers - highly efficient transformer models for NLU☆705Updated 3 months ago
- Profile the GPU memory usage of every line in a Pytorch code☆83Updated 7 years ago
- ☆358Updated last year
- Neural Text Generation with Unlikelihood Training☆309Updated 3 years ago
- Code for the Shortformer model, from the ACL 2021 paper by Ofir Press, Noah A. Smith and Mike Lewis.☆147Updated 3 years ago
- ☆64Updated 5 years ago
- Configure Python functions explicitly and safely☆126Updated 7 months ago
- ☆245Updated 5 years ago
- ☆47Updated 5 years ago
- Prune a model while finetuning or training.☆403Updated 3 years ago
- LaNMT: Latent-variable Non-autoregressive Neural Machine Translation with Deterministic Inference☆80Updated 3 years ago
- XtremeDistil framework for distilling/compressing massive multilingual neural network models to tiny and efficient models for AI at scale☆155Updated last year