Build, fit, and sample from cognitive models with JAX + NumPyro.
Cognax currently has three submodules:
cognax.decisions: tools for modeling decision makingcognax.joint: tools for joint modeling (e.g. neural and behavioral data)cognax.experimental: everything else
Warning
This project is still in early development. The API is subject to change, and the code has not been thoroughly bug tested.
You can install the latest development version of Cognax directly from GitHub using the following command:
pip install git+https://github.com/compmem/cognaxNote
JAX is not listed as a dependency, but it must be installed separately before using Cognax. See the JAX Docs for more information.