brainunit 0.0.3.post20241209

Physical Units and Unit-aware Mathematical System in JAX

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

Unit-Aware Computation Ecosystem

brainunit has been deeply integrated into following diverse projects, such as:

• brainstate: A State-based Transformation System for Program Compilation and Augmentation

• braintaichi: Leveraging Taichi Lang to customize brain dynamics operators

• braintools: The Common Toolbox for Brain Dynamics Programming.

• dendritex: Dendritic Modeling in JAX

• pinnx: Physics-Informed Neural Networks for Scientific Machine Learning in JAX.

• diffrax: Numerical differential equation solvers in JAX.

• jax-md: Differentiable Molecular Dynamics in JAX

• Catalax: JAX-based framework to model biological systems

• ...

See also the BDP ecosystem

We are building the brain dynamics programming ecosystem. brainunit has been deeply integrated into our BDP ecosystem.