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:
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 3 years ago
- ☆44Updated 2 years ago
- Benchmarking RL generalization in an interpretable way.☆161Updated 3 months ago
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated 2 years ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆48Updated 4 years ago
- ☆46Updated 2 years ago
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago
- Code for Model-Free Opponent Shaping (ICML 2022)☆19Updated 2 years ago
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆76Updated 2 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
- Benchmarked implementations of Offline RL Algorithms.☆75Updated 6 months ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆37Updated 7 months ago
- Simple maze environments using mujoco-py☆56Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆86Updated 2 years ago
- LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization☆35Updated 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
- Representation Learning for RL☆126Updated 2 years ago
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆48Updated last year
- The Starcraft Multi-Agent challenge lite☆40Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆79Updated last year
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆21Updated last year
- Safe Model-based Reinforcement Learning with Robust Cross-Entropy Method☆66Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆115Updated 3 years ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆65Updated last year
- Fast and flexible multi-agent gridworld reinforcement learning environments.☆44Updated 5 months ago
- Github repo for HIDIO: Hierarchical Reinforcement Learning by Discovering Intrinsic Options☆46Updated 3 years ago
- IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL☆43Updated this week
- Unofficial re-implementation of "Learning Latent Dynamics for Planning from Pixels" (https://arxiv.org/abs/1811.04551 ) with PyTorch☆47Updated 5 years ago
- Code for MOPO: Model-based Offline Policy Optimization☆188Updated 3 years ago