AlexGoldie / rl-learned-optimization
Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
☆22Updated last week
Alternatives and similar repositories for rl-learned-optimization:
Users that are interested in rl-learned-optimization are comparing it to the libraries listed below
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆44Updated this week
- Unified Implementations of Offline Reinforcement Learning Algorithms☆64Updated last week
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆55Updated last year
- Learning diverse options through the Laplacian representation.☆23Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆84Updated last year
- ☆75Updated 6 months ago
- ☆18Updated 3 months ago
- ☆76Updated last month
- Corax: Core RL in JAX☆37Updated last year
- Conservative Q learning in Jax☆53Updated 2 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 8 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆10Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆100Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆68Updated last year
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆50Updated 2 weeks ago
- Synthetic Experience Replay☆93Updated 11 months ago
- Highly scalable 2D JAX physics engine.☆55Updated last month
- ☆82Updated 2 months ago
- ☆47Updated 2 years ago
- Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023☆46Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆80Updated last year
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆62Updated last year
- A collection of matrix games in JAX☆11Updated 5 months ago
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆91Updated 3 weeks ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆52Updated last year
- POPGym Library in JAX☆11Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆86Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆80Updated 5 months ago
- Synchronized Curriculum Learning for RL Agents☆44Updated last month
- ☆41Updated 9 months ago