google-deepmind / zipfian_environments
☆28Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for zipfian_environments
- Docker containers of baseline agents for the Crafter environment☆28Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- ☆41Updated last month
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …☆35Updated 6 months ago
- Accelerated replay buffers in JAX☆40Updated 2 years ago
- Vectorization techniques for fast population-based training.☆54Updated 2 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 years ago
- Reinforcement Learning with Latent Flow☆43Updated 3 years ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆63Updated 3 months ago
- An implementation of MuZero in JAX.☆53Updated 2 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- Baselines for gymnax 🤖☆60Updated last year
- Proto-RL: Reinforcement Learning with Prototypical Representations☆82Updated 2 years ago
- GPT implementation in Flax☆18Updated 2 years ago
- ☆17Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"☆42Updated last year
- A2C is a special case of PPO!☆19Updated 2 years ago
- General Modules for JAX☆58Updated 3 months ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆83Updated 9 months ago
- flexible meta-learning in jax☆12Updated last year
- Fast reinforcement learning research☆55Updated 4 months ago
- ☆17Updated 5 months ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- ☆36Updated last year
- A collection of RL algorithms written in JAX.☆95Updated 2 years ago
- Scalable Opponent Shaping Experiments in JAX☆21Updated 7 months ago
- An Open-Ended Agentic Simulator☆28Updated 3 months ago