SuReLI / RRLS
Robust Reinforcement Learning Suite
☆18Updated 3 months ago
Related projects: ⓘ
- JAX implementation of RL algorithms and vectorized environments☆32Updated 8 months ago
- Modular Single-file Reinfocement Learning Algorithms Library☆37Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆53Updated 3 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆33Updated last week
- Baselines for gymnax 🤖☆57Updated last year
- Various reinforcement learning algorithms written in Jax + Flax☆21Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 2 years ago
- ☆41Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆58Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- ☆17Updated last year
- Official PyTorch implementation of "Rethinking Value Function Learning for Generalization in Reinforcement Learning" (NeurIPS 2022)☆14Updated last year
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆46Updated 10 months ago
- ☆25Updated last week
- The Starcraft Multi-Agent challenge lite☆32Updated last week
- Vectorization techniques for fast population-based training.☆52Updated 2 years ago
- A lightweight RL library inspired from salina☆13Updated 2 weeks ago
- An Open-Ended Agentic Simulator☆17Updated last month
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆16Updated 8 months ago
- Docker containers of baseline agents for the Crafter environment☆27Updated 2 years ago
- ☆37Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆96Updated 2 years ago
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆41Updated 2 months ago
- Rainbow DQN implementation accompanying the paper "Fast and Data-Efficient Training of Rainbow" which reaches 205.7 median HNS after 10M …☆43Updated 2 years ago
- ☆33Updated last year
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning☆68Updated last month
- ☆56Updated 3 weeks ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆76Updated last year
- ☆41Updated 5 months ago