microsoft / ReinMax
Beyond Straight-Through
☆94Updated last year
Alternatives and similar repositories for ReinMax:
Users that are interested in ReinMax are comparing it to the libraries listed below
- ☆91Updated last year
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆45Updated last month
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆80Updated this week
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated last year
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆63Updated 6 months ago
- ☆81Updated last year
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆95Updated 7 months ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago
- ☆51Updated 9 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆99Updated 2 years ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 3 months ago
- ☆127Updated last year
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- ☆31Updated 5 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆86Updated 3 years ago
- ☆81Updated 8 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆65Updated 7 months ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆79Updated last year
- ☆60Updated 3 years ago
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆53Updated 3 weeks ago
- ☆45Updated last year
- NF-Layers for constructing neural functionals.☆82Updated last year
- Randomized Positional Encodings Boost Length Generalization of Transformers☆80Updated last year
- Reparameterized Discrete Diffusion Models for Text Generation☆96Updated 2 years ago
- Official Repository of Pretraining Without Attention (BiGS), BiGS is the first model to achieve BERT-level transfer learning on the GLUE …☆116Updated last year