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

Project Samples

Project Activity

See All Activity >

Categories

Machine Learning

License

MIT License

Follow Reformer PyTorch

Reformer PyTorch Web Site

Other Useful Business Software
Ship Agents Faster Icon
Ship Agents Faster

Transform your applications and workflows into powerful agentic systems at global scale.

Gemini Enterprise Agent Platform lets you rapidly build, scale, govern and optimize production-ready agents grounded in your organization's data. The platform enables developers to build custom or pre-built agents for virtually any use case. New customers get $300 in free credits.
Get Started Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of Reformer PyTorch!

Additional Project Details

Operating Systems

Linux, Mac, Windows

Programming Language

Python

Related Categories

Python Machine Learning Software

Registered

2024-08-12