kchua / mbrl-jaxLinks
MBRL library in JAX
☆10Updated 2 years ago
Alternatives and similar repositories for mbrl-jax
Users that are interested in mbrl-jax are comparing it to the libraries listed below
Sorting:
- ☆24Updated last year
- Official repo for Offline RL for Online RL☆17Updated last year
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆39Updated 8 months ago
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆40Updated 2 years ago
- ☆36Updated 2 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆33Updated 2 years ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆30Updated 2 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated last year
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30Updated 3 years ago
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆17Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- [CoRL 2020] COG: Connecting New Skills to Past Experience with Offline Reinforcement Learning☆33Updated 4 years ago
- ☆29Updated 4 years ago
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Updated 2 years ago
- Repository for the paper: "Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulation" @ NeurIPS 2022☆18Updated 2 years ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆26Updated 2 years ago
- Official repository for Paper "Offline Goal-Conditioned Reinforcement Learning via f-Advantage Regression" (NeurIPS 2022)☆35Updated last year
- Code base for paper: Reparameterized Policy Learning for Multimodal Trajectory Optimization☆27Updated last year
- Deep PILCO PyTorch Implementation☆15Updated 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
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024☆23Updated last year
- A modified benchmark for designing and controlling 2D Voxel-based Soft Robots☆35Updated last year
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆30Updated last year
- This repository contains the code for RL for POMDPs through learning an Approximate Information State.☆21Updated 3 years ago
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆12Updated 2 years ago
- ☆23Updated last year
- Authors' PyTorch implementation of 'Recomposing the Reinforcement Learning Building-Blocks with Hypernetworks' (HypeRL)☆24Updated 4 years ago
- Bipedal Skills Benchmark for Reinforcement Learning☆25Updated 2 years ago