FLAIROx / jafar
JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
☆32Updated last week
Related projects ⓘ
Alternatives and complementary repositories for jafar
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 3 months ago
- ☆79Updated this week
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆57Updated last year
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆21Updated 6 months ago
- ☆16Updated 3 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated 11 months ago
- ☆53Updated 3 weeks ago
- ☆61Updated 2 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 "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- An Open-Ended Agentic Simulator☆22Updated 2 months ago
- a simple and scalable agent for training adaptive policies with sequence-based RL☆89Updated this week
- ☆40Updated 3 months ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆17Updated 10 months ago
- ☆22Updated 4 months ago
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆85Updated 11 months ago
- Reinforcement learning on general 2D physics environments in JAX☆10Updated this week
- Conservative Q learning in Jax☆50Updated last year
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)☆74Updated 3 months ago
- OGBench: Benchmarking Offline Goal-Conditioned RL☆72Updated last week
- Foundation Policies with Hilbert Representations (ICML 2024)☆72Updated 6 months ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆83Updated 9 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- Reinforcement Learning via Supervised Learning☆68Updated 2 years ago
- Skeleton for scalable and flexible Jax RL implementations☆63Updated last year
- Corax: Core RL in JAX☆34Updated 8 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆93Updated 3 months ago
- Learning diverse options through the Laplacian representation.☆22Updated 10 months ago
- General Modules for JAX☆58Updated 3 months ago
- Clean single-file implementation of offline RL algorithms in JAX☆93Updated 2 months ago