MichalBortkiewicz / JaxGCRL
Goal-Conditioned Reinforcement Learning with JAX
☆127Updated this week
Alternatives and similar repositories for JaxGCRL:
Users that are interested in JaxGCRL are comparing it to the libraries listed below
- A benchmark for offline goal-conditioned RL and offline RL☆139Updated 2 weeks ago
- Clean single-file implementation of offline RL algorithms in JAX☆135Updated 2 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆142Updated last week
- ☆47Updated 2 years ago
- Skeleton for scalable and flexible Jax RL implementations☆74Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆141Updated last year
- ☆191Updated 3 months ago
- Jax/Flax Implementation of TD-MPC2☆58Updated last week
- ☆77Updated 3 weeks ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆100Updated 9 months ago
- Highly scalable 2D JAX physics engine.☆52Updated last week
- Contains JAX implementation of algorithms for inverse reinforcement learning☆71Updated 7 months ago
- ☆18Updated last month
- Foundation Policies with Hilbert Representations (ICML 2024)☆80Updated 11 months ago
- ☆74Updated 7 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆67Updated 9 months ago
- ☆73Updated 4 months ago
- ☆41Updated 8 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆64Updated last year
- ☆43Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆109Updated 7 months ago
- General Modules for JAX☆64Updated 3 weeks ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆111Updated 3 years ago
- Conservative Q learning in Jax☆53Updated 2 years ago
- a simple and scalable agent for training adaptive policies with sequence-based RL☆114Updated last month
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆87Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated 2 years ago
- Transformer-based World Models☆78Updated last year
- Deep Hierarchical Planning from Pixels☆95Updated 2 years ago