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
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆101Updated 2 years ago
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago
- ☆41Updated last year
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- Simple maze environments using mujoco-py☆54Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆69Updated 10 months ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- Advantage weighted Actor Critic for Offline RL☆50Updated 2 years ago
- The official repository of Decoupled Reinforcement Learning to Stabilise Intrinsically-Motivated Exploration" (AAMAS 2022)☆27Updated 3 years ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆44Updated 4 years ago
- ☆17Updated 2 years ago
- ☆47Updated 2 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated last year
- ☆14Updated 3 years ago
- ☆53Updated last year
- ☆42Updated 3 years ago
- Conservative Q learning in Jax☆53Updated 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
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- DecentralizedLearning☆24Updated 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
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆62Updated 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
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated last year
- Deep Learning Project☆21Updated 5 years ago
- Recurrent continuous reinforcement learning algorithms implemented in Pytorch.☆50Updated 3 years ago
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆74Updated 2 years ago
- Implementation of Multi-Game Decision Transformers in PyTorch☆46Updated 2 years ago