AutonomousAgentsLab / cr-dv3
DreamerV3 implementation of Curious Replay, a method for prioritizing experience replay that is tailored to model-based reinforcement learning agents.
☆36Updated last year
Alternatives and similar repositories for cr-dv3:
Users that are interested in cr-dv3 are comparing it to the libraries listed below
- ☆66Updated 4 months ago
- Simple JAX Graphics Library.☆29Updated 2 months ago
- ☆31Updated 9 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆54Updated 9 months ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- Baselines for gymnax 🤖☆61Updated last year
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …☆35Updated 7 months ago
- POPGym Library in JAX☆11Updated 9 months ago
- Highly scalable 2D JAX physics engine.☆42Updated last week
- Corax: Core RL in JAX☆36Updated 10 months ago
- Skeleton for scalable and flexible Jax RL implementations☆68Updated last year
- ☆34Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆43Updated 2 weeks ago
- ☆43Updated 3 months ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆15Updated this week
- ☆20Updated 2 years ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆13Updated 7 months ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆25Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆59Updated last year
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- A PyTorch Implementation of Skipper☆20Updated 3 months ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆79Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆47Updated last year
- GPT implementation in Flax☆18Updated 3 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆67Updated 5 months ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- General Modules for JAX☆62Updated 5 months ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆76Updated 9 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆19Updated last month
- ☆24Updated 6 months ago