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
- ☆219Updated 5 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 4 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆376Updated 4 years ago
- Code for the paper "Are Sixteen Heads Really Better than One?"☆172Updated 5 years ago
- LM Pretraining with PyTorch/TPU☆136Updated 5 years ago
- A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!☆113Updated 2 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 5 years ago
- ☆48Updated 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
- Profile the GPU memory usage of every line in a Pytorch code☆83Updated 7 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆314Updated 2 years ago
- FairSeq repo with Apollo optimizer☆114Updated last year
- Method to improve inference time for BERT. This is an implementation of the paper titled "PoWER-BERT: Accelerating BERT Inference via Pro…☆62Updated this week
- This repository contains the code for running the character-level Sandwich Transformers from our ACL 2020 paper on Improving Transformer …☆55Updated 4 years ago
- LaNMT: Latent-variable Non-autoregressive Neural Machine Translation with Deterministic Inference☆79Updated 4 years ago
- Cascaded Text Generation with Markov Transformers☆129Updated 2 years ago
- Configure Python functions explicitly and safely☆127Updated 10 months ago
- DeeBERT: Dynamic Early Exiting for Accelerating BERT Inference☆159Updated 3 years ago
- Fast Block Sparse Matrices for Pytorch☆549Updated 4 years ago
- Code for the Shortformer model, from the ACL 2021 paper by Ofir Press, Noah A. Smith and Mike Lewis.☆147Updated 4 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Hyperparameter Search for AllenNLP☆139Updated 6 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆300Updated 3 months ago
- ☆64Updated 5 years ago
- ☆44Updated 5 years ago
- Implementation of a Transformer, but completely in Triton☆274Updated 3 years ago
- This repository contains code to replicate the no-longer publicly available Toronto BookCorpus dataset☆49Updated 3 years ago