ToruOwO / mimexView external linksLinks
PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"
☆16May 17, 2023Updated 2 years ago
Alternatives and similar repositories for mimex
Users that are interested in mimex are comparing it to the libraries listed below
Sorting:
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- ☆10Jun 27, 2024Updated last year
- ☆13Apr 25, 2024Updated last year
- A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation (ICLR2023)☆14Feb 3, 2023Updated 3 years ago
- [ICLR 2023] Choreographer: a world-model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able t…☆41Jun 18, 2024Updated last year
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆17Oct 23, 2022Updated 3 years ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Aug 31, 2020Updated 5 years ago
- Bipedal Skills Benchmark for Reinforcement Learning☆25Oct 27, 2022Updated 3 years ago
- Code for Abstract-to-Executable Trajectory Translation for One Shot Task Generalization (ICML 2023)☆23May 12, 2023Updated 2 years ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆22Jul 21, 2025Updated 6 months ago
- [ICLR 2024 oral] Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆29Mar 1, 2024Updated last year
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆162Dec 16, 2025Updated 2 months ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Apr 19, 2024Updated last year
- Evaluation of TD-MPC2.☆21Jan 21, 2024Updated 2 years ago
- Chain-of-Thought Predictive Control☆57May 1, 2023Updated 2 years ago
- Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.☆31Nov 4, 2024Updated last year
- Implementation of CASCADE in Learning General World Models in a Handful of Reward-Free Deployments (NeurIPS 22).☆29Oct 25, 2022Updated 3 years ago
- Jaxpr Visualisation Tool☆35Dec 22, 2024Updated last year
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- Benchmarked implementations of Offline RL Algorithms.☆76Mar 4, 2025Updated 11 months ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆30Oct 29, 2023Updated 2 years ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Feb 6, 2023Updated 3 years ago
- ☆35Jan 29, 2023Updated 3 years ago
- Simple JAX Graphics Library.☆36Nov 3, 2024Updated last year
- Repo to reproduce the First-Explore paper results☆39Dec 25, 2024Updated last year
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- ☆34Aug 7, 2024Updated last year
- ☆38Oct 3, 2023Updated 2 years ago
- ☆35Jan 4, 2023Updated 3 years ago
- Code for our NeurIPS 2020 paper Improving Generalization in Reinforcement Learning with Mixture Regularization☆35Oct 22, 2020Updated 5 years ago
- A snappy + easy + pretty TUI debugger for Python.☆69Oct 3, 2025Updated 4 months ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆41Feb 27, 2024Updated last year
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆345Oct 3, 2022Updated 3 years ago
- Real-Time RTUs☆11Jan 2, 2025Updated last year
- DreamSmooth: Improving Model-Based RL with Reward Smoothing (ICLR 2024)☆12May 6, 2024Updated last year
- The Ecoacoustic Dataset from Arctic North Slope Alaska☆11May 29, 2025Updated 8 months ago
- A PyTorch Implementation of DF-GAN☆10Mar 26, 2022Updated 3 years ago
- ☆10Nov 17, 2022Updated 3 years ago