microsoft / ReinMaxLinks
Beyond Straight-Through
☆102Updated 2 years ago
Alternatives and similar repositories for ReinMax
Users that are interested in ReinMax are comparing it to the libraries listed below
Sorting:
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- ☆106Updated 2 years ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆124Updated 2 weeks ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Updated 2 years ago
- ☆51Updated last year
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆33Updated 2 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆126Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆67Updated 11 months ago
- ☆138Updated last year
- ☆49Updated last year
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆73Updated 6 months ago
- ☆98Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆94Updated 3 years ago
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- Curse-of-memory phenomenon of RNNs in sequence modelling☆19Updated 3 months ago
- Official code for the paper "Attention as a Hypernetwork"☆40Updated last year
- Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"☆120Updated last year
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆66Updated last year
- Standalone Product Key Memory module in Pytorch - for augmenting Transformer models☆82Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- Sparse Backpropagation for Mixture-of-Expert Training☆30Updated last year
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆62Updated 5 months ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆56Updated last year
- ☆56Updated 10 months ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆81Updated 2 years ago
- NF-Layers for constructing neural functionals.☆88Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆123Updated 6 months ago