gebob19 / rl_with_jaxLinks
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:
- ☆43Updated 2 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 3 years ago
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆18Updated 4 years ago
- Implementation of Tactical Optimistic and Pessimistic value estimation☆26Updated last year
- Deep Learning Project☆22Updated 5 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆74Updated last year
- ☆47Updated 2 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆86Updated 3 years ago
- ☆17Updated last year
- Advantage weighted Actor Critic for Offline RL☆50Updated 2 years ago
- ☆53Updated last year
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- Model-Based Offline Reinforcement Learning☆50Updated 4 years ago
- Skeleton for scalable and flexible Jax RL implementations☆82Updated last year
- Simple maze environments using mujoco-py☆56Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- ☆31Updated 5 years ago
- PyTorch implementation of (Deep) Reinforcement Learning (RL) algorithms☆22Updated 2 years ago
- Unofficial re-implementation of "Learning Latent Dynamics for Planning from Pixels" (https://arxiv.org/abs/1811.04551 ) with PyTorch☆47Updated 5 years ago
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆76Updated 2 years ago
- A collection of RL algorithms written in JAX.☆98Updated 2 years ago
- The official repository of Decoupled Reinforcement Learning to Stabilise Intrinsically-Motivated Exploration" (AAMAS 2022)☆27Updated 3 years ago
- Conservative Q learning in Jax☆54Updated 2 years ago
- ☆17Updated 2 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆71Updated 11 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆53Updated 3 weeks ago
- Synchronized Curriculum Learning for RL Agents☆45Updated 2 months ago
- ☆45Updated 2 years ago