gebob19 / rl_with_jax
clear single-file JAX implementations of common RL algorithms
☆16Updated 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 Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 2 years ago
- ☆41Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆64Updated 7 months ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆17Updated 4 years ago
- ☆46Updated 2 years ago
- Advantage weighted Actor Critic for Offline RL☆48Updated 2 years ago
- Simple maze environments using mujoco-py☆54Updated last year
- ☆54Updated 11 months ago
- Corax: Core RL in JAX☆36Updated 11 months ago
- ☆30Updated 5 years ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆44Updated 4 years ago
- The official repository of Decoupled Reinforcement Learning to Stabilise Intrinsically-Motivated Exploration" (AAMAS 2022)☆27Updated 2 years ago
- V-MPO torch version with DMLab30 and GTrXL☆12Updated 3 years ago
- Code for "Hierarchical Skills for Efficient Exploration" HSD-3 Algorithm and Baselines☆49Updated 2 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆102Updated 2 years ago
- We investigate the effect of populations on finding good solutions to the robust MDP☆28Updated 3 years ago
- ☆35Updated 2 years ago
- Conservative Q learning in Jax☆52Updated last year
- Various reinforcement learning algorithms written in Jax + Flax☆23Updated last year
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Deep Learning Project☆20Updated 5 years ago
- ☆17Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆107Updated 2 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆85Updated 3 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆138Updated last year
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆60Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆67Updated 6 months ago
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago
- PyTorch IMPALA implementation☆25Updated 5 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆12Updated 6 months ago