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 Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Updated 2 years ago
- ☆107Updated 2 years ago
- ☆100Updated last year
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆130Updated this week
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆77Updated 7 months ago
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆33Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- ☆51Updated last year
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆56Updated last year
- ☆141Updated last year
- ☆49Updated last year
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆67Updated 11 months ago
- ☆57Updated 11 months ago
- Stick-breaking attention☆60Updated 2 months ago
- [ICLR 2023] Official implementation of Transnormer in our ICLR 2023 paper - Toeplitz Neural Network for Sequence Modeling☆80Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆53Updated 9 months ago
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- ☆34Updated 6 months ago
- Official code for the paper "Attention as a Hypernetwork"☆42Updated last year
- ☆32Updated last year
- NF-Layers for constructing neural functionals.☆88Updated last year
- Accelerated First Order Parallel Associative Scan☆188Updated last year
- Sparse Backpropagation for Mixture-of-Expert Training☆30Updated last year
- Code for the paper "The Impact of Positional Encoding on Length Generalization in Transformers", NeurIPS 2023☆137Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated last year
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆94Updated 3 years ago
- [ICML 2022] Latent Diffusion Energy-Based Model for Interpretable Text Modeling☆66Updated 3 years ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆125Updated 6 months ago