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
Related projects ⓘ
Alternatives and complementary repositories for offline_baselines_jax
- ☆13Updated 2 years ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆59Updated last year
- a modular reinforcement learning library with JAX agents☆22Updated last year
- ☆23Updated 2 years ago
- A modified benchmark for designing and controlling 2D Voxel-based Soft Robots☆32Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆57Updated 5 months ago
- [NeurIPS 2024] Maximum Entropy Reinforcement Learning via Energy-Based Normalizing Flow☆16Updated 3 weeks ago
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆11Updated last year
- ☆20Updated 6 months ago
- Source files to replicate experiments in my ICLR 2022 paper.☆62Updated 4 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Official repo for Offline RL for Online RL☆14Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆23Updated 2 years ago
- ☆55Updated last month
- Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…☆19Updated 7 months ago
- Temporally Correlated Episodic Reinforcement Learning, ICLR 24☆13Updated 7 months ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated 2 years ago
- Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation☆62Updated last year
- Fast reinforcement learning research☆55Updated 5 months ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆31Updated last year
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated last year
- Scripts to recreate the D4RL datasets with Minari☆18Updated last month
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 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
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆18Updated 2 years ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆12Updated 5 months ago
- ☆22Updated 5 months ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated last year
- ☆42Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆63Updated last year