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
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆18Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- ☆21Updated last year
- ☆36Updated 2 years ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30Updated 3 years ago
- ☆24Updated last year
- Official repo for Offline RL for Online RL☆16Updated last year
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆39Updated 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
- Repository for the paper: "Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulation" @ NeurIPS 2022☆18Updated last year
- ☆29Updated 4 years ago
- ☆47Updated 2 years ago
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆40Updated 7 months ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated 2 years ago
- This repository contains the code for RL for POMDPs through learning an Approximate Information State.☆21Updated 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.☆71Updated 11 months ago
- ☆24Updated 11 months ago
- ☆15Updated 2 years ago
- ☆16Updated last year
- ☆43Updated 2 years ago
- Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024☆23Updated last year
- ☆23Updated 3 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆74Updated last year
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆29Updated 2 years ago
- Change-Based Exploration Transfer☆36Updated 3 years ago
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆24Updated 2 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆17Updated last year
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆33Updated 2 years ago