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