Researchers and developers building or simulating discrete-state Active Inference generative models in Python.

pymdp

A Python implementation of Active Inference for partially observable Markov Decision Processes.

Lead: Conor Heins

pymdp is an open-source Python package for discrete-state (POMDP) Active Inference, widely used across the research community for building and simulating generative models. It is an external ecosystem implementation, distinct from the Institute's own projects, that the Institute references and builds learning materials around.

Overview

pymdp implements discrete-state Active Inference for partially observable Markov Decision Processes, and has been introduced to the community via a PubMed-indexed step-by-step tutorial. It is maintained by the infer-actively organization, led by Conor Heins, and is a foundational tool referenced across the Institute's Computational domain page and learning materials.

Participate

Contributions, issues, and discussion happen directly on the upstream GitHub repository; the Institute's Discord is a good place to find collaborators building on top of it.

Related resources

Public links for this page