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
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆312Updated last year
- Transformers without Tears: Improving the Normalization of Self-Attention☆131Updated 9 months ago
- Understanding the Difficulty of Training Transformers☆328Updated 2 years ago
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆432Updated 2 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- FairSeq repo with Apollo optimizer☆110Updated last year
- Code for the paper "Are Sixteen Heads Really Better than One?"☆171Updated 4 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 4 years ago
- LM Pretraining with PyTorch/TPU☆134Updated 5 years ago
- FastFormers - highly efficient transformer models for NLU☆704Updated last year
- Configure Python functions explicitly and safely☆126Updated 4 months ago
- Prune a model while finetuning or training.☆400Updated 2 years ago
- XtremeDistil framework for distilling/compressing massive multilingual neural network models to tiny and efficient models for AI at scale☆153Updated last year
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆88Updated 4 years ago
- Fast Block Sparse Matrices for Pytorch☆546Updated 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…☆61Updated 2 years ago
- PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset☆123Updated 5 years ago
- Neural Text Generation with Unlikelihood Training☆309Updated 3 years ago
- [ACL 2022] Structured Pruning Learns Compact and Accurate Models https://arxiv.org/abs/2204.00408☆195Updated last year
- Library for 8-bit optimizers and quantization routines.☆717Updated 2 years ago
- Tracking the progress in non-autoregressive generation (translation, transcription, etc.)☆307Updated 2 years ago
- A masked language modeling objective to train a model to predict any subset of the target words, conditioned on both the input text and a…☆242Updated 3 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Code for the Shortformer model, from the ACL 2021 paper by Ofir Press, Noah A. Smith and Mike Lewis.☆146Updated 3 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆119Updated 3 years ago
- Cascaded Text Generation with Markov Transformers☆129Updated 2 years ago
- Torch Distributed Experimental☆115Updated 7 months ago
- Adaptive Softmax implementation for PyTorch☆80Updated 5 years ago