subho406 / Recurrent-Linear-Transformers
Implementation of Recurrent Linear Transformers in Jax+Flax.
☆14Updated 10 months ago
Related projects: ⓘ
- ☆25Updated 5 months ago
- ☆28Updated this week
- ☆24Updated 2 months ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆13Updated last year
- Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spaces, NeurIPS 2021☆11Updated 2 years ago
- Parallel Associative Scan for Language Models☆16Updated 8 months ago
- Efficient PScan implementation in PyTorch☆15Updated 8 months ago
- ☆12Updated 8 months ago
- ☆42Updated 7 months ago
- ☆16Updated this week
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆26Updated 2 years ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Updated 6 years ago
- ☆23Updated 10 months ago
- ☆13Updated 7 months ago
- Curse-of-memory phenomenon of RNNs in sequence modelling☆17Updated this week
- Implements the Messenger environment and EMMA model.☆22Updated last year
- Generalised UDRL☆37Updated 2 years ago
- ☆17Updated 3 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆35Updated last month
- Code for our TMLR paper "Distributional GFlowNets with Quantile Flows".☆9Updated 7 months ago
- Parallelizing non-linear sequential models over the sequence length☆40Updated last month
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆17Updated 10 months ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆11Updated 4 years ago
- Recursive Bayesian Networks☆11Updated last week
- PyTorch Package For Quasimetric Learning☆38Updated last year
- ☆22Updated 10 months ago
- ☆24Updated 2 weeks ago
- ☆14Updated last month
- Dateset Reset Policy Optimization☆27Updated 5 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆60Updated 4 months ago