kenjyoung / dreamerv2_JAX
An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.
☆14Updated last year
Related projects ⓘ
Alternatives and complementary repositories for dreamerv2_JAX
- Docker containers of baseline agents for the Crafter environment☆28Updated 2 years ago
- Code for "Possibility Before Utility: Learning And Using Hierarchical Affordances" ICLR 2022☆13Updated 2 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆25Updated last year
- ☆41Updated last month
- code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"☆24Updated 2 years ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆18Updated 2 years ago
- ☆17Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆23Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Updated 2 years ago
- Latent Dynamics Mixture, NeurIPS 2021☆17Updated 2 years ago
- Official repo for Offline RL for Online RL☆14Updated last year
- Code for AAAI 2023 paper "Hypernetworks for Zero-shot Transfer in Reinforcement Learning"☆16Updated last year
- Implementation of Proximal Policy Optimization in Jax+Flax☆16Updated last year
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆14Updated 6 months ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated last year
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆35Updated 2 weeks ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Updated 3 years ago
- ☆20Updated 6 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- ☆23Updated 2 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 years ago
- ☆21Updated 6 months ago
- ☆27Updated 3 years ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆28Updated last year
- Simple JAX Graphics Library.☆13Updated last week
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆17Updated last year
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆32Updated last year
- Implementation (TensorFlow/keras) of the DreamerV3 model-based RL algorithm by Hafner et al. 2023☆3Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆18Updated 4 months ago