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"☆102Updated 3 years ago
- ☆45Updated 2 years 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
- Benchmarking RL generalization in an interpretable way.☆166Updated last week
- Unofficial re-implementation of "Learning Latent Dynamics for Planning from Pixels" (https://arxiv.org/abs/1811.04551 ) with PyTorch☆46Updated 5 years ago
- Simple maze environments using mujoco-py☆56Updated last year
- ☆48Updated 2 years ago
- Author's PyTorch implementation of TD7 for online and offline RL☆151Updated 2 years ago
- A collection of RL algorithms written in JAX.☆104Updated 3 years ago
- Benchmarked implementations of Offline RL Algorithms.☆74Updated 7 months ago
- Code for MOPO: Model-based Offline Policy Optimization☆188Updated 3 years ago
- ☆114Updated 2 years ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆20Updated 5 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
- Skeleton for scalable and flexible Jax RL implementations☆88Updated 2 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆76Updated 2 years ago
- ☆31Updated 6 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆57Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆79Updated last year
- Partially Observable Process Gym☆202Updated 4 months ago
- Representation Learning for RL☆127Updated 2 years ago
- ☆43Updated 4 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆116Updated 3 years ago
- Code accompanying the paper Adversarially Trained Actor Critic for Offline Reinforcement Learning by Ching-An Cheng*, Tengyang Xie*, Nan …☆71Updated 2 years ago
- PyTorch implementation of (Deep) Reinforcement Learning (RL) algorithms☆25Updated 3 years ago
- Advantage weighted Actor Critic for Offline RL☆50Updated 3 years ago
- Author's PyTorch implementation of Randomized Ensembled Double Q-Learning (REDQ) algorithm.☆173Updated 11 months ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆38Updated 8 months ago