Stars
Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)
A Python implementation of active inference for Markov Decision Processes
Tigramite is a python package for causal inference with a focus on time series data. The Tigramite documentation is at
Reliable, minimal and scalable library for evaluating and conducting world model research
Evaluating and reproducing real-world robot manipulation policies (e.g., RT-1, RT-1-X, Octo) in simulation under common setups (e.g., Google Robot, WidowX+Bridge) (CoRL 2024)
#UAI2020 Codes for PAC-Bayesian Contrastive Unsupervised Representation Learning
A Paper List for Humanoid Robot Learning.
DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 Spotlight.
NeurIPS 2023: Safety-Gymnasium: A Unified Safe Reinforcement Learning Benchmark
A curated list of awesome model based RL resources (continually updated)
A curated list of awesome exploration RL resources (continually updated)
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Distributional Successor Features Enable Zero-Shot Policy Optimization
Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"
A fast and simple implementation of learning algorithms for robotics.
A Curated List of Awesome Works in World Modeling, Aiming to Serve as a One-stop Resource for Researchers, Practitioners, and Enthusiasts Interested in World Modeling.
A benchmark environment for fully cooperative human-AI performance.
robosuite: A Modular Simulation Framework and Benchmark for Robot Learning
Safe Multi-Agent Robosuite benchmark for safe multi-agent reinforcement learning research.
Reinforcement Learning through Active Inference
Official Implementation for the paper "SR-AIF: Solving Sparse-Reward Robotic Tasks from Pixels with Active Inference and World Models"
Isaac Lab API, powered by MuJoCo-Warp, for RL and robotics research.
Some tools to aid visualization of common tasks in PVRNN
Finding project directories in Python (data science) projects, just like in R rprojroot and here packages
Pytorch implementation of Hyperspherical Variational Auto-Encoders
Python library for Representational Similarity Analysis