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 PyPI Downloads

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

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.post20241216
File Interpreter ABI Platform
brainunit-0.0.3.post20241216-py2.py3-none-any.whl Python 2, Python 3 none any Details

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

Download URL brainunit-0.0.3.post20241216-py2.py3-none-any.whl
Size 283.0 kB
Tags Python 2 Python 3
SHA-256 checksum
How to use checksums
2c8b66c42600fb3a28c8446706ae0f05e957b1c3b6ff6eb74476d6f07b1ce42a
BLAKE2b-256 checksum
How to use checksums
61ee90582542d2cfba2f4e9be943b2ff0a3a4cbd3d715b4b34e4974358079e1f
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.post20241216 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