kenjyoung / dreamerv2_JAXView external linksLinks
An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.
☆18Jan 16, 2023Updated 3 years ago
Alternatives and similar repositories for dreamerv2_JAX
Users that are interested in dreamerv2_JAX are comparing it to the libraries listed below
Sorting:
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Probabilistic inference for models of behaviour☆10Oct 13, 2025Updated 4 months ago
- Repository for the paper: "Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulation" @ NeurIPS 2022☆21Jul 10, 2023Updated 2 years ago
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- Deep Multi-Agent Reinforcement Learning with StarCraft 2☆10Sep 27, 2020Updated 5 years ago
- (Official) PyTorch implementation for Efficient Episodic Memory Utilization of Cooperative Multi-Agent Reinforcement Learning (EMU) in IC…☆12May 23, 2024Updated last year
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 7 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- ☆14Oct 7, 2022Updated 3 years ago
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Feb 20, 2023Updated 2 years ago
- [ICLR 2023] Choreographer: a world-model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able t…☆41Jun 18, 2024Updated last year
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆20Mar 22, 2022Updated 3 years ago
- A simple example of randomized ensembled double q learning☆19Sep 3, 2021Updated 4 years ago
- Official implementation of A Mixture of Surprises for Unsupervised Reinforcement Learning☆23Nov 16, 2022Updated 3 years ago
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Jul 6, 2023Updated 2 years ago
- Implementation of 6 DQN extension methods using Pytorch. (RAINBOW)☆16Dec 7, 2020Updated 5 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated 9 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- The repository is for Reinforcement-Learning Uncertainty research, in which we investigate various uncertain factors in RL.☆23Jun 16, 2023Updated 2 years ago
- Code for paper Rethinking the Data Annotation Process for Multi-view 3D Pose Estimation with Active Learning and Self-Training☆22Apr 18, 2023Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated 11 months ago
- Bayesian active RL (BARL) and trajectory information planning (TIP)☆26Oct 11, 2022Updated 3 years ago
- Today I Learned☆21Jan 1, 2025Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆163Jun 23, 2023Updated 2 years ago
- An open-source reinforcement learning environment build toolkit for ROS and Gazebo.☆22Jul 22, 2022Updated 3 years ago
- ☆26Apr 26, 2024Updated last year
- Neural Graphical models are neural network based graphical models that offer richer representation, faster inference & sampling☆30Aug 12, 2025Updated 6 months ago
- code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"☆24Dec 29, 2021Updated 4 years ago
- Unity로 멀티 에이전트 강화학습(MARL) 수행하기 위한 프레임 워크 제공☆24Apr 17, 2022Updated 3 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆30Oct 29, 2023Updated 2 years ago
- ☆86Jan 9, 2026Updated last month
- Simple JAX Graphics Library.☆36Nov 3, 2024Updated last year
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- ☆132May 8, 2020Updated 5 years ago
- A Simplified Pytorch Version of the Dreamer Algorithm☆149Jul 24, 2023Updated 2 years ago
- A modified benchmark for designing and controlling 2D Voxel-based Soft Robots☆38Nov 18, 2023Updated 2 years ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆38Jun 11, 2025Updated 8 months ago
- Main repository for the Quantum Hackathon in Bilbao☆10Dec 9, 2019Updated 6 years ago