symoon11 / dreamerv3-flax
Flax Implementation of DreamerV3 on Crafter
☆14Updated last week
Alternatives and similar repositories for dreamerv3-flax:
Users that are interested in dreamerv3-flax are comparing it to the libraries listed below
- ☆18Updated last month
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Various reinforcement learning algorithms written in Jax + Flax☆24Updated last year
- ☆47Updated 2 years ago
- ☆9Updated last year
- General Modules for JAX☆64Updated 2 weeks ago
- ☆73Updated 4 months ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆48Updated last year
- POPGym Library in JAX☆11Updated 10 months ago
- Baselines for gymnax 🤖☆66Updated last year
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆47Updated last month
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- Pytorch implementation of DreamerV2: Mastering Atari with Discrete World Models, based on the original implementation☆19Updated 2 years ago
- Reproduction of Dreamerv1 and v2 in pytorch for deepmind control suite☆35Updated 2 years ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆64Updated last year
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 4 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆69Updated 6 months ago
- Code for the papers Hypernetworks in Meta-Reinforcement Learning (Beck et al., 2022) and Recurrent Hypernetworks are Surprisingly Strong …☆12Updated 7 months ago
- ☆74Updated 6 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated 2 years ago
- ☆43Updated 5 months ago
- ☆35Updated 2 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆15Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆9Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago