Skip to main content

Physical units and unit-aware mathematical system in JAX

Header image of brainunit.

Supported Python Version LICENSE Documentation Status PyPI version Continuous Integration

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.

Other unofficial projects include:

  • 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.

Metadata

Release files for brainunit 0.0.3.post20241212

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Built distribution (wheel)

Table of built distributions (wheels) for brainunit 0.0.3.post20241212
File Interpreter ABI Platform
brainunit-0.0.3.post20241212-py2.py3-none-any.whl Python 3, Python 2 none any Details

Release files / brainunit-0.0.3.post20241212-py2.py3-none-any.whl

Download URL brainunit-0.0.3.post20241212-py2.py3-none-any.whl
Size 260.9 kB
Tags Python 2 Python 3
SHA-256 checksum
How to use checksums
ec6b7ac06be5035c17dc77c0ee36642dc17d49c72f766afc966f9d7065926fc3
BLAKE2b-256 checksum
How to use checksums
0543f14569e1a7d85c949fa56586b68cdec5ab652f59ee5116baba3fd756bdef
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.2 CPython/3.11.5

Release history Release notifications | RSS feed

0.5.2

2 release files

0.5.1

2 release files

0.5.0

2 release files

0.4.0

2 release files

0.3.2

2 release files

0.3.1

2 release files

0.3.0

2 release files

0.2.2

2 release files

0.2.1

2 release files

0.2.0

2 release files

0.1.4

2 release files

0.1.3

2 release files

0.1.2

2 release files

0.1.1

2 release files

0.1.0

2 release files

0.0.17

2 release files

0.0.16

2 release files

0.0.15

1 release file

0.0.14

1 release file

0.0.13

1 release file

0.0.12

1 release file

0.0.11

1 release file

0.0.10

1 release file

0.0.9

1 release file

0.0.8

1 release file

0.0.7

1 release file

0.0.6

1 release file

0.0.5

1 release file

0.0.4

1 release file

This release

0.0.3.post20241212 This release

1 release file

0.0.3

1 release file

0.0.2

1 release file

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page