microsoft / ReinMax
Beyond Straight-Through
☆93Updated last year
Alternatives and similar repositories for ReinMax:
Users that are interested in ReinMax are comparing it to the libraries listed below
- ☆78Updated last year
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆97Updated last year
- ☆30Updated 9 months ago
- Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆28Updated 2 months ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆58Updated 3 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- ☆123Updated 10 months ago
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆91Updated 4 months ago
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆44Updated last year
- Stick-breaking attention☆41Updated this week
- Sparse Backpropagation for Mixture-of-Expert Training☆27Updated 6 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆121Updated last year
- ☆78Updated 10 months ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆26Updated 3 years ago
- ☆51Updated 7 months ago
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆47Updated 8 months ago
- ☆51Updated 7 months ago
- NF-Layers for constructing neural functionals.☆79Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆61Updated 5 months ago
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆46Updated last year
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆57Updated last year
- [ICML 2022] Latent Diffusion Energy-Based Model for Interpretable Text Modeling☆64Updated 2 years ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model