gebob19 / rl_with_jax
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:
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated last year
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆18Updated 4 years ago
- ☆47Updated 2 years ago
- ☆42Updated 2 years ago
- Unofficial re-implementation of "Learning Latent Dynamics for Planning from Pixels" (https://arxiv.org/abs/1811.04551 ) with PyTorch☆47Updated 4 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated 2 years ago
- ☆36Updated 2 years ago
- Code for Latent Action Space for Offline Reinforcement Learning [CoRL 2020]☆52Updated 3 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆74Updated 11 months ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆46Updated 4 years ago
- We investigate the effect of populations on finding good solutions to the robust MDP☆28Updated 4 years ago
- Simple maze environments using mujoco-py☆54Updated last year
- 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
- NeurIPS Reproducibility Challenge 2019☆20Updated 5 years ago
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago
- ☆53Updated last year
- Deep Learning Project☆22Updated 5 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 3 years ago
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆40Updated 6 months ago
- Code for "Hierarchical Skills for Efficient Exploration" HSD-3 Algorithm and Baselines☆50Updated 2 years ago
- ☆17Updated 2 years ago
- ☆15Updated 2 years ago
- Implementation of the skill discovery algorithm described in ICLR submission "Option Discovery using Deep Skill Chaining"☆28Updated 5 years ago
- The official repository of Decoupled Reinforcement Learning to Stabilise Intrinsically-Motivated Exploration" (AAMAS 2022)☆27Updated 3 years ago
- ☆42Updated 3 years ago
- ☆17Updated 3 years ago
- DecentralizedLearning☆24Updated 2 years ago
- ☆31Updated 2 years ago
- Advantage weighted Actor Critic for Offline RL☆50Updated 2 years ago