qw3rtman / robust-world-model-planningLinks
Code for "Closing the Train-Test Gap in World Models for Gradient-Based Planning"
☆74Updated last week
Alternatives and similar repositories for robust-world-model-planning
Users that are interested in robust-world-model-planning are comparing it to the libraries listed below
Sorting:
- ☆83Updated last year
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆115Updated last year
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆98Updated last year
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆79Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆123Updated 2 weeks ago
- ☆52Updated 3 months ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆28Updated last year
- Implementation of Danijar's latest iteration for his Dreamer line of work☆158Updated this week
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆28Updated 2 years ago
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆162Updated last month
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆36Updated 6 months ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆151Updated 8 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆40Updated last year
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks☆36Updated last year
- This is the code repository for IntPhys 2, a video benchmark designed to evaluate the intuitive physics understanding of deep learning mo…☆93Updated 3 months ago
- Code for "Baba Is AI: Break the Rules to Beat the Benchmark"☆41Updated 4 months ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆29Updated 7 months ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆21Updated last year
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated 4 months ago
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆214Updated 11 months ago
- ☆35Updated 7 months ago
- Synchronized Curriculum Learning for RL Agents☆119Updated 2 months ago
- minimal Energy-based transformer☆42Updated last month
- This repository is the official implementation of the TRAC optimizer in Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement …☆32Updated 8 months ago
- ☆31Updated last year
- The Official PyTorch Implementation of "Brain-like Variational Inference" (NeurIPS 2025 Paper)☆65Updated 2 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆107Updated 3 months ago
- ☆78Updated 8 months ago
- ☆13Updated 5 months ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆85Updated 2 months ago