Cognax

Build, fit, and sample from cognitive models with JAX + NumPyro.

Cognax currently has three submodules:

• cognax.decisions: tools for modeling decision making
• cognax.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.

Installation

You can install the latest development version of Cognax directly from GitHub using the following command:

pip install git+https://github.com/compmem/cognax

Development

Cognax uses uv for local development. To create a development environment and install the project with its development dependencies, run:

uv sync

Run tests through uv:

uv run pytest