argmax-ai / aime
Official repository for our paper on "Action Inference by Maximising Evidence: Zero-Shot Imitation from Observation with World Models"
☆12Updated 11 months ago
Related projects ⓘ
Alternatives and complementary repositories for aime
- Official repository for paper "Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching" (ICML…☆25Updated last year
- PWM: Policy Learning with Large World Models☆37Updated 3 months ago
- ☆43Updated 4 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆73Updated 6 months ago
- Codebase for ICLR 2023 paper, "SMART: Self-supervised Multi-task pretrAining with contRol Transformers"☆50Updated 9 months ago
- [NeurIPS 2022] Official implementation of the paper: "Human-AI Shared Control via Policy Dissection"☆48Updated last year
- ☆45Updated 9 months ago
- Official repo for Offline RL for Online RL☆14Updated last year
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆98Updated last year
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆24Updated last year
- [NeurIPS 2024] GenRL: Multimodal foundation world models allow grounding language and video prompts into embodied domains, by turning the…☆59Updated 3 months ago
- Codebase of NeurIPS 2022 paper ''Planning for Sample Efficient Imitation Learning''☆35Updated 2 years ago
- ☆22Updated 5 months ago
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆59Updated 7 months ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆12Updated 5 months ago
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- [ICLR 2023] Choreographer: a model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able to effi…☆36Updated 5 months ago
- Meta-RL Model-Based Algorithm☆27Updated 5 months ago
- ☆28Updated last year
- ☆15Updated last year
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆44Updated 8 months ago
- Temporally Correlated Episodic Reinforcement Learning, ICLR 24☆13Updated 7 months ago
- code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"☆24Updated 2 years ago
- ☆14Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated last year
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆41Updated last year
- [CoRL 2020] COG: Connecting New Skills to Past Experience with Offline Reinforcement Learning☆31Updated 4 years ago
- Evaluation of TD-MPC2.☆22Updated 10 months ago
- Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…☆18Updated 7 months ago