mjyoo2 / offline_baselines_jax
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
- ☆13Updated 2 years ago
- A modified benchmark for designing and controlling 2D Voxel-based Soft Robots☆34Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆24Updated 2 years ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆60Updated last year
- ☆23Updated 9 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated last month
- ☆23Updated 2 years ago
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆12Updated last year
- ☆18Updated last year
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆34Updated last year
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated 2 years ago
- ☆21Updated 9 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
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 2 years ago
- A set of environments utilizing pybullet for simulation of robotic manipulation tasks.☆25Updated 3 years ago
- ☆46Updated 2 years ago
- Code for "Possibility Before Utility: Learning And Using Hierarchical Affordances" ICLR 2022☆14Updated 2 years ago
- Deep Reinforcement Learning Framework done with PyTorch☆31Updated 3 weeks ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Updated last year
- a modular reinforcement learning library with JAX agents☆22Updated last year
- [ICLR 2023] Choreographer: a model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able to effi…☆39Updated 7 months ago
- ☆31Updated 10 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆27Updated last year
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆14Updated 8 months ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆17Updated 2 months ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated last year
- ☆47Updated last year
- DreamerV3 implementation of Curious Replay, a method for prioritizing experience replay that is tailored to model-based reinforcement lea…☆35Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆67Updated 6 months ago
- [CoRL 2020] COG: Connecting New Skills to Past Experience with Offline Reinforcement Learning☆32Updated 4 years ago