GitHub repository: tMiy4/pymdp_test
Description: pymdp_jax_test