gebob19 / rl_with_jaxLinks
clear single-file JAX implementations of common RL algorithms
☆17Updated 4 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:
- ☆47Updated 3 weeks ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆103Updated 3 years ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆68Updated last year
- Benchmarked implementations of Offline RL Algorithms.☆76Updated 9 months ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆40Updated 9 months ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆48Updated 5 years ago
- Benchmarking RL generalization in an interpretable way.☆170Updated 3 weeks ago
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated 2 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆78Updated 3 years ago
- ☆51Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆119Updated 3 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆92Updated 4 years ago
- ☆18Updated 3 years ago
- Code for Model-Free Opponent Shaping (ICML 2022)☆20Updated 3 years ago
- Author's PyTorch implementation of Randomized Ensembled Double Q-Learning (REDQ) algorithm.☆176Updated last year
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆69Updated 3 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆18Updated last year
- IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL☆45Updated last month
- Simple maze environments using mujoco-py☆57Updated last year
- Author's PyTorch implementation of TD7 for online and offline RL☆156Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆81Updated last year
- Deep Learning Project☆23Updated 5 years ago
- Representation Learning for RL☆128Updated 2 years ago
- Fast and flexible multi-agent gridworld reinforcement learning environments.☆46Updated 8 months ago
- Implementation of Multi-Game Decision Transformers in PyTorch☆47Updated 2 years ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆19Updated 5 years ago
- Skeleton for scalable and flexible Jax RL implementations☆92Updated 2 years ago
- A pytorch reprelication of the model-based reinforcement learning algorithm MBPO☆182Updated 3 years ago
- A collection of RL algorithms written in JAX.☆104Updated 3 years ago