FLAIROx / jafar
JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
☆38Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for jafar
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 3 months ago
- ☆92Updated this week
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆58Updated last year
- ☆55Updated last month
- a simple and scalable agent for training adaptive policies with sequence-based RL☆91Updated this week
- Skeleton for scalable and flexible Jax RL implementations☆63Updated last year
- OGBench: Benchmarking Offline Goal-Conditioned RL☆79Updated 3 weeks ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆21Updated 7 months ago
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)☆76Updated 3 months ago
- Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023☆44Updated 6 months ago
- ☆17Updated 4 months ago
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆85Updated last year
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆95Updated 5 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated last year
- ☆22Updated 5 months ago
- Synthetic Experience Replay☆73Updated 5 months ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆33Updated 8 months ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆22Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆72Updated 7 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆133Updated last year
- Simple JAX Graphics Library.☆23Updated 2 weeks ago
- ☆63Updated 3 months ago
- Conservative Q learning in Jax☆51Updated last year
- Reinforcement Learning via Supervised Learning☆68Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆105Updated 2 months ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆83Updated 9 months ago
- ☆40Updated 4 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆87Updated 11 months ago