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
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆85Updated 6 months ago
- ☆18Updated 2 years ago
- ☆24Updated 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
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆12Updated last year
- ☆23Updated 2 years ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆16Updated last year
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆51Updated 3 weeks ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆26Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆75Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆25Updated 2 years ago
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆37Updated 2 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated 11 months ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- ☆22Updated last year
- Official repo for Offline RL for Online RL☆17Updated last year
- ☆25Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆17Updated 11 months ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Updated 2 years ago
- ☆19Updated last month
- ☆47Updated 2 years ago
- Implementation of Jump-Start Reinforcement Learning (JSRL) with Stable Baselines3☆31Updated last year
- Code for AAAI 2023 paper "Hypernetworks for Zero-shot Transfer in Reinforcement Learning"☆20Updated 2 years ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆57Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆82Updated last year
- ☆42Updated 2 years ago
- A modified benchmark for designing and controlling 2D Voxel-based Soft Robots☆35Updated last year
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Updated 2 years ago