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:
- This repository contains the code for RL for POMDPs through learning an Approximate Information State.☆21Updated 4 years ago
- ☆19Updated 7 months ago
- ☆30Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated 2 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated 2 years ago
- ☆19Updated last year
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆40Updated 3 years ago
- ☆25Updated last year
- ☆16Updated 3 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- ☆29Updated 4 years ago
- ☆36Updated 2 years ago
- Bridging State and History Representations: Understanding Self-Predictive RL -- ICLR 2024☆21Updated last year
- Source code to reproduce experiments from Mendez et al., ICLR '22☆21Updated 3 years ago
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆30Updated 4 years ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆64Updated 3 months ago
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆16Updated 2 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 3 years ago
- Authors' PyTorch implementation of 'Recomposing the Reinforcement Learning Building-Blocks with Hypernetworks' (HypeRL)☆24Updated 4 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆29Updated last year
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆40Updated last month
- ☆23Updated 3 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆79Updated last year
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30Updated 3 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated last month
- ☆46Updated 2 years ago
- ☆31Updated 2 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆67Updated 3 years ago
- COOM: Benchmarking Continual Reinforcement Learning on Doom☆18Updated 6 months ago