brainunit 0.0.16__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (48) hide show
  1. brainunit/__init__.py +65 -0
  2. brainunit/_base.py +20 -0
  3. brainunit/_celsius.py +20 -0
  4. brainunit/_unit_common.py +19 -0
  5. brainunit/_unit_constants.py +20 -0
  6. brainunit/_unit_shortcuts.py +20 -0
  7. brainunit/autograd/__init__.py +30 -0
  8. brainunit/autograd/_hessian.py +19 -0
  9. brainunit/autograd/_jacobian.py +19 -0
  10. brainunit/autograd/_value_and_grad.py +19 -0
  11. brainunit/autograd/_vector_grad.py +19 -0
  12. brainunit/constants.py +20 -0
  13. brainunit/fft/__init__.py +25 -0
  14. brainunit/fft/_fft_change_unit.py +19 -0
  15. brainunit/fft/_fft_keep_unit.py +20 -0
  16. brainunit/lax/__init__.py +45 -0
  17. brainunit/lax/_lax_accept_unitless.py +19 -0
  18. brainunit/lax/_lax_array_creation.py +19 -0
  19. brainunit/lax/_lax_change_unit.py +20 -0
  20. brainunit/lax/_lax_keep_unit.py +20 -0
  21. brainunit/lax/_lax_linalg.py +20 -0
  22. brainunit/lax/_lax_remove_unit.py +20 -0
  23. brainunit/lax/_misc.py +20 -0
  24. brainunit/linalg/__init__.py +25 -0
  25. brainunit/linalg/_linalg_change_unit.py +19 -0
  26. brainunit/linalg/_linalg_keep_unit.py +19 -0
  27. brainunit/linalg/_linalg_remove_unit.py +20 -0
  28. brainunit/math/__init__.py +61 -0
  29. brainunit/math/_activation.py +20 -0
  30. brainunit/math/_alias.py +20 -0
  31. brainunit/math/_einops.py +20 -0
  32. brainunit/math/_fun_accept_unitless.py +20 -0
  33. brainunit/math/_fun_array_creation.py +20 -0
  34. brainunit/math/_fun_change_unit.py +20 -0
  35. brainunit/math/_fun_keep_unit.py +20 -0
  36. brainunit/math/_fun_remove_unit.py +20 -0
  37. brainunit/math/_misc.py +20 -0
  38. brainunit/math/fft.py +20 -0
  39. brainunit/math/linalg.py +20 -0
  40. brainunit/sparse/__init__.py +22 -0
  41. brainunit/sparse/_block_csr.py +20 -0
  42. brainunit/sparse/_block_ell.py +20 -0
  43. brainunit/sparse/_coo.py +20 -0
  44. brainunit/sparse/_csr.py +20 -0
  45. brainunit-0.0.16.dist-info/METADATA +174 -0
  46. brainunit-0.0.16.dist-info/RECORD +48 -0
  47. brainunit-0.0.16.dist-info/WHEEL +5 -0
  48. brainunit-0.0.16.dist-info/top_level.txt +1 -0
brainunit/__init__.py ADDED
@@ -0,0 +1,65 @@
1
+ # Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+
17
+ import saiunit
18
+
19
+ __version__ = saiunit.__version__
20
+
21
+ from . import autograd
22
+ from . import constants
23
+ from . import fft
24
+ from . import lax
25
+ from . import linalg
26
+ from . import math
27
+ from . import sparse
28
+ from ._base import *
29
+ from ._base import __all__ as _base_all
30
+ from ._celsius import *
31
+ from ._celsius import __all__ as _celsius_all
32
+ from ._unit_common import *
33
+ from ._unit_common import __all__ as _common_all
34
+ from ._unit_constants import *
35
+ from ._unit_constants import __all__ as _constants_all
36
+ from ._unit_shortcuts import *
37
+ from ._unit_shortcuts import __all__ as _std_units_all
38
+
39
+ __all__ = (
40
+ [
41
+ 'math',
42
+ 'linalg',
43
+ 'autograd',
44
+ 'fft',
45
+ 'constants',
46
+ 'sparse'
47
+ ] +
48
+ _common_all +
49
+ _std_units_all +
50
+ _base_all +
51
+ _constants_all +
52
+ _celsius_all
53
+ )
54
+ del _common_all, _std_units_all, _base_all, _celsius_all, _constants_all, saiunit
55
+
56
+ # old version compatibility
57
+ avogadro_constant = constants.avogadro
58
+ boltzmann_constant = constants.boltzmann
59
+ electric_constant = constants.electric
60
+ electron_mass = constants.electron_mass
61
+ elementary_charge = constants.elementary_charge
62
+ faraday_constant = constants.faraday
63
+ gas_constant = constants.gas
64
+ magnetic_constant = constants.magnetic
65
+ molar_mass_constant = constants.molar_mass
brainunit/_base.py ADDED
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit._base import *
19
+ from saiunit._base import __all__
20
+
brainunit/_celsius.py ADDED
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit._celsius import *
19
+ from saiunit._celsius import __all__
20
+
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit._unit_common import *
19
+ from saiunit._unit_common import __all__
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit._unit_constants import *
19
+ from saiunit._unit_constants import __all__
20
+
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit._unit_shortcuts import *
19
+ from saiunit._unit_shortcuts import __all__
20
+
@@ -0,0 +1,30 @@
1
+ # Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+
17
+ __all__ = [
18
+ 'value_and_grad',
19
+ 'grad',
20
+ 'vector_grad',
21
+ 'jacobian',
22
+ 'jacrev',
23
+ 'jacfwd',
24
+ 'hessian',
25
+ ]
26
+
27
+ from ._hessian import *
28
+ from ._jacobian import *
29
+ from ._value_and_grad import *
30
+ from ._vector_grad import *
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.autograd._hessian import *
19
+ from saiunit.autograd._hessian import __all__
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.autograd._jacobian import *
19
+ from saiunit.autograd._jacobian import __all__
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.autograd._value_and_grad import *
19
+ from saiunit.autograd._value_and_grad import __all__
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.autograd._vector_grad import *
19
+ from saiunit.autograd._vector_grad import __all__
brainunit/constants.py ADDED
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.constants import *
19
+ from saiunit.constants import __all__
20
+
@@ -0,0 +1,25 @@
1
+ # Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ from ._fft_change_unit import *
17
+ from ._fft_change_unit import __all__ as _fft_change_unit_all
18
+ from ._fft_keep_unit import *
19
+ from ._fft_keep_unit import __all__ as _fft_keep_unit_all
20
+
21
+ __all__ = (_fft_change_unit_all +
22
+ _fft_keep_unit_all)
23
+
24
+ del (_fft_change_unit_all,
25
+ _fft_keep_unit_all,)
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.fft._fft_change_unit import *
19
+ from saiunit.fft._fft_change_unit import __all__
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.fft._fft_keep_unit import *
19
+ from saiunit.fft._fft_keep_unit import __all__
20
+
@@ -0,0 +1,45 @@
1
+ # Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ from ._lax_accept_unitless import *
17
+ from ._lax_accept_unitless import __all__ as _lax_accept_unitless_all
18
+ from ._lax_array_creation import *
19
+ from ._lax_array_creation import __all__ as _lax_array_creation_all
20
+ from ._lax_change_unit import *
21
+ from ._lax_change_unit import __all__ as _lax_change_unit_all
22
+ from ._lax_keep_unit import *
23
+ from ._lax_keep_unit import __all__ as _lax_keep_unit_all
24
+ from ._lax_linalg import *
25
+ from ._lax_linalg import __all__ as _lax_linalg_all
26
+ from ._lax_remove_unit import *
27
+ from ._lax_remove_unit import __all__ as _lax_remove_unit_all
28
+ from ._misc import *
29
+ from ._misc import __all__ as _lax_misc_all
30
+
31
+ __all__ = (_lax_accept_unitless_all +
32
+ _lax_array_creation_all +
33
+ _lax_change_unit_all +
34
+ _lax_keep_unit_all +
35
+ _lax_linalg_all +
36
+ _lax_remove_unit_all +
37
+ _lax_misc_all)
38
+
39
+ del (_lax_accept_unitless_all,
40
+ _lax_array_creation_all,
41
+ _lax_change_unit_all,
42
+ _lax_keep_unit_all,
43
+ _lax_linalg_all,
44
+ _lax_remove_unit_all,
45
+ _lax_misc_all)
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.lax._lax_accept_unitless import *
19
+ from saiunit.lax._lax_accept_unitless import __all__
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.lax._lax_array_creation import *
19
+ from saiunit.lax._lax_array_creation import __all__
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.lax._lax_change_unit import *
19
+ from saiunit.lax._lax_change_unit import __all__
20
+
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.lax._lax_keep_unit import *
19
+ from saiunit.lax._lax_keep_unit import __all__
20
+
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.lax._lax_linalg import *
19
+ from saiunit.lax._lax_linalg import __all__
20
+
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.lax._lax_remove_unit import *
19
+ from saiunit.lax._lax_remove_unit import __all__
20
+
brainunit/lax/_misc.py ADDED
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.lax._misc import *
19
+ from saiunit.lax._misc import __all__
20
+
@@ -0,0 +1,25 @@
1
+ # Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ from ._linalg_change_unit import *
17
+ from ._linalg_change_unit import __all__ as _linalg_change_unit_all
18
+ from ._linalg_keep_unit import *
19
+ from ._linalg_keep_unit import __all__ as _linalg_keep_unit_all
20
+
21
+ __all__ = (_linalg_change_unit_all +
22
+ _linalg_keep_unit_all)
23
+
24
+ del (_linalg_change_unit_all,
25
+ _linalg_keep_unit_all)
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.linalg._linalg_change_unit import *
19
+ from saiunit.linalg._linalg_change_unit import __all__
@@ -0,0 +1,19 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.linalg._linalg_keep_unit import *
19
+ from saiunit.linalg._linalg_keep_unit import __all__
@@ -0,0 +1,20 @@
1
+ # Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+
16
+ # -*- coding: utf-8 -*-
17
+
18
+ from saiunit.linalg._linalg_remove_unit import *
19
+ from saiunit.linalg._linalg_remove_unit import __all__
20
+