kyonofx / tia
[ICML 2021] Learning Task Informed Abstractions -- a representation learning approach for model-based RL in complex visual domains
☆19Updated 3 years ago
Alternatives and similar repositories for tia:
Users that are interested in tia are comparing it to the libraries listed below
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated 2 years ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆34Updated last year
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆31Updated 3 years ago
- Code base for paper: Reparameterized Policy Learning for Multimodal Trajectory Optimization☆26Updated last year
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated last year
- ☆22Updated 3 years ago
- ☆23Updated 9 months ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆29Updated 2 years ago
- Code for "Hierarchical Skills for Efficient Exploration" HSD-3 Algorithm and Baselines☆49Updated 2 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆29Updated last year
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Updated 2 years ago
- ☆34Updated 2 months ago
- Unofficial PyTorch implementation (replicating paper results) of Implicit Q-Learning (In-sample Q-Learning) for offline RL☆22Updated 3 months ago
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆29Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated 2 months ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆60Updated last year
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆14Updated 8 months ago
- ☆12Updated 2 years ago
- Source files to replicate experiments in my ICLR 2022 paper.☆67Updated 7 months ago
- ☆14Updated 10 months ago
- ☆16Updated 10 months ago
- Official release of CompoSuite, a compositional RL benchmark☆47Updated last year
- ☆41Updated last year
- ☆17Updated 3 years ago
- ☆35Updated 2 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆66Updated 2 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 5 months ago