symoon11 / dreamerv3-flax
Flax Implementation of DreamerV3 on Crafter
☆12Updated last week
Alternatives and similar repositories for dreamerv3-flax:
Users that are interested in dreamerv3-flax are comparing it to the libraries listed below
- ☆9Updated last year
- Various reinforcement learning algorithms written in Jax + Flax☆23Updated last year
- Pytorch implementation of DreamerV2: Mastering Atari with Discrete World Models, based on the original implementation☆18Updated 2 years ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆18Updated 3 weeks ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- ☆73Updated 3 months ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- ☆48Updated 2 years ago
- General Modules for JAX☆63Updated 6 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆47Updated last week
- Reproduction of Dreamerv1 and v2 in pytorch for deepmind control suite☆35Updated 2 years ago
- An implementation of MuZero in JAX.☆55Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆48Updated last year
- Simple JAX Graphics Library.☆30Updated 3 months ago
- Conservative Q learning in Jax☆52Updated 2 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆12Updated last year
- Baselines for gymnax 🤖☆63Updated last year
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- Deep Reinforcement Learning Framework done with PyTorch☆32Updated this week
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- Rainbow DQN implementation accompanying the paper "Fast and Data-Efficient Training of Rainbow" which reaches 205.7 median HNS after 10M …☆43Updated 3 years ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆20Updated last week
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆62Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 3 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆20Updated 2 months ago
- Code for the papers Hypernetworks in Meta-Reinforcement Learning (Beck et al., 2022) and Recurrent Hypernetworks are Surprisingly Strong …☆12Updated 6 months ago
- Dreamer on JAX☆15Updated 3 years ago