microsoft / infinibatchLinks
Efficient, check-pointed data loading for deep learning with massive data sets.
☆210Updated 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☆332Updated 3 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆134Updated last year
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆102Updated 5 years ago
- 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
- Profile the GPU memory usage of every line in a Pytorch code☆83Updated 7 years ago
- ☆219Updated 5 years ago
- Code for the paper "Are Sixteen Heads Really Better than One?"☆173Updated 5 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 2 months ago
- LM Pretraining with PyTorch/TPU☆136Updated 6 years ago
- ☆48Updated 5 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆316Updated 2 years ago
- FairSeq repo with Apollo optimizer☆114Updated last year
- Implementation of https://arxiv.org/abs/1904.00962☆377Updated 5 years ago
- Configure Python functions explicitly and safely☆128Updated last year
- ☆65Updated 5 years ago
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆433Updated 3 years ago
- Implementation of a Transformer, but completely in Triton☆277Updated 3 years ago
- Fast Block Sparse Matrices for Pytorch☆550Updated 4 years ago
- A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!☆113Updated 2 years ago
- This repository contains the code for running the character-level Sandwich Transformers from our ACL 2020 paper on Improving Transformer …☆55Updated 4 years ago
- This repository contains code to replicate the no-longer publicly available Toronto BookCorpus dataset☆49Updated 3 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆301Updated 6 months ago
- Cascaded Text Generation with Markov Transformers☆129Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆250Updated 3 years ago
- Hyperparameter Search for AllenNLP☆140Updated 9 months ago
- PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset☆123Updated 6 years ago
- ☆87Updated 3 years ago
- Neural Text Generation with Unlikelihood Training☆310Updated 4 years ago
- LaNMT: Latent-variable Non-autoregressive Neural Machine Translation with Deterministic Inference☆79Updated 4 years ago