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:
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆70Updated last year
- ☆13Updated 2 years ago
- ☆23Updated 2 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆17Updated last year
- [ICLR 2023] Choreographer: a model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able to effi…☆40Updated 11 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆57Updated last year
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆26Updated 2 years ago
- Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆25Updated last year
- Scripts to recreate the D4RL datasets with Minari☆19Updated 3 weeks ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆16Updated 11 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆27Updated 2 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆84Updated 6 months ago
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆12Updated last year
- Official repo for Offline RL for Online RL☆16Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆25Updated 2 years ago
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆11Updated last year
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆32Updated 7 months 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
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆37Updated 2 years ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- ☆18Updated last year
- ExORL: Exploratory Data for Offline Reinforcement Learning☆115Updated 3 years ago
- ☆32Updated last week
- Implementation of Jump-Start Reinforcement Learning (JSRL) with Stable Baselines3☆31Updated last year
- ☆21Updated last year
- Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…☆28Updated last year
- ☆31Updated last year
- Code for Tackling Long-Horizon Tasks with Model-based Offline Reinforcement Learning☆12Updated 4 months ago