This is a Pytorch implementation of Reformer. It includes LSH attention, reversible network, and chunking. It has been validated with an auto-regressive task (enwik8).
Features
- Positional Embeddings
- Masking
- Documentation available
- Examples available
- Reformer Encoder Decoder Architecture
- Product Key Memory
- Customize Feedforward
- Helpers for training auto-regressive models
Categories
Machine LearningLicense
MIT LicenseFollow Reformer PyTorch
Other Useful Business Software
$300 Free Credits to Build on Google Cloud
Put your $300 in credit toward real workloads, then keep building with free monthly usage for 20+ products. No commitment and no charge until you upgrade.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of Reformer PyTorch!