FLAIROx / jafar
JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
☆63Updated 3 months ago
Alternatives and similar repositories for jafar:
Users that are interested in jafar are comparing it to the libraries listed below
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)☆97Updated 9 months ago
- [NeurIPS 2024] GenRL: Multimodal-foundation world models enable grounding language and video prompts into embodied domains, by turning th…☆78Updated last month
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆68Updated last year
- ☆24Updated 10 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 8 months ago
- PWM: Policy Learning with Large World Models☆45Updated 2 months ago
- Using advances in generative modeling to learn reward functions from unlabeled videos.☆129Updated last year
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆44Updated last year
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆55Updated last year
- DrM, a visual RL algorithm, minimizes the dormant ratio to guide exploration-exploitation trade-offs, achieving significant improvements …☆73Updated 11 months ago
- Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023☆46Updated last year
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆151Updated 8 months ago
- The official implementation of flow Q-learning (FQL)☆140Updated last month
- ☆56Updated 2 years ago
- Masked World Models for Visual Control☆122Updated last year
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆82Updated 5 months ago
- Code release for "Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning" (NeurIPS 2023), https://ar…☆62Updated 7 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆15Updated 3 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆77Updated 11 months ago
- ☆32Updated last year
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆66Updated 11 months ago
- Improving Token-Based World Models with Parallel Observation Prediction (ICML 2024)☆14Updated last month
- [ICLR 2023] Choreographer: a model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able to effi…☆40Updated 10 months ago
- Transformer-based World Models☆81Updated 2 years ago
- ☆82Updated 2 months ago
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆30Updated 6 months ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆37Updated last year
- ☆76Updated 8 months ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆84Updated last year