pymdp: A Python library for active inference in discrete state spaces

Conor Heins, Beren Millidge, Daphne Demekas, Brennan Klein, Karl J. Friston, Iain D. Couzin, Alexander Tschantz