hrshtv / pytorch-lmuView external linksLinks
A PyTorch implementation of Legendre Memory Units (LMUs) and its FFT variant
☆43Dec 22, 2021Updated 4 years ago
Alternatives and similar repositories for pytorch-lmu
Users that are interested in pytorch-lmu are comparing it to the libraries listed below
Sorting:
- Implementation of the Legendre Memory Unit in PyTorch☆22Dec 17, 2019Updated 6 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Jan 16, 2022Updated 4 years ago
- ☆13Apr 25, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 2 months ago
- ☆16Jul 16, 2024Updated last year
- Griffin MQA + Hawk Linear RNN Hybrid☆88Apr 26, 2024Updated last year
- ☆12May 18, 2022Updated 3 years ago
- Hidden Engrams: Long Term Memory for Transformer Model Inference☆35Jun 26, 2021Updated 4 years ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆20Oct 21, 2025Updated 3 months ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Aug 31, 2020Updated 5 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 2 months ago
- Behavioural cloning solution to MineRL2020 competition☆18Mar 6, 2021Updated 4 years ago
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆20Oct 6, 2021Updated 4 years ago
- Neurorobotics 2020☆18Aug 14, 2023Updated 2 years ago
- [PAMI 2021] Gating Revisited: Deep Multi-layer RNNs That Can Be Trained☆18Dec 3, 2021Updated 4 years ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Feb 9, 2023Updated 3 years ago
- ☆19Mar 1, 2023Updated 2 years ago
- ☆51Jan 28, 2024Updated 2 years ago
- A minimal home grid world environment to evaluate language understanding in interactive agents.☆24Sep 6, 2023Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- Code for the experiments done in the paper "GAN-Aimbots: Using Machine Learning for Cheating in First Person Shooters"☆24May 13, 2022Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- ☆35Sep 16, 2025Updated 5 months ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆29Updated this week
- Official Code for "Relative Entropy Pathwise Policy Optimization"☆42Jan 17, 2026Updated 3 weeks ago
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 2 years ago
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …☆35May 21, 2024Updated last year
- A modular implementation of PPO, and soon hopefully other algorithms.☆26Jan 16, 2024Updated 2 years ago
- Jaxpr Visualisation Tool☆35Dec 22, 2024Updated last year
- ☆33May 15, 2024Updated last year
- MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research☆39Jul 14, 2025Updated 7 months ago
- Toolkit of Causal Model-based Reinforcement Learning.☆33Jun 5, 2023Updated 2 years ago
- CARMA Streets is a component of CARMA ecosystem, which enables such a coordination among different transportation users. This component p…☆11Aug 21, 2025Updated 5 months ago
- Maximum Entropy Population Based Training for Zero-Shot Human-AI Coordination☆26Nov 29, 2022Updated 3 years ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- Code for the paper "PALBERT: Teaching ALBERT to Ponder", NeurIPS 2022 Spotlight☆37Apr 8, 2023Updated 2 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆71Jul 17, 2025Updated 6 months ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆81Feb 13, 2025Updated last year
- Official implementation for the paper "Offline Meta RL - Identifiability Challenges and Effective Data Collection Strategies", NeurIPS 20…☆31Nov 23, 2021Updated 4 years ago