LICENSE
README.md
pyproject.toml
setup.cfg
inferactively_pymdp.egg-info/PKG-INFO
inferactively_pymdp.egg-info/SOURCES.txt
inferactively_pymdp.egg-info/dependency_links.txt
inferactively_pymdp.egg-info/requires.txt
inferactively_pymdp.egg-info/top_level.txt
pymdp/__init__.py
pymdp/agent.py
pymdp/algos.py
pymdp/control.py
pymdp/distribution.py
pymdp/inference.py
pymdp/learning.py
pymdp/likelihoods.py
pymdp/maths.py
pymdp/utils.py
pymdp/envs/__init__.py
pymdp/envs/cue_chaining.py
pymdp/envs/env.py
pymdp/envs/generalized_tmaze.py
pymdp/envs/graph_worlds.py
pymdp/envs/grid_world.py
pymdp/envs/rollout.py
pymdp/envs/tmaze.py
pymdp/envs/assets/cheese.png
pymdp/envs/assets/mouse.png
pymdp/envs/assets/shock.png
pymdp/legacy/__init__.py
pymdp/legacy/agent.py
pymdp/legacy/control.py
pymdp/legacy/default_models.py
pymdp/legacy/inference.py
pymdp/legacy/learning.py
pymdp/legacy/maths.py
pymdp/legacy/utils.py
pymdp/legacy/algos/__init__.py
pymdp/legacy/algos/fpi.py
pymdp/legacy/algos/mmp.py
pymdp/legacy/algos/mmp_old.py
pymdp/legacy/envs/__init__.py
pymdp/legacy/envs/env.py
pymdp/legacy/envs/grid_worlds.py
pymdp/legacy/envs/tmaze.py
pymdp/legacy/envs/visual_foraging.py
pymdp/planning/__init__.py
pymdp/planning/mcts.py
pymdp/planning/si.py
pymdp/planning/visualize.py
test/test_SPM_validation.py
test/test_agent.py
test/test_agent_jax.py
test/test_categorical_observations.py
test/test_control.py
test/test_control_jax.py
test/test_cue_chaining_env.py
test/test_demos.py
test/test_distribution.py
test/test_env.py
test/test_fpi.py
test/test_grid_world_parity.py
test/test_hmm_associative_scan.py
test/test_inductive_inference_jax.py
test/test_infer_states_optimized.py
test/test_inference.py
test/test_inference_jax.py
test/test_jax_sparse_backend.py
test/test_learning.py
test/test_learning_jax.py
test/test_message_passing_jax.py
test/test_mmp.py
test/test_param_info_gain_jax.py
test/test_pybefit_model_fitting.py
test/test_rollout_function.py
test/test_sophisticated_inference_jax.py
test/test_tmaze_envs.py
test/test_tmaze_recoverability.py
test/test_utils.py
test/test_utils_jax.py
test/test_vfe_jax.py
test/test_wrappers.py