google-deepmind / csuite
☆41Updated last month
Related projects ⓘ
Alternatives and complementary repositories for csuite
- Docker containers of baseline agents for the Crafter environment☆28Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- ☆36Updated last year
- Simple JAX Graphics Library.☆23Updated 2 weeks ago
- ☆56Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated last year
- Baselines for gymnax 🤖☆60Updated last year
- General Modules for JAX☆58Updated 3 months ago
- Change-Based Exploration Transfer☆36Updated 2 years ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆78Updated 2 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Object Centric Atari games☆48Updated this week
- Skeleton for scalable and flexible Jax RL implementations☆63Updated last year
- ExORL: Exploratory Data for Offline Reinforcement Learning☆105Updated 2 years ago
- Vectorization techniques for fast population-based training.☆54Updated 2 years ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆12Updated 2 years ago
- Proto-RL: Reinforcement Learning with Prototypical Representations☆82Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆133Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …☆35Updated 6 months ago
- Fast reinforcement learning research☆55Updated 4 months ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆14Updated last year
- An implementation of MuZero in JAX.☆53Updated 2 years ago
- ☆29Updated 3 years ago
- ☆28Updated last year
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆42Updated 4 months ago
- ☆42Updated 2 years ago
- ☆42Updated last year