brainunit 0.2.1__tar.gz → 0.3.0__tar.gz

{brainunit-0.2.1 → brainunit-0.3.0}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: brainunit
-Version: 0.2.1
+Version: 0.3.0
 Summary: Physical Units and Unit-Aware Mathematical System for General-Purpose Brain Dynamics Modeling
 Author-email: BrainUnit Developers <chao.brain@qq.com>
 License-Expression: Apache-2.0
 Project-URL: homepage, https://github.com/chaobrain/brainunit
 Project-URL: repository, https://github.com/chaobrain/brainunit
 Project-URL: Bug Tracker, https://github.com/chaobrain/brainunit/issues
-Project-URL: Documentation, https://brainunit.readthedocs.io/
+Project-URL: Documentation, https://brainx.chaobrain.com/brainunit/
 Keywords: physical unit,physical quantity,brain modeling
 Classifier: Natural Language :: English
 Classifier: Operating System :: OS Independent
@@ -30,10 +30,12 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: numpy
 Requires-Dist: typing_extensions
-Requires-Dist: saiunit==0.2.1
+Requires-Dist: saiunit==0.3.0
 Provides-Extra: testing
 Requires-Dist: pytest; extra == "testing"
 Requires-Dist: brainstate; extra == "testing"
+Provides-Extra: jax
+Requires-Dist: jax; extra == "jax"
 Provides-Extra: cpu
 Requires-Dist: jax[cpu]; extra == "cpu"
 Provides-Extra: cuda12
@@ -42,12 +44,22 @@ Provides-Extra: cuda13
 Requires-Dist: jax[cuda13]; extra == "cuda13"
 Provides-Extra: tpu
 Requires-Dist: jax[tpu]; extra == "tpu"
+Provides-Extra: cupy
+Requires-Dist: cupy-cuda12x>=13.0; extra == "cupy"
+Provides-Extra: torch
+Requires-Dist: torch>=2.0; extra == "torch"
+Provides-Extra: dask
+Requires-Dist: dask[array]>=2024.1; extra == "dask"
+Provides-Extra: ndonnx
+Requires-Dist: ndonnx>=0.9; extra == "ndonnx"
+Provides-Extra: all
+Requires-Dist: brainunit[cupy,dask,jax,ndonnx,torch]; extra == "all"
 
 <h1 align='center'>BrainUnit</h1>
 <h2 align='center'>Physical units and unit-aware math system for general-purpose brain dynamics modeling</h2>
 
 <p align="center">
-  	<img alt="Header image of brainunit." src="https://raw.githubusercontent.com/chaobrain/saiunit/main/docs/_static/brainunit.png" width=30%>
+  	<img alt="Header image of brainunit." src="https://brainx.chaobrain.com/images/brainunit.webp" width=30%>
 </p> 
 
 
@@ -55,7 +67,7 @@ Requires-Dist: jax[tpu]; extra == "tpu"
 <p align="center">
 	<a href="https://pypi.org/project/brainunit/"><img alt="Supported Python Version" src="https://img.shields.io/pypi/pyversions/brainunit"></a>
 	<a href="https://github.com/chaobrain/brainunit/blob/main/LICENSE"><img alt="LICENSE" src="https://img.shields.io/badge/License-Apache%202.0-blue.svg"></a>
-    <a href='https://brainunit.readthedocs.io/?badge=latest'>
+    <a href='https://brainx.chaobrain.com/braintools/'>
         <img src='https://readthedocs.org/projects/brainunit/badge/?version=latest' alt='Documentation Status' />
     </a>  	
     <a href="https://badge.fury.io/py/brainunit"><img alt="PyPI version" src="https://badge.fury.io/py/brainunit.svg"></a>
@@ -67,7 +79,7 @@ Requires-Dist: jax[tpu]; extra == "tpu"
 
 [BrainUnit](https://github.com/chaobrain/brainunit) provides physical units and unit-aware mathematical system in JAX for brain dynamics modeling. It introduces rigoirous physical units into high-performance AI-driven abstract numerical computing. 
 
-BrainUnit is initially designed to enable unit-aware computations in brain dynamics modeling (see our [BDP ecosystem](https://ecosystem-for-brain-dynamics.readthedocs.io/)). However, its features and capacities can be applied to general domains in scientific computing and AI4Science. Starting in 2025/02, BrainUnit has been fully integrated into [SAIUnit](https://github.com/chaobrain/saiunit) (the **Unit** system for **S**cientific **AI**). 
+BrainUnit is initially designed to enable unit-aware computations in brain dynamics modeling (see our [BDP ecosystem](https://brainx.chaobrain.com/)). However, its features and capacities can be applied to general domains in scientific computing and AI4Science. Starting in 2025/02, BrainUnit has been fully integrated into [SAIUnit](https://github.com/chaobrain/saiunit) (the **Unit** system for **S**cientific **AI**). 
 
 Functionalities are the same for both ``brainunit`` and ``saiunit``, and their functions and data structures are interoperable, sharing the same set of APIs, and eliminating any potential conflicts. This meas that 
 
@@ -87,13 +99,13 @@ For users primarily engaged in general scientific computing, `saiunit` is likely
 
 ## Documentation
 
-The official documentation of BrainUnit is hosted on Read the Docs: [https://brainunit.readthedocs.io](https://brainunit.readthedocs.io)
+The official documentation of BrainUnit is hosted on Read the Docs: [https://brainx.chaobrain.com/brainunit](https://brainx.chaobrain.com/brainunit)
 
 
 
 ## Features
 
-`brainunit` can be seamlessly integrated into every aspect of our [brain modeling ecosystem](https://brainmodeling.readthedocs.io/), such as, the checkpointing of [braintools](https://github.com/chaobrain/braintools), the event-driven operators in [brainevent](https://github.com/chaobrain/brainevent), the state-based JIT compilation in [brainstate](https://github.com/chaobrain/brainstate), online learning rules in [brainscale](https://github.com/chaobrain/brainscale), or event more. 
+`brainunit` can be seamlessly integrated into every aspect of our [brain modeling ecosystem](https://brainx.chaobrain.com/), such as, the checkpointing of [braintools](https://github.com/chaobrain/braintools), the event-driven operators in [brainevent](https://github.com/chaobrain/brainevent), the state-based JIT compilation in [brainstate](https://github.com/chaobrain/brainstate), online learning rules in [brainscale](https://github.com/chaobrain/brainscale), or event more. 
 
 A quick example for this kind of integration:
 
@@ -175,4 +187,4 @@ If you use `brainunit` in your research, please consider citing the following pa
 
 ## See also the ecosystem
 
-``brainunit`` is one part of our [brain modeling ecosystem](https://brainmodeling.readthedocs.io/). 
+``brainunit`` is one part of our [brain modeling ecosystem](https://brainx.chaobrain.com/). 

{brainunit-0.2.1 → brainunit-0.3.0}/README.md

@@ -2,7 +2,7 @@
 <h2 align='center'>Physical units and unit-aware math system for general-purpose brain dynamics modeling</h2>
 
 <p align="center">
-  	<img alt="Header image of brainunit." src="https://raw.githubusercontent.com/chaobrain/saiunit/main/docs/_static/brainunit.png" width=30%>
+  	<img alt="Header image of brainunit." src="https://brainx.chaobrain.com/images/brainunit.webp" width=30%>
 </p> 
 
 
@@ -10,7 +10,7 @@
 <p align="center">
 	<a href="https://pypi.org/project/brainunit/"><img alt="Supported Python Version" src="https://img.shields.io/pypi/pyversions/brainunit"></a>
 	<a href="https://github.com/chaobrain/brainunit/blob/main/LICENSE"><img alt="LICENSE" src="https://img.shields.io/badge/License-Apache%202.0-blue.svg"></a>
-    <a href='https://brainunit.readthedocs.io/?badge=latest'>
+    <a href='https://brainx.chaobrain.com/braintools/'>
         <img src='https://readthedocs.org/projects/brainunit/badge/?version=latest' alt='Documentation Status' />
     </a>  	
     <a href="https://badge.fury.io/py/brainunit"><img alt="PyPI version" src="https://badge.fury.io/py/brainunit.svg"></a>
@@ -22,7 +22,7 @@
 
 [BrainUnit](https://github.com/chaobrain/brainunit) provides physical units and unit-aware mathematical system in JAX for brain dynamics modeling. It introduces rigoirous physical units into high-performance AI-driven abstract numerical computing. 
 
-BrainUnit is initially designed to enable unit-aware computations in brain dynamics modeling (see our [BDP ecosystem](https://ecosystem-for-brain-dynamics.readthedocs.io/)). However, its features and capacities can be applied to general domains in scientific computing and AI4Science. Starting in 2025/02, BrainUnit has been fully integrated into [SAIUnit](https://github.com/chaobrain/saiunit) (the **Unit** system for **S**cientific **AI**). 
+BrainUnit is initially designed to enable unit-aware computations in brain dynamics modeling (see our [BDP ecosystem](https://brainx.chaobrain.com/)). However, its features and capacities can be applied to general domains in scientific computing and AI4Science. Starting in 2025/02, BrainUnit has been fully integrated into [SAIUnit](https://github.com/chaobrain/saiunit) (the **Unit** system for **S**cientific **AI**). 
 
 Functionalities are the same for both ``brainunit`` and ``saiunit``, and their functions and data structures are interoperable, sharing the same set of APIs, and eliminating any potential conflicts. This meas that 
 
@@ -42,13 +42,13 @@ For users primarily engaged in general scientific computing, `saiunit` is likely
 
 ## Documentation
 
-The official documentation of BrainUnit is hosted on Read the Docs: [https://brainunit.readthedocs.io](https://brainunit.readthedocs.io)
+The official documentation of BrainUnit is hosted on Read the Docs: [https://brainx.chaobrain.com/brainunit](https://brainx.chaobrain.com/brainunit)
 
 
 
 ## Features
 
-`brainunit` can be seamlessly integrated into every aspect of our [brain modeling ecosystem](https://brainmodeling.readthedocs.io/), such as, the checkpointing of [braintools](https://github.com/chaobrain/braintools), the event-driven operators in [brainevent](https://github.com/chaobrain/brainevent), the state-based JIT compilation in [brainstate](https://github.com/chaobrain/brainstate), online learning rules in [brainscale](https://github.com/chaobrain/brainscale), or event more. 
+`brainunit` can be seamlessly integrated into every aspect of our [brain modeling ecosystem](https://brainx.chaobrain.com/), such as, the checkpointing of [braintools](https://github.com/chaobrain/braintools), the event-driven operators in [brainevent](https://github.com/chaobrain/brainevent), the state-based JIT compilation in [brainstate](https://github.com/chaobrain/brainstate), online learning rules in [brainscale](https://github.com/chaobrain/brainscale), or event more. 
 
 A quick example for this kind of integration:
 
@@ -130,4 +130,4 @@ If you use `brainunit` in your research, please consider citing the following pa
 
 ## See also the ecosystem
 
-``brainunit`` is one part of our [brain modeling ecosystem](https://brainmodeling.readthedocs.io/). 
+``brainunit`` is one part of our [brain modeling ecosystem](https://brainx.chaobrain.com/). 

{brainunit-0.2.1 → brainunit-0.3.0}/brainunit/__init__.py

@@ -18,6 +18,7 @@ __version__ = saiunit.__version__
 __version_info__ = saiunit.__version_info__
 
 from . import autograd
+from saiunit._matplotlib_compat import enable_matplotlib_support
 from . import constants
 from . import fft
 from . import lax
@@ -34,6 +35,18 @@ from ._base_dimension import (
     get_dim_for_display,
     get_or_create_dimension,
 )
+from ._backend import (
+    get_default_backend,
+    is_cupy_array,
+    is_dask_array,
+    is_jax_array,
+    is_ndonnx_array,
+    is_numpy_array,
+    is_torch_array,
+    set_default_backend,
+    using_backend,
+)
+from ._exceptions import BackendError
 from ._base_getters import (
     array_with_unit,
     assert_quantity,
@@ -48,14 +61,15 @@ from ._base_getters import (
     have_same_dim,
     is_dimensionless,
     is_scalar_type,
+    is_unit_equal_math,
     is_unitless,
     maybe_decimal,
     split_mantissa_unit,
     unit_scale_align_to_first,
 )
-from ._base_quantity import Quantity, compatible_with_equinox
+from ._base_quantity import Quantity
 from ._base_unit import UNITLESS, Unit, add_standard_unit, parse_unit
-from ._celsius import celsius2kelvin, kelvin2celsius
+from ._celsius import celsius2kelvin, kelvin2celsius, fahrenheit2kelvin, kelvin2fahrenheit
 from ._misc import maybe_custom_array, maybe_custom_array_tree
 from ._unit_common import *
 from ._unit_common import __all__ as _common_all
@@ -100,6 +114,18 @@ __all__ = [
               'DIMENSIONLESS',
               'DimensionMismatchError',
               'UnitMismatchError',
+              'BackendError',
+
+              # _backend
+              'get_default_backend',
+              'set_default_backend',
+              'using_backend',
+              'is_jax_array',
+              'is_numpy_array',
+              'is_cupy_array',
+              'is_torch_array',
+              'is_dask_array',
+              'is_ndonnx_array',
               'get_or_create_dimension',
               'get_dim_for_display',
 
@@ -125,12 +151,12 @@ __all__ = [
               'assert_quantity',
               'have_same_dim',
               'has_same_unit',
+              'is_unit_equal_math',
               'unit_scale_align_to_first',
               'array_with_unit',
 
               # _base_quantity
               'Quantity',
-              'compatible_with_equinox',
 
               # _base_decorators
               'check_dims',
@@ -140,6 +166,11 @@ __all__ = [
               # _celsius
               'celsius2kelvin',
               'kelvin2celsius',
+              'fahrenheit2kelvin',
+              'kelvin2fahrenheit',
+
+              # _matplotlib_compat
+              'enable_matplotlib_support',
 
               # old version compatibility
               'avogadro_constant',

brainunit-0.3.0/brainunit/_backend.py

@@ -0,0 +1,17 @@
+# Copyright 2026 BrainX Ecosystem Limited. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+from saiunit._backend import *
+from saiunit._backend import __all__

brainunit-0.3.0/brainunit/_exceptions.py

@@ -0,0 +1,17 @@
+# Copyright 2026 BrainX Ecosystem Limited. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+from saiunit._exceptions import *
+from saiunit._exceptions import __all__

{brainunit-0.2.1 → brainunit-0.3.0}/brainunit/linalg/__init__.py

@@ -17,9 +17,13 @@ from ._linalg_change_unit import *
 from ._linalg_change_unit import __all__ as _linalg_change_unit_all
 from ._linalg_keep_unit import *
 from ._linalg_keep_unit import __all__ as _linalg_keep_unit_all
+from ._linalg_remove_unit import *
+from ._linalg_remove_unit import __all__ as _linalg_remove_unit_all
 
 __all__ = (_linalg_change_unit_all +
-              _linalg_keep_unit_all)
+              _linalg_keep_unit_all +
+              _linalg_remove_unit_all)
 
 del (_linalg_change_unit_all,
-     _linalg_keep_unit_all)
+     _linalg_keep_unit_all,
+     _linalg_remove_unit_all)

{brainunit-0.2.1 → brainunit-0.3.0}/brainunit.egg-info/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: brainunit
-Version: 0.2.1
+Version: 0.3.0
 Summary: Physical Units and Unit-Aware Mathematical System for General-Purpose Brain Dynamics Modeling
 Author-email: BrainUnit Developers <chao.brain@qq.com>
 License-Expression: Apache-2.0
 Project-URL: homepage, https://github.com/chaobrain/brainunit
 Project-URL: repository, https://github.com/chaobrain/brainunit
 Project-URL: Bug Tracker, https://github.com/chaobrain/brainunit/issues
-Project-URL: Documentation, https://brainunit.readthedocs.io/
+Project-URL: Documentation, https://brainx.chaobrain.com/brainunit/
 Keywords: physical unit,physical quantity,brain modeling
 Classifier: Natural Language :: English
 Classifier: Operating System :: OS Independent
@@ -30,10 +30,12 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: numpy
 Requires-Dist: typing_extensions
-Requires-Dist: saiunit==0.2.1
+Requires-Dist: saiunit==0.3.0
 Provides-Extra: testing
 Requires-Dist: pytest; extra == "testing"
 Requires-Dist: brainstate; extra == "testing"
+Provides-Extra: jax
+Requires-Dist: jax; extra == "jax"
 Provides-Extra: cpu
 Requires-Dist: jax[cpu]; extra == "cpu"
 Provides-Extra: cuda12
@@ -42,12 +44,22 @@ Provides-Extra: cuda13
 Requires-Dist: jax[cuda13]; extra == "cuda13"
 Provides-Extra: tpu
 Requires-Dist: jax[tpu]; extra == "tpu"
+Provides-Extra: cupy
+Requires-Dist: cupy-cuda12x>=13.0; extra == "cupy"
+Provides-Extra: torch
+Requires-Dist: torch>=2.0; extra == "torch"
+Provides-Extra: dask
+Requires-Dist: dask[array]>=2024.1; extra == "dask"
+Provides-Extra: ndonnx
+Requires-Dist: ndonnx>=0.9; extra == "ndonnx"
+Provides-Extra: all
+Requires-Dist: brainunit[cupy,dask,jax,ndonnx,torch]; extra == "all"
 
 <h1 align='center'>BrainUnit</h1>
 <h2 align='center'>Physical units and unit-aware math system for general-purpose brain dynamics modeling</h2>
 
 <p align="center">
-  	<img alt="Header image of brainunit." src="https://raw.githubusercontent.com/chaobrain/saiunit/main/docs/_static/brainunit.png" width=30%>
+  	<img alt="Header image of brainunit." src="https://brainx.chaobrain.com/images/brainunit.webp" width=30%>
 </p> 
 
 
@@ -55,7 +67,7 @@ Requires-Dist: jax[tpu]; extra == "tpu"
 <p align="center">
 	<a href="https://pypi.org/project/brainunit/"><img alt="Supported Python Version" src="https://img.shields.io/pypi/pyversions/brainunit"></a>
 	<a href="https://github.com/chaobrain/brainunit/blob/main/LICENSE"><img alt="LICENSE" src="https://img.shields.io/badge/License-Apache%202.0-blue.svg"></a>
-    <a href='https://brainunit.readthedocs.io/?badge=latest'>
+    <a href='https://brainx.chaobrain.com/braintools/'>
         <img src='https://readthedocs.org/projects/brainunit/badge/?version=latest' alt='Documentation Status' />
     </a>  	
     <a href="https://badge.fury.io/py/brainunit"><img alt="PyPI version" src="https://badge.fury.io/py/brainunit.svg"></a>
@@ -67,7 +79,7 @@ Requires-Dist: jax[tpu]; extra == "tpu"
 
 [BrainUnit](https://github.com/chaobrain/brainunit) provides physical units and unit-aware mathematical system in JAX for brain dynamics modeling. It introduces rigoirous physical units into high-performance AI-driven abstract numerical computing. 
 
-BrainUnit is initially designed to enable unit-aware computations in brain dynamics modeling (see our [BDP ecosystem](https://ecosystem-for-brain-dynamics.readthedocs.io/)). However, its features and capacities can be applied to general domains in scientific computing and AI4Science. Starting in 2025/02, BrainUnit has been fully integrated into [SAIUnit](https://github.com/chaobrain/saiunit) (the **Unit** system for **S**cientific **AI**). 
+BrainUnit is initially designed to enable unit-aware computations in brain dynamics modeling (see our [BDP ecosystem](https://brainx.chaobrain.com/)). However, its features and capacities can be applied to general domains in scientific computing and AI4Science. Starting in 2025/02, BrainUnit has been fully integrated into [SAIUnit](https://github.com/chaobrain/saiunit) (the **Unit** system for **S**cientific **AI**). 
 
 Functionalities are the same for both ``brainunit`` and ``saiunit``, and their functions and data structures are interoperable, sharing the same set of APIs, and eliminating any potential conflicts. This meas that 
 
@@ -87,13 +99,13 @@ For users primarily engaged in general scientific computing, `saiunit` is likely
 
 ## Documentation
 
-The official documentation of BrainUnit is hosted on Read the Docs: [https://brainunit.readthedocs.io](https://brainunit.readthedocs.io)
+The official documentation of BrainUnit is hosted on Read the Docs: [https://brainx.chaobrain.com/brainunit](https://brainx.chaobrain.com/brainunit)
 
 
 
 ## Features
 
-`brainunit` can be seamlessly integrated into every aspect of our [brain modeling ecosystem](https://brainmodeling.readthedocs.io/), such as, the checkpointing of [braintools](https://github.com/chaobrain/braintools), the event-driven operators in [brainevent](https://github.com/chaobrain/brainevent), the state-based JIT compilation in [brainstate](https://github.com/chaobrain/brainstate), online learning rules in [brainscale](https://github.com/chaobrain/brainscale), or event more. 
+`brainunit` can be seamlessly integrated into every aspect of our [brain modeling ecosystem](https://brainx.chaobrain.com/), such as, the checkpointing of [braintools](https://github.com/chaobrain/braintools), the event-driven operators in [brainevent](https://github.com/chaobrain/brainevent), the state-based JIT compilation in [brainstate](https://github.com/chaobrain/brainstate), online learning rules in [brainscale](https://github.com/chaobrain/brainscale), or event more. 
 
 A quick example for this kind of integration:
 
@@ -175,4 +187,4 @@ If you use `brainunit` in your research, please consider citing the following pa
 
 ## See also the ecosystem
 
-``brainunit`` is one part of our [brain modeling ecosystem](https://brainmodeling.readthedocs.io/). 
+``brainunit`` is one part of our [brain modeling ecosystem](https://brainx.chaobrain.com/). 

{brainunit-0.2.1 → brainunit-0.3.0}/brainunit.egg-info/SOURCES.txt

@@ -1,18 +1,21 @@
 README.md
 pyproject.toml
 brainunit/__init__.py
+brainunit/_backend.py
 brainunit/_base_decorators.py
 brainunit/_base_dimension.py
 brainunit/_base_getters.py
 brainunit/_base_quantity.py
 brainunit/_base_unit.py
 brainunit/_celsius.py
+brainunit/_exceptions.py
 brainunit/_misc.py
 brainunit/_unit_common.py
 brainunit/_unit_constants.py
 brainunit/_unit_shortcuts.py
 brainunit/constants.py
 brainunit/custom_array.py
+brainunit/py.typed
 brainunit/typing.py
 brainunit.egg-info/PKG-INFO
 brainunit.egg-info/SOURCES.txt

brainunit-0.3.0/brainunit.egg-info/requires.txt

@@ -0,0 +1,37 @@
+numpy
+typing_extensions
+saiunit==0.3.0
+
+[all]
+brainunit[cupy,dask,jax,ndonnx,torch]
+
+[cpu]
+jax[cpu]
+
+[cuda12]
+jax[cuda12]
+
+[cuda13]
+jax[cuda13]
+
+[cupy]
+cupy-cuda12x>=13.0
+
+[dask]
+dask[array]>=2024.1
+
+[jax]
+jax
+
+[ndonnx]
+ndonnx>=0.9
+
+[testing]
+pytest
+brainstate
+
+[torch]
+torch>=2.0
+
+[tpu]
+jax[tpu]

{brainunit-0.2.1 → brainunit-0.3.0}/pyproject.toml

@@ -14,9 +14,12 @@ exclude = [
     "brainunit/__init__.py"
 ]
 
+[tool.setuptools.package-data]
+brainunit = ["py.typed"]
+
 [project]
 name = "brainunit"
-version = "0.2.1"
+version = "0.3.0"
 description = "Physical Units and Unit-Aware Mathematical System for General-Purpose Brain Dynamics Modeling"
 readme = 'README.md'
 license = 'Apache-2.0'
@@ -48,7 +51,7 @@ keywords = ['physical unit', 'physical quantity', 'brain modeling']
 dependencies = [
     'numpy',
     'typing_extensions',
-    'saiunit==0.2.1',
+    'saiunit==0.3.0',
 ]
 
 [tool.flit.module]
@@ -58,11 +61,17 @@ name = "brainunit"
 homepage = 'https://github.com/chaobrain/brainunit'
 repository = 'https://github.com/chaobrain/brainunit'
 "Bug Tracker" = "https://github.com/chaobrain/brainunit/issues"
-"Documentation" = "https://brainunit.readthedocs.io/"
+"Documentation" = "https://brainx.chaobrain.com/brainunit/"
 
 [project.optional-dependencies]
 testing = ["pytest", "brainstate"]
+jax = ["jax"]
 cpu = ["jax[cpu]"]
 cuda12 = ["jax[cuda12]"]
 cuda13 = ["jax[cuda13]"]
 tpu = ["jax[tpu]"]
+cupy = ["cupy-cuda12x>=13.0"]
+torch = ["torch>=2.0"]
+dask = ["dask[array]>=2024.1"]
+ndonnx = ["ndonnx>=0.9"]
+all = ["brainunit[jax,cupy,torch,dask,ndonnx]"]

brainunit-0.2.1/brainunit.egg-info/requires.txt

@@ -1,19 +0,0 @@
-numpy
-typing_extensions
-saiunit==0.2.1
-
-[cpu]
-jax[cpu]
-
-[cuda12]
-jax[cuda12]
-
-[cuda13]
-jax[cuda13]
-
-[testing]
-pytest
-brainstate
-
-[tpu]
-jax[tpu]