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
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆90Updated this week
- ☆94Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated last year
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆47Updated 2 months ago
- ☆82Updated last year
- ☆127Updated last year
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆99Updated 2 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆57Updated last year
- ☆51Updated 10 months ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆123Updated last year
- ☆61Updated 3 years ago
- ☆51Updated 2 years ago
- ☆30Updated 4 months ago
- ☆20Updated 2 months ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆65Updated 6 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- NF-Layers for constructing neural functionals.☆82Updated last year
- Sequence Modeling with Structured State Spaces☆63Updated 2 years ago
- ☆37Updated last year
- ☆32Updated last year
- [ICML 2022] Latent Diffusion Energy-Based Model for Interpretable Text Modeling☆65Updated 2 years ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆72Updated 3 months ago
- Differentiable Top-k Classification Learning☆80Updated 2 years ago
- Stick-breaking attention☆50Updated last month
- ☆22Updated 3 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated 2 years ago
- ☆81Updated 8 months ago