brainunit 0.2.0__tar.gz → 0.2.2__tar.gz

{brainunit-0.2.0 → brainunit-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brainunit
-Version: 0.2.0
+Version: 0.2.2
 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
@@ -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.0
+Requires-Dist: saiunit==0.2.2
 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,6 +44,16 @@ 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>

{brainunit-0.2.0 → brainunit-0.2.2}/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,6 +61,7 @@ from ._base_getters import (
     have_same_dim,
     is_dimensionless,
     is_scalar_type,
+    is_unit_equal_math,
     is_unitless,
     maybe_decimal,
     split_mantissa_unit,
@@ -55,7 +69,7 @@ from ._base_getters import (
 )
 from ._base_quantity import Quantity, compatible_with_equinox
 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,6 +151,7 @@ __all__ = [
               'assert_quantity',
               'have_same_dim',
               'has_same_unit',
+              'is_unit_equal_math',
               'unit_scale_align_to_first',
               'array_with_unit',
 
@@ -140,6 +167,11 @@ __all__ = [
               # _celsius
               'celsius2kelvin',
               'kelvin2celsius',
+              'fahrenheit2kelvin',
+              'kelvin2fahrenheit',
+
+              # _matplotlib_compat
+              'enable_matplotlib_support',
 
               # old version compatibility
               'avogadro_constant',

brainunit-0.2.2/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.2.2/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.0 → brainunit-0.2.2}/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.0 → brainunit-0.2.2}/brainunit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brainunit
-Version: 0.2.0
+Version: 0.2.2
 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
@@ -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.0
+Requires-Dist: saiunit==0.2.2
 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,6 +44,16 @@ 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>

{brainunit-0.2.0 → brainunit-0.2.2}/brainunit.egg-info/SOURCES.txt

@@ -1,12 +1,14 @@
 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
@@ -52,7 +54,5 @@ brainunit/math/_misc.py
 brainunit/math/fft.py
 brainunit/math/linalg.py
 brainunit/sparse/__init__.py
-brainunit/sparse/_block_csr.py
-brainunit/sparse/_block_ell.py
 brainunit/sparse/_coo.py
 brainunit/sparse/_csr.py

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

@@ -0,0 +1,37 @@
+numpy
+typing_extensions
+saiunit==0.2.2
+
+[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.0 → brainunit-0.2.2}/pyproject.toml

@@ -16,7 +16,7 @@ exclude = [
 
 [project]
 name = "brainunit"
-version = "0.2.0"
+version = "0.2.2"
 description = "Physical Units and Unit-Aware Mathematical System for General-Purpose Brain Dynamics Modeling"
 readme = 'README.md'
 license = 'Apache-2.0'
@@ -48,7 +48,7 @@ keywords = ['physical unit', 'physical quantity', 'brain modeling']
 dependencies = [
     'numpy',
     'typing_extensions',
-    'saiunit==0.2.0',
+    'saiunit==0.2.2',
 ]
 
 [tool.flit.module]
@@ -62,7 +62,13 @@ repository = 'https://github.com/chaobrain/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.0/brainunit/sparse/_block_csr.py

@@ -1,20 +0,0 @@
-# Copyright 2025 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.
-# ==============================================================================
-
-# -*- coding: utf-8 -*-
-
-from saiunit.sparse._block_csr import *
-from saiunit.sparse._block_csr import __all__
-

brainunit-0.2.0/brainunit/sparse/_block_ell.py

@@ -1,20 +0,0 @@
-# Copyright 2025 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.
-# ==============================================================================
-
-# -*- coding: utf-8 -*-
-
-from saiunit.sparse._block_ell import *
-from saiunit.sparse._block_ell import __all__
-

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

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