kchua / mbrl-jax
MBRL library in JAX
☆9Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for mbrl-jax
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆17Updated last year
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Updated last year
- ☆21Updated 7 months ago
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆34Updated 2 years ago
- Official repo for Offline RL for Online RL☆14Updated last year
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆36Updated 3 weeks ago
- An open-source framework to benchmark and assess safety specifications of Reinforcement Learning problems.☆11Updated last year
- ☆20Updated 6 months ago
- Source code to reproduce experiments from Mendez et al., ICLR '22☆20Updated 2 years ago
- Bayesian active RL (BARL) and trajectory information planning (TIP)☆24Updated 2 years ago
- ☆28Updated 3 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆11Updated last year
- A modified benchmark for designing and controlling 2D Voxel-based Soft Robots☆32Updated last year
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆14Updated last year
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated last year
- ☆34Updated last year
- Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024☆22Updated 7 months ago
- ☆30Updated 3 months ago
- Safe Reinforcement Learning with Natural Language Constraints☆14Updated 3 years ago
- [CoRL 2020] COG: Connecting New Skills to Past Experience with Offline Reinforcement Learning☆31Updated 4 years ago
- Trajectory-wise Multiple Choice Learning for Dynamics Generalization in Reinforcement Learning (NeurIPS 2020)☆39Updated 4 years ago
- ☆22Updated 7 months ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆27Updated last year
- Code for AAAI 2023 paper "Hypernetworks for Zero-shot Transfer in Reinforcement Learning"☆16Updated last year
- IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL☆35Updated 2 months ago
- PyTorch implementation of our paper Reinforcement Learning with Random Delays (ICLR 2020)☆39Updated 2 years ago
- Bipedal Skills Benchmark for Reinforcement Learning☆23Updated 2 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆62Updated 4 months ago
- ☆38Updated last year