gebob19 / rl_with_jaxLinks
clear single-file JAX implementations of common RL algorithms
☆16Updated 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:
- ☆48Updated 2 months ago
- Benchmarking RL generalization in an interpretable way.☆173Updated 2 months ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆103Updated 3 years ago
- Representation Learning for RL☆130Updated 2 years ago
- Fast and flexible multi-agent gridworld reinforcement learning environments.☆48Updated 10 months ago
- IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL☆45Updated 3 months ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆40Updated 11 months ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆20Updated 5 years ago
- Code for MOPO: Model-based Offline Policy Optimization☆191Updated 3 years ago
- Partially Observable Process Gym☆211Updated 7 months ago
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated 2 years ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆22Updated 2 years ago
- Benchmarked implementations of Offline RL Algorithms.☆76Updated 10 months ago
- Code for Latent Action Space for Offline Reinforcement Learning [CoRL 2020]☆53Updated 4 years ago
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆79Updated 3 years ago
- LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization☆38Updated 3 years ago
- Author's PyTorch implementation of Randomized Ensembled Double Q-Learning (REDQ) algorithm.☆176Updated last year
- Code for Model-Free Opponent Shaping (ICML 2022)☆20Updated 3 years ago
- ☆52Updated 3 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆62Updated 3 weeks ago
- A collection of RL algorithms written in JAX.☆104Updated 3 years ago
- Offline Risk-Averse Actor-Critic (O-RAAC). A model-free RL algorithm for risk-averse RL in a fully offline setting☆35Updated 4 years ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆48Updated 5 years ago
- Model-Based Offline Reinforcement Learning☆52Updated 5 years ago
- ☆59Updated 2 years ago
- Author's PyTorch implementation of TD7 for online and offline RL☆162Updated 2 years ago
- The Starcraft Multi-Agent challenge lite☆46Updated last year
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- Learning Laplacian Representations in Reinforcement Learning☆18Updated 5 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19Updated last year