kchua / mbrl-jax
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
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆18Updated last year
- ☆23Updated 2 years ago
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆36Updated 2 years ago
- ☆28Updated 3 years ago
- ☆21Updated 9 months ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆29Updated 2 years ago
- ☆34Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated 2 years ago
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆38Updated 2 months ago
- ☆30Updated last year
- Deep PILCO PyTorch Implementation☆14Updated last year
- This repository contains the code for RL for POMDPs through learning an Approximate Information State.☆20Updated 3 years ago
- Source code to reproduce experiments from Mendez et al., ICLR '22☆21Updated 2 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆67Updated 6 months ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated last year
- a modular reinforcement learning library with JAX agents☆22Updated last year
- Codes for the study "Variational Recurrent Models for Solving Partially Observable Control Tasks", published as a conference paper at ICL…☆51Updated 4 years ago
- ☆40Updated last year
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆31Updated 3 years ago
- Official repo for Offline RL for Online RL☆15Updated last year
- ☆21Updated 8 months ago
- ☆34Updated 4 years ago
- ☆16Updated 2 years ago
- Formulating Model-based RL Dynamics as a continuous rather then one step prediction☆35Updated 2 years ago
- Bayesian active RL (BARL) and trajectory information planning (TIP)☆24Updated 2 years ago
- Bipedal Skills Benchmark for Reinforcement Learning☆24Updated 2 years ago
- ☆21Updated 3 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆63Updated 7 months ago
- ☆45Updated last year
- Code base for paper: Reparameterized Policy Learning for Multimodal Trajectory Optimization☆26Updated last year