junmokane / spatially-aware-transformer
☆8Updated last month
Alternatives and similar repositories for spatially-aware-transformer:
Users that are interested in spatially-aware-transformer are comparing it to the libraries listed below
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆11Updated 10 months ago
- PWM: Policy Learning with Large World Models☆39Updated 4 months ago
- ☆29Updated last year
- ☆21Updated 9 months ago
- ☆24Updated 6 months ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆18Updated 2 years ago
- JAX implementation of WSRL and RL baselines☆18Updated last week
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆61Updated 9 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- [NeurIPS 2024] GenRL: Multimodal-foundation world models enable grounding language and video prompts into embodied domains, by turning th…☆63Updated this week
- Temporally Correlated Episodic Reinforcement Learning, ICLR 24☆13Updated 9 months ago
- [RA-L 2024] Novel action spaces leveraging redundancy in 7 DoF arms enable efficient & precise learning in robotic manipulation☆17Updated 7 months ago
- Code release for the paper "Goal Representations for Instruction Following: A Semi-Supervised Language Interface to Control"☆14Updated 9 months ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆29Updated 2 years ago
- A Dual-RL method DVL: Dual-V Learning for offline and online reinforcement learning☆9Updated last year
- ☆34Updated last month
- code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"☆24Updated 3 years ago
- Official repo for Offline RL for Online RL☆15Updated last year
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Bingham Policy Parameterization for 3D Rotations in Reinforcement Learning☆24Updated 2 years ago
- ☆17Updated last year
- ☆14Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 5 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆78Updated 8 months ago
- Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆25Updated 10 months ago
- ☆54Updated last year
- [NeurIPS 2022] Official implementation of the paper: "Human-AI Shared Control via Policy Dissection"☆49Updated last year
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆27Updated 11 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆27Updated last year
- Code for "DittoGym: Learning to Control Soft Shape-Shifting Robots" by Suning Huang, Boyuan Chen, Huazhe Xu, and Vincent Sitzmann.☆26Updated 8 months ago