gebob19 / rl_with_jaxLinks
clear single-file JAX implementations of common RL algorithms
☆17Updated 3 years ago
Alternatives and similar repositories for rl_with_jax
Users that are interested in rl_with_jax are comparing it to the libraries listed below
Sorting:
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 3 years ago
- Benchmarking RL generalization in an interpretable way.☆157Updated last month
- ☆42Updated 2 years ago
- Simple maze environments using mujoco-py☆57Updated last year
- Author's PyTorch implementation of TD7 for online and offline RL☆144Updated last year
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆47Updated 4 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆115Updated 3 years ago
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆48Updated last year
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated last year
- ☆54Updated last year
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆87Updated 4 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- Code for MOPO: Model-based Offline Policy Optimization☆179Updated 3 years ago
- Representation Learning for RL☆126Updated 2 years ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆18Updated 4 years ago
- Deep Learning Project☆22Updated 5 years ago
- Benchmarked implementations of Offline RL Algorithms.☆74Updated 4 months ago
- Author's PyTorch implementation of Randomized Ensembled Double Q-Learning (REDQ) algorithm.☆170Updated 8 months ago
- ☆47Updated 2 years ago
- We investigate the effect of populations on finding good solutions to the robust MDP☆28Updated 4 years ago
- The Starcraft Multi-Agent challenge lite☆41Updated 10 months ago
- Learning Laplacian Representations in Reinforcement Learning☆16Updated 4 years ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- Partially Observable Process Gym☆193Updated last month
- ☆31Updated 6 years ago
- ☆56Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆145Updated 2 years ago
- Author's PyTorch implementation of LAP and PAL with TD3 and DDQN☆36Updated 3 years ago
- A pytorch reprelication of the model-based reinforcement learning algorithm MBPO☆170Updated 3 years ago
- Model-Based Offline Reinforcement Learning☆50Updated 4 years ago