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.
1 maintainer
Metadata
1 maintainer
Release files for brainunit 0.0.3.post20241206
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| brainunit-0.0.3.post20241206-py2.py3-none-any.whl | Python 2, Python 3 | none | any | Details |
Release files / brainunit-0.0.3.post20241206-py2.py3-none-any.whl
| Download URL | brainunit-0.0.3.post20241206-py2.py3-none-any.whl |
|---|---|
| Size | 258.1 kB |
| Tags | Python 2 Python 3 |
|
SHA-256 checksum How to use checksums |
dbbbdfdb387bf18a8f2e81b456c7c6efcdac6bf73020d2bd67bd5d08f396f19f
|
|
BLAKE2b-256 checksum How to use checksums |
cbe235bad11952c92c33f795a507744052748dc1d9ceccd7a42aeb2abd87a963
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.11.5
|
