mjyoo2 / offline_baselines_jaxLinks
Offline Baselines with JAX is a set of implementations of reinforcement learning algorithms in JAX. This library is based on Stable Baselines 3, and JAXRL.
☆5Updated last year
Alternatives and similar repositories for offline_baselines_jax
Users that are interested in offline_baselines_jax are comparing it to the libraries listed below
Sorting:
- ☆24Updated last year
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆12Updated 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
- Modular Single-file Reinfocement Learning Algorithms Library☆38Updated 2 years ago
- ☆23Updated 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
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Official repo for Offline RL for Online RL☆17Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆25Updated 2 years ago
- ☆31Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated last year
- A modified benchmark for designing and controlling 2D Voxel-based Soft Robots☆35Updated last year
- [ICLR 2023] Choreographer: a model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able to effi…☆41Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆70Updated last year
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 3 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19Updated last year
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30Updated 3 years ago
- ☆36Updated 2 years ago
- MBRL library in JAX☆10Updated 2 years ago
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆40Updated 2 years ago
- PlanDQ: Hierarchical Plan Orchestration via D-Conductor and Q-Performer☆10Updated last year
- Temporally Correlated Episodic Reinforcement Learning, ICLR 24☆12Updated last year
- Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024☆23Updated last year
- ☆23Updated last year
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆27Updated 2 years ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆27Updated 2 years ago
- Fast reinforcement learning research☆61Updated 7 months ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆30Updated 2 years ago
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆29Updated 2 years ago
- ☆42Updated 3 years ago