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:
- Keras implementation of Legendre Memory Units☆214Nov 21, 2025Updated 3 months ago
- 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
- ☆29Jul 9, 2024Updated last year
- ☆13Apr 25, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 3 months ago
- Gym env for Slay the Spire☆17Dec 31, 2024Updated last year
- Примеры пропозалов для подачи заявки в Open.TLab☆28Dec 15, 2022Updated 3 years ago
- ☆12May 18, 2022Updated 3 years ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆20Oct 21, 2025Updated 4 months ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Aug 31, 2020Updated 5 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 3 months ago
- Behavioural cloning solution to MineRL2020 competition☆18Mar 6, 2021Updated 5 years ago
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆21Oct 6, 2021Updated 4 years ago
- A benchmark to test and compare your pcg algorithm against each other☆22Sep 22, 2025Updated 5 months ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆21May 18, 2023Updated 2 years ago
- Neurorobotics 2020☆18Aug 14, 2023Updated 2 years ago
- ☆19Mar 1, 2023Updated 3 years ago
- Resources for the paper titled "Evaluating Latent Space Robustness and Uncertainty of EEG-ML Models under Realistic Distribution Shifts".…☆19Jul 12, 2023Updated 2 years ago
- ☆51Jan 28, 2024Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- 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
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆31Mar 2, 2026Updated last week
- Official Code for "Relative Entropy Pathwise Policy Optimization"☆46Feb 27, 2026Updated last week
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 2 years ago
- A modular implementation of PPO, and soon hopefully other algorithms.☆26Jan 16, 2024Updated 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
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- Jaxpr Visualisation Tool☆36Dec 22, 2024Updated last year
- Toolkit of Causal Model-based Reinforcement Learning.☆33Jun 5, 2023Updated 2 years ago
- MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research☆39Jul 14, 2025Updated 7 months ago
- Maximum Entropy Population Based Training for Zero-Shot Human-AI Coordination☆26Nov 29, 2022Updated 3 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 6 months ago
- Code for the paper "PALBERT: Teaching ALBERT to Ponder", NeurIPS 2022 Spotlight☆37Apr 8, 2023Updated 2 years ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆81Feb 13, 2025Updated last year