proroklab / ffm
Reinforcement Learning with Fast and Forgetful Memory
☆25Updated last year
Alternatives and similar repositories for ffm:
Users that are interested in ffm are comparing it to the libraries listed below
- ☆47Updated 2 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated last year
- General Modules for JAX☆64Updated last month
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- ☆35Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆48Updated this week
- ☆74Updated last week
- Contains JAX implementation of algorithms for inverse reinforcement learning☆71Updated 7 months ago
- Skeleton for scalable and flexible Jax RL implementations☆74Updated last year
- Baselines for gymnax 🤖☆66Updated last year
- ☆41Updated 8 months ago
- Goal-Conditioned Reinforcement Learning with JAX☆134Updated this week
- Simple single-file baselines for Q-Learning in pure-GPU setting☆150Updated last week
- ExORL: Exploratory Data for Offline Reinforcement Learning☆111Updated 3 years ago
- Conservative Q learning in Jax☆53Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆99Updated last year
- ☆73Updated 4 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆66Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆141Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆67Updated 9 months ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆39Updated this week
- A toolkit for auto-generation of OpenAI Gym environments from RDDL description files.☆78Updated this week
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆78Updated 3 months ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆80Updated 11 months ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆138Updated 3 months ago
- ☆44Updated last year
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …☆35Updated 10 months ago
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆80Updated last month