schmidtdominik / Rainbow
Rainbow DQN implementation accompanying the paper "Fast and Data-Efficient Training of Rainbow" which reaches 205.7 median HNS after 10M frames. π
β43Updated 2 years ago
Related projects β
Alternatives and complementary repositories for Rainbow
- β29Updated 3 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β49Updated 2 years ago
- Scalable Opponent Shaping Experiments in JAXβ21Updated 7 months ago
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.β12Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"β57Updated 5 months ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ46Updated last year
- Docker containers of baseline agents for the Crafter environmentβ28Updated 2 years ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learningβ13Updated 2 years ago
- Simple JAX Graphics Library.β23Updated 3 weeks ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objectiveβ79Updated last year
- Accelerated replay buffers in JAXβ40Updated 2 years ago
- β41Updated 2 months ago
- AGAC: Adversarially Guided Actor-Criticβ47Updated 3 years ago
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"β21Updated last year
- Implementation of Tactical Optimistic and Pessimistic value estimationβ25Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)β59Updated last year
- Skeleton for scalable and flexible Jax RL implementationsβ63Updated last year
- β36Updated last year
- Vectorization techniques for fast population-based training.β54Updated 2 years ago
- β38Updated last year
- β29Updated last year
- β42Updated last year
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the β¦β83Updated 3 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β40Updated last week
- Repository for the QDgym code. A framework for Quality Diversity optimization benchmark tasks based OpenAI Gym.β21Updated 3 years ago
- β38Updated 3 years ago
- Modular Single-file Reinfocement Learning Algorithms Libraryβ37Updated last year
- Official code for "Pretraining Representations For Data-Efficient Reinforcement Learning" (NeurIPS 2021)β52Updated 3 years ago
- Repository for the PGA-MAP-Elites algorithm. PGA-MAP-Elites was developed to efficiently scale MAP-Elites to large genotypes and noisy dβ¦β35Updated 3 years ago
- Simplistic Pytorch Implementation of the Dreamer-RLβ20Updated 2 years ago