Physical units and unit-aware mathematical system in JAX

brainunit provides physical units and unit-aware mathematical system in JAX for brain dynamics and AI4Science

Installation

You can install brainunit via pip:

pip install brainunit --upgrade

Documentation

The official documentation is hosted on Read the Docs: https://brainunit.readthedocs.io

See also the BDP ecosystem

We are building the BDP ecosystem: https://ecosystem-for-brain-dynamics.readthedocs.io/