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:
- ☆23Updated 2 years ago
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆39Updated 2 years ago
- Official repository for paper "Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching" (ICML…☆26Updated 2 years ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30Updated 3 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- ☆36Updated 2 years ago
- This repository contains the code for RL for POMDPs through learning an Approximate Information State.☆21Updated 3 years ago
- ☆24Updated last year
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆39Updated 7 months ago
- ☆25Updated 3 years ago
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆17Updated 2 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆33Updated 2 years ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Updated 3 years ago
- Source code to reproduce experiments from Mendez et al., ICLR '22☆21Updated 2 years ago
- ☆47Updated 2 years ago
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆20Updated 3 years ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆18Updated 4 years ago
- Code base for paper: Reparameterized Policy Learning for Multimodal Trajectory Optimization☆26Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆25Updated 2 years ago
- A Dual-RL method DVL: Dual-V Learning for offline and online reinforcement learning☆11Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Official repo for Offline RL for Online RL☆17Updated last year
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆30Updated 4 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated 2 years ago
- ☆12Updated 3 years ago
- Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024☆23Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated 11 months ago
- Deep PILCO PyTorch Implementation☆15Updated 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
- ☆29Updated 4 years ago