microsoft / infinibatchLinks
Efficient, check-pointed data loading for deep learning with massive data sets.
☆209Updated 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☆330Updated 3 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆102Updated 4 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆133Updated last year
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 4 years ago
- Training Transformer-XL on 128 GPUs☆141Updated 5 years ago
- ☆219Updated 5 years ago
- A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!☆113Updated 2 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆314Updated 2 years ago
- ☆48Updated 5 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆376Updated 4 years ago
- LM Pretraining with PyTorch/TPU☆136Updated 5 years ago
- FairSeq repo with Apollo optimizer☆114Updated last year
- Code for the paper "Are Sixteen Heads Really Better than One?"☆172Updated 5 years ago
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆432Updated 3 years ago
- Cascaded Text Generation with Markov Transformers☆129Updated 2 years ago
- Profile the GPU memory usage of every line in a Pytorch code☆83Updated 7 years ago
- Fast Block Sparse Matrices for Pytorch☆547Updated 4 years ago
- Method to improve inference time for BERT. This is an implementation of the paper titled "PoWER-BERT: Accelerating BERT Inference via Pro…☆62Updated 3 weeks ago
- LaNMT: Latent-variable Non-autoregressive Neural Machine Translation with Deterministic Inference☆79Updated 4 years ago
- Implementation of a Transformer, but completely in Triton☆275Updated 3 years ago
- ☆44Updated 5 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆301Updated 4 months ago
- Configure Python functions explicitly and safely