brainstate 0.2.1__py2.py3-none-any.whl → 0.2.2__py2.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.
- brainstate/__init__.py +167 -169
- brainstate/_compatible_import.py +340 -340
- brainstate/_compatible_import_test.py +681 -681
- brainstate/_deprecation.py +210 -210
- brainstate/_deprecation_test.py +2297 -2319
- brainstate/_error.py +45 -45
- brainstate/_state.py +2157 -1652
- brainstate/_state_test.py +1129 -52
- brainstate/_utils.py +47 -47
- brainstate/environ.py +1495 -1495
- brainstate/environ_test.py +1223 -1223
- brainstate/graph/__init__.py +22 -22
- brainstate/graph/_node.py +240 -240
- brainstate/graph/_node_test.py +589 -589
- brainstate/graph/_operation.py +1620 -1624
- brainstate/graph/_operation_test.py +1147 -1147
- brainstate/mixin.py +1447 -1433
- brainstate/mixin_test.py +1017 -1017
- brainstate/nn/__init__.py +146 -137
- brainstate/nn/_activations.py +1100 -1100
- brainstate/nn/_activations_test.py +354 -354
- brainstate/nn/_collective_ops.py +635 -633
- brainstate/nn/_collective_ops_test.py +774 -774
- brainstate/nn/_common.py +226 -226
- brainstate/nn/_common_test.py +134 -154
- brainstate/nn/_conv.py +2010 -2010
- brainstate/nn/_conv_test.py +849 -849
- brainstate/nn/_delay.py +575 -575
- brainstate/nn/_delay_test.py +243 -243
- brainstate/nn/_dropout.py +618 -618
- brainstate/nn/_dropout_test.py +480 -477
- brainstate/nn/_dynamics.py +870 -1267
- brainstate/nn/_dynamics_test.py +53 -67
- brainstate/nn/_elementwise.py +1298 -1298
- brainstate/nn/_elementwise_test.py +829 -829
- brainstate/nn/_embedding.py +408 -408
- brainstate/nn/_embedding_test.py +156 -156
- brainstate/nn/_event_fixedprob.py +233 -233
- brainstate/nn/_event_fixedprob_test.py +115 -115
- brainstate/nn/_event_linear.py +83 -83
- brainstate/nn/_event_linear_test.py +121 -121
- brainstate/nn/_exp_euler.py +254 -254
- brainstate/nn/_exp_euler_test.py +377 -377
- brainstate/nn/_linear.py +744 -744
- brainstate/nn/_linear_test.py +475 -475
- brainstate/nn/_metrics.py +1070 -1070
- brainstate/nn/_metrics_test.py +611 -611
- brainstate/nn/_module.py +391 -384
- brainstate/nn/_module_test.py +40 -40
- brainstate/nn/_normalizations.py +1334 -1334
- brainstate/nn/_normalizations_test.py +699 -699
- brainstate/nn/_paddings.py +1020 -1020
- brainstate/nn/_paddings_test.py +722 -722
- brainstate/nn/_poolings.py +2239 -2239
- brainstate/nn/_poolings_test.py +952 -952
- brainstate/nn/_rnns.py +946 -946
- brainstate/nn/_rnns_test.py +592 -592
- brainstate/nn/_utils.py +216 -216
- brainstate/nn/_utils_test.py +401 -401
- brainstate/nn/init.py +809 -809
- brainstate/nn/init_test.py +180 -180
- brainstate/random/__init__.py +270 -270
- brainstate/random/{_rand_funs.py → _fun.py} +3938 -3938
- brainstate/random/{_rand_funs_test.py → _fun_test.py} +638 -640
- brainstate/random/_impl.py +672 -0
- brainstate/random/{_rand_seed.py → _seed.py} +675 -675
- brainstate/random/{_rand_seed_test.py → _seed_test.py} +48 -48
- brainstate/random/{_rand_state.py → _state.py} +1320 -1617
- brainstate/random/{_rand_state_test.py → _state_test.py} +551 -551
- brainstate/transform/__init__.py +56 -59
- brainstate/transform/_ad_checkpoint.py +176 -176
- brainstate/transform/_ad_checkpoint_test.py +49 -49
- brainstate/transform/_autograd.py +1025 -1025
- brainstate/transform/_autograd_test.py +1289 -1289
- brainstate/transform/_conditions.py +316 -316
- brainstate/transform/_conditions_test.py +220 -220
- brainstate/transform/_error_if.py +94 -94
- brainstate/transform/_error_if_test.py +52 -52
- brainstate/transform/_find_state.py +200 -0
- brainstate/transform/_find_state_test.py +84 -0
- brainstate/transform/_jit.py +399 -399
- brainstate/transform/_jit_test.py +143 -143
- brainstate/transform/_loop_collect_return.py +675 -675
- brainstate/transform/_loop_collect_return_test.py +58 -58
- brainstate/transform/_loop_no_collection.py +283 -283
- brainstate/transform/_loop_no_collection_test.py +50 -50
- brainstate/transform/_make_jaxpr.py +2176 -2016
- brainstate/transform/_make_jaxpr_test.py +1634 -1510
- brainstate/transform/_mapping.py +607 -529
- brainstate/transform/_mapping_test.py +104 -194
- brainstate/transform/_progress_bar.py +255 -255
- brainstate/transform/_unvmap.py +256 -256
- brainstate/transform/_util.py +286 -286
- brainstate/typing.py +837 -837
- brainstate/typing_test.py +780 -780
- brainstate/util/__init__.py +27 -27
- brainstate/util/_others.py +1024 -1024
- brainstate/util/_others_test.py +962 -962
- brainstate/util/_pretty_pytree.py +1301 -1301
- brainstate/util/_pretty_pytree_test.py +675 -675
- brainstate/util/_pretty_repr.py +462 -462
- brainstate/util/_pretty_repr_test.py +696 -696
- brainstate/util/filter.py +945 -945
- brainstate/util/filter_test.py +911 -911
- brainstate/util/struct.py +910 -910
- brainstate/util/struct_test.py +602 -602
- {brainstate-0.2.1.dist-info → brainstate-0.2.2.dist-info}/METADATA +108 -108
- brainstate-0.2.2.dist-info/RECORD +111 -0
- {brainstate-0.2.1.dist-info → brainstate-0.2.2.dist-info}/licenses/LICENSE +202 -202
- brainstate/transform/_eval_shape.py +0 -145
- brainstate/transform/_eval_shape_test.py +0 -38
- brainstate/transform/_random.py +0 -171
- brainstate-0.2.1.dist-info/RECORD +0 -111
- {brainstate-0.2.1.dist-info → brainstate-0.2.2.dist-info}/WHEEL +0 -0
- {brainstate-0.2.1.dist-info → brainstate-0.2.2.dist-info}/top_level.txt +0 -0
@@ -1,108 +1,108 @@
|
|
1
|
-
Metadata-Version: 2.4
|
2
|
-
Name: brainstate
|
3
|
-
Version: 0.2.
|
4
|
-
Summary: A State-based Transformation System for Brain Modeling.
|
5
|
-
Author-email: BrainState Developers <chao.brain@qq.com>
|
6
|
-
License: Apache-2.0 license
|
7
|
-
Project-URL: homepage, https://github.com/chaobrain/brainstate
|
8
|
-
Project-URL: repository, https://github.com/chaobrain/brainstate
|
9
|
-
Project-URL: Documentation, https://brainstate.readthedocs.io/
|
10
|
-
Project-URL: Source Code, https://github.com/chaobrain/brainstate
|
11
|
-
Project-URL: Bug Tracker, https://github.com/chaobrain/brainstate/issues
|
12
|
-
Keywords: computational neuroscience,brain-inspired computing,brain simulation,brain modeling,spiking neural networks
|
13
|
-
Classifier: Natural Language :: English
|
14
|
-
Classifier: Operating System :: OS Independent
|
15
|
-
Classifier: Development Status :: 4 - Beta
|
16
|
-
Classifier: Intended Audience :: Developers
|
17
|
-
Classifier: Intended Audience :: Science/Research
|
18
|
-
Classifier: Programming Language :: Python :: 3
|
19
|
-
Classifier: Programming Language :: Python :: 3.10
|
20
|
-
Classifier: Programming Language :: Python :: 3.11
|
21
|
-
Classifier: Programming Language :: Python :: 3.12
|
22
|
-
Classifier: Programming Language :: Python :: 3.13
|
23
|
-
Classifier: License :: OSI Approved :: Apache Software License
|
24
|
-
Classifier: Programming Language :: Python
|
25
|
-
Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
|
26
|
-
Classifier: Topic :: Scientific/Engineering :: Mathematics
|
27
|
-
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
28
|
-
Classifier: Topic :: Software Development :: Libraries
|
29
|
-
Requires-Python: >=3.10
|
30
|
-
Description-Content-Type: text/markdown
|
31
|
-
License-File: LICENSE
|
32
|
-
Requires-Dist: numpy>=1.15
|
33
|
-
Requires-Dist: tqdm
|
34
|
-
Requires-Dist: brainunit
|
35
|
-
Requires-Dist: brainevent
|
36
|
-
Provides-Extra: cpu
|
37
|
-
Requires-Dist: jax[cpu]; extra == "cpu"
|
38
|
-
Requires-Dist: brainunit; extra == "cpu"
|
39
|
-
Requires-Dist: brainevent; extra == "cpu"
|
40
|
-
Provides-Extra: cuda12
|
41
|
-
Requires-Dist: jax[cuda12]; extra == "cuda12"
|
42
|
-
Requires-Dist: brainunit; extra == "cuda12"
|
43
|
-
Requires-Dist: brainevent; extra == "cuda12"
|
44
|
-
Provides-Extra: cuda13
|
45
|
-
Requires-Dist: jax[cuda13]; extra == "cuda13"
|
46
|
-
Requires-Dist: brainunit; extra == "cuda13"
|
47
|
-
Requires-Dist: brainevent; extra == "cuda13"
|
48
|
-
Provides-Extra: tpu
|
49
|
-
Requires-Dist: jax[tpu]; extra == "tpu"
|
50
|
-
Requires-Dist: brainunit; extra == "tpu"
|
51
|
-
Requires-Dist: brainevent; extra == "tpu"
|
52
|
-
Provides-Extra: testing
|
53
|
-
Requires-Dist: absl-py; extra == "testing"
|
54
|
-
Requires-Dist: pytest; extra == "testing"
|
55
|
-
Requires-Dist: jax; extra == "testing"
|
56
|
-
Requires-Dist: brainunit; extra == "testing"
|
57
|
-
Requires-Dist: brainevent; extra == "testing"
|
58
|
-
Dynamic: license-file
|
59
|
-
|
60
|
-
|
61
|
-
# A ``State``-based Transformation System for Brain Modeling
|
62
|
-
|
63
|
-
|
64
|
-
|
65
|
-
<p align="center">
|
66
|
-
<img alt="Header image of brainstate." src="https://raw.githubusercontent.com/chaobrain/brainstate/main/docs/_static/brainstate.png" width=40%>
|
67
|
-
</p>
|
68
|
-
|
69
|
-
|
70
|
-
|
71
|
-
<p align="center">
|
72
|
-
<a href="https://pypi.org/project/brainstate/"><img alt="Supported Python Version" src="https://img.shields.io/pypi/pyversions/brainstate"></a>
|
73
|
-
<a href="https://github.com/chaobrain/brainstate/blob/main/LICENSE"><img alt="LICENSE" src="https://img.shields.io/badge/License-Apache%202.0-blue.svg"></a>
|
74
|
-
<a href='https://brainstate.readthedocs.io/?badge=latest'>
|
75
|
-
<img src='https://readthedocs.org/projects/brainstate/badge/?version=latest' alt='Documentation Status' />
|
76
|
-
</a>
|
77
|
-
<a href="https://badge.fury.io/py/brainstate"><img alt="PyPI version" src="https://badge.fury.io/py/brainstate.svg"></a>
|
78
|
-
<a href="https://github.com/chaobrain/brainstate/actions/workflows/CI.yml"><img alt="Continuous Integration" src="https://github.com/chaobrain/brainstate/actions/workflows/CI.yml/badge.svg"></a>
|
79
|
-
<a href="https://pepy.tech/projects/brainstate"><img src="https://static.pepy.tech/badge/brainstate" alt="PyPI Downloads"></a>
|
80
|
-
<a href="https://doi.org/10.5281/zenodo.14970015"><img src="https://zenodo.org/badge/811300394.svg" alt="DOI"></a>
|
81
|
-
</p>
|
82
|
-
|
83
|
-
|
84
|
-
## Installation
|
85
|
-
|
86
|
-
You can install ``brainstate`` via pip:
|
87
|
-
|
88
|
-
```bash
|
89
|
-
pip install brainstate --upgrade
|
90
|
-
```
|
91
|
-
|
92
|
-
Alternatively, you can install `BrainX`, which bundles `brainstate` with other compatible packages for a comprehensive brain modeling ecosystem:
|
93
|
-
|
94
|
-
```bash
|
95
|
-
pip install BrainX -U
|
96
|
-
```
|
97
|
-
|
98
|
-
## Documentation
|
99
|
-
|
100
|
-
The official documentation is hosted on Read the Docs: [https://brainstate.readthedocs.io/](https://brainstate.readthedocs.io/)
|
101
|
-
|
102
|
-
|
103
|
-
|
104
|
-
## See also the ecosystem
|
105
|
-
|
106
|
-
``brainstate`` is one part of our brain simulation ecosystem: https://brainmodeling.readthedocs.io/
|
107
|
-
|
108
|
-
|
1
|
+
Metadata-Version: 2.4
|
2
|
+
Name: brainstate
|
3
|
+
Version: 0.2.2
|
4
|
+
Summary: A State-based Transformation System for Brain Modeling.
|
5
|
+
Author-email: BrainState Developers <chao.brain@qq.com>
|
6
|
+
License: Apache-2.0 license
|
7
|
+
Project-URL: homepage, https://github.com/chaobrain/brainstate
|
8
|
+
Project-URL: repository, https://github.com/chaobrain/brainstate
|
9
|
+
Project-URL: Documentation, https://brainstate.readthedocs.io/
|
10
|
+
Project-URL: Source Code, https://github.com/chaobrain/brainstate
|
11
|
+
Project-URL: Bug Tracker, https://github.com/chaobrain/brainstate/issues
|
12
|
+
Keywords: computational neuroscience,brain-inspired computing,brain simulation,brain modeling,spiking neural networks
|
13
|
+
Classifier: Natural Language :: English
|
14
|
+
Classifier: Operating System :: OS Independent
|
15
|
+
Classifier: Development Status :: 4 - Beta
|
16
|
+
Classifier: Intended Audience :: Developers
|
17
|
+
Classifier: Intended Audience :: Science/Research
|
18
|
+
Classifier: Programming Language :: Python :: 3
|
19
|
+
Classifier: Programming Language :: Python :: 3.10
|
20
|
+
Classifier: Programming Language :: Python :: 3.11
|
21
|
+
Classifier: Programming Language :: Python :: 3.12
|
22
|
+
Classifier: Programming Language :: Python :: 3.13
|
23
|
+
Classifier: License :: OSI Approved :: Apache Software License
|
24
|
+
Classifier: Programming Language :: Python
|
25
|
+
Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
|
26
|
+
Classifier: Topic :: Scientific/Engineering :: Mathematics
|
27
|
+
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
28
|
+
Classifier: Topic :: Software Development :: Libraries
|
29
|
+
Requires-Python: >=3.10
|
30
|
+
Description-Content-Type: text/markdown
|
31
|
+
License-File: LICENSE
|
32
|
+
Requires-Dist: numpy>=1.15
|
33
|
+
Requires-Dist: tqdm
|
34
|
+
Requires-Dist: brainunit
|
35
|
+
Requires-Dist: brainevent
|
36
|
+
Provides-Extra: cpu
|
37
|
+
Requires-Dist: jax[cpu]; extra == "cpu"
|
38
|
+
Requires-Dist: brainunit; extra == "cpu"
|
39
|
+
Requires-Dist: brainevent; extra == "cpu"
|
40
|
+
Provides-Extra: cuda12
|
41
|
+
Requires-Dist: jax[cuda12]; extra == "cuda12"
|
42
|
+
Requires-Dist: brainunit; extra == "cuda12"
|
43
|
+
Requires-Dist: brainevent; extra == "cuda12"
|
44
|
+
Provides-Extra: cuda13
|
45
|
+
Requires-Dist: jax[cuda13]; extra == "cuda13"
|
46
|
+
Requires-Dist: brainunit; extra == "cuda13"
|
47
|
+
Requires-Dist: brainevent; extra == "cuda13"
|
48
|
+
Provides-Extra: tpu
|
49
|
+
Requires-Dist: jax[tpu]; extra == "tpu"
|
50
|
+
Requires-Dist: brainunit; extra == "tpu"
|
51
|
+
Requires-Dist: brainevent; extra == "tpu"
|
52
|
+
Provides-Extra: testing
|
53
|
+
Requires-Dist: absl-py; extra == "testing"
|
54
|
+
Requires-Dist: pytest; extra == "testing"
|
55
|
+
Requires-Dist: jax; extra == "testing"
|
56
|
+
Requires-Dist: brainunit; extra == "testing"
|
57
|
+
Requires-Dist: brainevent; extra == "testing"
|
58
|
+
Dynamic: license-file
|
59
|
+
|
60
|
+
|
61
|
+
# A ``State``-based Transformation System for Brain Modeling
|
62
|
+
|
63
|
+
|
64
|
+
|
65
|
+
<p align="center">
|
66
|
+
<img alt="Header image of brainstate." src="https://raw.githubusercontent.com/chaobrain/brainstate/main/docs/_static/brainstate.png" width=40%>
|
67
|
+
</p>
|
68
|
+
|
69
|
+
|
70
|
+
|
71
|
+
<p align="center">
|
72
|
+
<a href="https://pypi.org/project/brainstate/"><img alt="Supported Python Version" src="https://img.shields.io/pypi/pyversions/brainstate"></a>
|
73
|
+
<a href="https://github.com/chaobrain/brainstate/blob/main/LICENSE"><img alt="LICENSE" src="https://img.shields.io/badge/License-Apache%202.0-blue.svg"></a>
|
74
|
+
<a href='https://brainstate.readthedocs.io/?badge=latest'>
|
75
|
+
<img src='https://readthedocs.org/projects/brainstate/badge/?version=latest' alt='Documentation Status' />
|
76
|
+
</a>
|
77
|
+
<a href="https://badge.fury.io/py/brainstate"><img alt="PyPI version" src="https://badge.fury.io/py/brainstate.svg"></a>
|
78
|
+
<a href="https://github.com/chaobrain/brainstate/actions/workflows/CI.yml"><img alt="Continuous Integration" src="https://github.com/chaobrain/brainstate/actions/workflows/CI.yml/badge.svg"></a>
|
79
|
+
<a href="https://pepy.tech/projects/brainstate"><img src="https://static.pepy.tech/badge/brainstate" alt="PyPI Downloads"></a>
|
80
|
+
<a href="https://doi.org/10.5281/zenodo.14970015"><img src="https://zenodo.org/badge/811300394.svg" alt="DOI"></a>
|
81
|
+
</p>
|
82
|
+
|
83
|
+
|
84
|
+
## Installation
|
85
|
+
|
86
|
+
You can install ``brainstate`` via pip:
|
87
|
+
|
88
|
+
```bash
|
89
|
+
pip install brainstate --upgrade
|
90
|
+
```
|
91
|
+
|
92
|
+
Alternatively, you can install `BrainX`, which bundles `brainstate` with other compatible packages for a comprehensive brain modeling ecosystem:
|
93
|
+
|
94
|
+
```bash
|
95
|
+
pip install BrainX -U
|
96
|
+
```
|
97
|
+
|
98
|
+
## Documentation
|
99
|
+
|
100
|
+
The official documentation is hosted on Read the Docs: [https://brainstate.readthedocs.io/](https://brainstate.readthedocs.io/)
|
101
|
+
|
102
|
+
|
103
|
+
|
104
|
+
## See also the ecosystem
|
105
|
+
|
106
|
+
``brainstate`` is one part of our brain simulation ecosystem: https://brainmodeling.readthedocs.io/
|
107
|
+
|
108
|
+
|
@@ -0,0 +1,111 @@
|
|
1
|
+
brainstate/__init__.py,sha256=XZZctdLIf_6-IJIbeTyoEHR6r3pyKF2n6MfikCffASA,5915
|
2
|
+
brainstate/_compatible_import.py,sha256=7thV_2F0FD5AF2DETjBfmtNb_2ZQzki8NxFgC62frg0,11037
|
3
|
+
brainstate/_compatible_import_test.py,sha256=6ka26Sa_Kk6F-Ar1HR6UaKJTHquXcUCWglgXBUOovcg,22762
|
4
|
+
brainstate/_deprecation.py,sha256=gSh36_TWLBgQAo0gNfOzscV9ssa26k3te9y25BG6O2w,8381
|
5
|
+
brainstate/_deprecation_test.py,sha256=vnmcZ7l_FmQRgsMYUA9wMiWL1ltPc01plUTK-pWKCzk,87550
|
6
|
+
brainstate/_error.py,sha256=6A5ILy17ZMMZIjS8LkajTZBDRnwv_Qait5x__h2Levo,1522
|
7
|
+
brainstate/_state.py,sha256=G5SCjxi42Lb1EoqaIZcyy0fmjNonwrQLLb4ujKt_99E,77867
|
8
|
+
brainstate/_state_test.py,sha256=ptul4-BHagOIhnAvnWmb2BFyXnIfWpK0rQClLxhzLBY,39895
|
9
|
+
brainstate/_utils.py,sha256=cmUyO9ds1etrrpV4ucp1G8mDqE15g4ZtbivblH_cD9o,1613
|
10
|
+
brainstate/environ.py,sha256=BmQsvo1aZaMpckHXlJ45dZh9DUdnHHP9Q9JdLNCa9wA,42169
|
11
|
+
brainstate/environ_test.py,sha256=RdVmeP7irbk3_qNjwWoy-DSdpkTRpxqAABtNGZcbB2w,42418
|
12
|
+
brainstate/mixin.py,sha256=QoZPhBs4hGXTzTfXaF0M8XhsbcbOtT4Ylm54D7pCOSA,45124
|
13
|
+
brainstate/mixin_test.py,sha256=6WmqJf34kT3Z5WaiCNDo3OV3ci0DIHCQp474zECDxEU,34718
|
14
|
+
brainstate/typing.py,sha256=pYiNI-9oHpH7HfjRKYxugK03KGiamCwweagMyO0rsi4,26301
|
15
|
+
brainstate/typing_test.py,sha256=2mmMW0uAzIo3_VXpT5Boq79BohxYkzBlHexBysFUGII,26240
|
16
|
+
brainstate/graph/__init__.py,sha256=kGVtHAnkiWR5MqDYQU0G3AobWnioGeDqjILA--RyDz8,846
|
17
|
+
brainstate/graph/_node.py,sha256=_XH8xx6_glsCK4KCsQnarACK8meyhCdfh3nWfUDko0k,6407
|
18
|
+
brainstate/graph/_node_test.py,sha256=sD2DS0AhDKOU5ZQm0cYz0llnJ6D60ftNfDpztk4i8cM,18687
|
19
|
+
brainstate/graph/_operation.py,sha256=5EN_qL08M_BIV-L3YkVP3OwlsDuNKmVV5NaCLvrl7kw,53971
|
20
|
+
brainstate/graph/_operation_test.py,sha256=IVyrJh4io3sDgtrTEIAItGzNs2XEf7rO1rvI1r_KiII,39119
|
21
|
+
brainstate/nn/__init__.py,sha256=JcaVVt01LNbAwxHymNmBUFAZwchtucyHq_sT70rvYc4,5245
|
22
|
+
brainstate/nn/_activations.py,sha256=6jHR67obYR1lpo-imVXmfd3m_NDyU0XZb8t-pVYDvUU,26917
|
23
|
+
brainstate/nn/_activations_test.py,sha256=Ikr8RYBaIpApVKUhY-XAWr6llEG7vYWS8YuqDHyTtBY,13438
|
24
|
+
brainstate/nn/_collective_ops.py,sha256=AHda68XjoNUyLOwxRhwQLnXr4zLBqnB8UDLAkfx2eAg,21282
|
25
|
+
brainstate/nn/_collective_ops_test.py,sha256=8mKQkfTjfwuO7DA1i_Yr4QD1yg2ZJgVpX7zhHFo9CuQ,25600
|
26
|
+
brainstate/nn/_common.py,sha256=UyJMJoVF9KfrToOX5Dbv-2s3CD49SsroLRfL17DLl4Q,7184
|
27
|
+
brainstate/nn/_common_test.py,sha256=xztvEPYCurvY9LiaZOtHjWcFrwJaYknA8_RLy_XHI1Y,4895
|
28
|
+
brainstate/nn/_conv.py,sha256=3cGToc5UoGN5jp4BUHlrK6md8O_0IcWMCBXFHBQg7nE,82106
|
29
|
+
brainstate/nn/_conv_test.py,sha256=65FlrteUxLQb8ckUUJFhaPZDsQZTlGVAp0HUGUJtt1M,30173
|
30
|
+
brainstate/nn/_delay.py,sha256=qssPPKgWeS-vcxKJG6ewn2bQ1KvdHHBrbI6bIJytSfI,22332
|
31
|
+
brainstate/nn/_delay_test.py,sha256=FzBb8vXfse8HEcEid83hpa6aag6oj90mtcHYsDs0DOE,10376
|
32
|
+
brainstate/nn/_dropout.py,sha256=UotjW0PQO4gypfhtSqzkR4UVGkY0kNBncEB_poGL7Sc,22555
|
33
|
+
brainstate/nn/_dropout_test.py,sha256=87ETtMZ7Aeaa0les7Z5ICO9KVKMewm_WWd-ITCFDo_w,21589
|
34
|
+
brainstate/nn/_dynamics.py,sha256=H83WUFWv6ZLBEn01U30qInjgfh8WAgZ5jwgXWA9TEkg,29161
|
35
|
+
brainstate/nn/_dynamics_test.py,sha256=XwTBCXXKN8hAaLFtQjmWEOf-ry0MJZmfgx9dAju-rXA,1744
|
36
|
+
brainstate/nn/_elementwise.py,sha256=4kKzrbKn5luwnpY8n7IeaMOtBVVie8oPHEc64hSn8-w,34858
|
37
|
+
brainstate/nn/_elementwise_test.py,sha256=sbWlUyTB8oiu3PRHObTvmUaob99EjIGYE6k6bEpg6K8,27296
|
38
|
+
brainstate/nn/_embedding.py,sha256=MXkGja6uCB2OsbwqArEQ8ez_GgwZ2s3DINaUnFyeZWk,14991
|
39
|
+
brainstate/nn/_embedding_test.py,sha256=Gc0y6gHMEagaDrBJoAYQZMDTd47TYQNVrenWUwLWK_w,6242
|
40
|
+
brainstate/nn/_event_fixedprob.py,sha256=ZEnIyjDksxtUWWG5GXLcF-RHR1S33DZTw-rN2lHKs0g,9395
|
41
|
+
brainstate/nn/_event_fixedprob_test.py,sha256=rvTKxEzKwvctQc8-AxXjJ4p4D-if1va2O8KqQYW5nxY,3836
|
42
|
+
brainstate/nn/_event_linear.py,sha256=d0J54Sf9zBl926BzXoy4Oc1p96h9veU3f8YhWrZLRPk,2554
|
43
|
+
brainstate/nn/_event_linear_test.py,sha256=qzcGplDIwxTnZOs4JzD5GX_oNBYtcYFNaf3GpWo8pZY,3765
|
44
|
+
brainstate/nn/_exp_euler.py,sha256=deKqWu5RvN_Jvj_QHg5j1xkKhlysBFhi3Qvo_Z-QNDo,8698
|
45
|
+
brainstate/nn/_exp_euler_test.py,sha256=21qomGOo96YLmlEQok2hCByAcRpmqREhSr3kxmhKOm8,13014
|
46
|
+
brainstate/nn/_linear.py,sha256=LLWBE6eBXjpXUvx5mjdwOjFnolNRGBxNb_ZbhsYOu4M,24201
|
47
|
+
brainstate/nn/_linear_test.py,sha256=5fHx4v4_54dH3Bsyapl2cobvVtMEu3DFR-jDKLbkJFw,17876
|
48
|
+
brainstate/nn/_metrics.py,sha256=TgALwv6i9La4Dm1WAkWDWxvxr9rkd7CJLGFy2sOGQbQ,36481
|
49
|
+
brainstate/nn/_metrics_test.py,sha256=XZiRndchRgEH0X8zsHzg0fsHNMxj43mnd883QChfSik,24104
|
50
|
+
brainstate/nn/_module.py,sha256=W9iXXmLxaX_QyNxukTxajHJfah7s9GqJXZRHQhZqCfU,13076
|
51
|
+
brainstate/nn/_module_test.py,sha256=znjB7FU5evJENQ1Pqw7ZlOGC5faQe4-4VpjW60H8UWI,1414
|
52
|
+
brainstate/nn/_normalizations.py,sha256=3ivJdQUhhWeMhDpzWhXhRQnu8dRR4zDzM1cOlPvPo2s,50374
|
53
|
+
brainstate/nn/_normalizations_test.py,sha256=y5n7aTaUHRkyAAjq4Oj8dfButJC3ehm4KdUB7226Bow,23350
|
54
|
+
brainstate/nn/_paddings.py,sha256=3u3dbRFtPSlIsLMBYZmHcYDQ6HFl0u_d-yP7ZoYcCrA,32415
|
55
|
+
brainstate/nn/_paddings_test.py,sha256=uY9CRexf9sM5V6AzTfzm8214y4IoGqEbahy92yaKsWM,27409
|
56
|
+
brainstate/nn/_poolings.py,sha256=aiLDTgtbDEse2OZlJONdU1CCsBjZtsWqsnL8ffTSGpk,86045
|
57
|
+
brainstate/nn/_poolings_test.py,sha256=cZ4lyutUY-Iti6rkGWjqWZ72fXF5QvwX9mki4CQ1xDs,34164
|
58
|
+
brainstate/nn/_rnns.py,sha256=FmuVsUwQCl7QbePmYokd-ZxKwVgDqa8FdCto_dNFDjk,32947
|
59
|
+
brainstate/nn/_rnns_test.py,sha256=gqPGz64i_QUEXi2fUBrIaBEPZjTs8hm8S6HyPbs3I6M,22112
|
60
|
+
brainstate/nn/_utils.py,sha256=VK-Se53e1q-Ip4AtMOZ3SUzYw8u2UllLJRLRtEFRCRE,7403
|
61
|
+
brainstate/nn/_utils_test.py,sha256=uim2SkfNHrBZzNDvN0WOK8qeZC1kaeOd-UQDvrn_M24,14266
|
62
|
+
brainstate/nn/init.py,sha256=7iLHrL-ZHpU-g5d0PlusaqmtkWO7X_KNr2eLx90oHrc,25656
|
63
|
+
brainstate/nn/init_test.py,sha256=bfby6kovvhbc7CCEaohtQawUQj3w3GvaMSDAdTiP2ps,6200
|
64
|
+
brainstate/random/__init__.py,sha256=yeWQ3RUcFXtXcDkFhaOYa_nwQ_M7hlH_W69YEcGF3Oo,8351
|
65
|
+
brainstate/random/_fun.py,sha256=fW2bc2i15sQNodmKT66ZCfXJSIZ8Ygxmn_Xf80Zjygw,135334
|
66
|
+
brainstate/random/_fun_test.py,sha256=EcHOcCOsmZPbm1n5TUyiTDxS13Pn3MJbq5IG-twbnPI,22866
|
67
|
+
brainstate/random/_impl.py,sha256=A79IK8YNZuN8RGLftASKHcZI0Cdsep7mGYNtCfKl2fQ,21830
|
68
|
+
brainstate/random/_seed.py,sha256=mLHqOu-lJQjsXo4nODACY78SSKgYayHx3n5IBZ9L6J0,24922
|
69
|
+
brainstate/random/_seed_test.py,sha256=Y2VCAkUzciDaCfYZWPe_Ewmi3MylK-WzfPA7TzorV8Q,1491
|
70
|
+
brainstate/random/_state.py,sha256=JPK6-jqFwrzsrb3lrxZ4GahP0TtXycUfWrJSdjHIEg8,42946
|
71
|
+
brainstate/random/_state_test.py,sha256=OfW0WxTpJZm_kT_7bZjJJ8ZtLM_X8NBF90h1sDSZxmw,19221
|
72
|
+
brainstate/transform/__init__.py,sha256=P7MAmt4pJYcpxLO30gFoT4BO_AHHM3VNOoG2j3OkNRk,2126
|
73
|
+
brainstate/transform/_ad_checkpoint.py,sha256=4dcNCEQVV_CPMSkE32URERDMpQHbyfdGeLT_Nvhyd4o,6912
|
74
|
+
brainstate/transform/_ad_checkpoint_test.py,sha256=fPXBjDxsLHbL2mhIU3x_F5BpitkXLpgIsRCRgm2Us6w,1697
|
75
|
+
brainstate/transform/_autograd.py,sha256=4zGSYa9TMn6bqzPJNLfU9UZGZRyYxmwMXTKZWO4w3QQ,39991
|
76
|
+
brainstate/transform/_autograd_test.py,sha256=saWG1_k3cRXpsyQDzQkOLGvsF7IIxG9aGnjrf5B3HNk,44112
|
77
|
+
brainstate/transform/_conditions.py,sha256=nLc_m0bLlybEPTi42feSlz2zTDIHB-BknSqPTIL_I0w,11376
|
78
|
+
brainstate/transform/_conditions_test.py,sha256=MEuqRq6IFmyORRDi0qWvNo4pWKFyc8aNrW1v9Saqxj0,8493
|
79
|
+
brainstate/transform/_error_if.py,sha256=e9tp3wT5p4bEyjn_Za_SrPNOG3OIoPBMIrvG2CsZzvw,2680
|
80
|
+
brainstate/transform/_error_if_test.py,sha256=yn-qcZ6lZUWciIif4fJOpdpKzJFAAdfgzZm6FfPeq7U,1848
|
81
|
+
brainstate/transform/_find_state.py,sha256=nUrVp_DUP78E2H8UHyhH8RL03kwcKjepnkDTqbISFmI,7405
|
82
|
+
brainstate/transform/_find_state_test.py,sha256=KahI4HSj6MTlC4ccCPyLiyERy3MryZd_iczGwFXb2bM,2913
|
83
|
+
brainstate/transform/_jit.py,sha256=qYsL3Z9nAAW0UyQe_AyvBEuJvqAan6iw3lN49o1oC0A,15421
|
84
|
+
brainstate/transform/_jit_test.py,sha256=ecw54dGQYdJq2J94itPrXBQrSDNvCY_htUD7z7y4HUM,4013
|
85
|
+
brainstate/transform/_loop_collect_return.py,sha256=HhjC2gq6qzliw4ofP16VxdtR5hW-NmDZdeHxuiLdYGk,25899
|
86
|
+
brainstate/transform/_loop_collect_return_test.py,sha256=BVK-b3CuDtTXciRaA_8t4751N4taQOnIPNzAelSts-k,1753
|
87
|
+
brainstate/transform/_loop_no_collection.py,sha256=ArPpNemMh4jJsq_vUWPxuagCnxTlONN--3P_-44qYq8,10156
|
88
|
+
brainstate/transform/_loop_no_collection_test.py,sha256=3bRo9_Oaypbw3asEevrgTK0WksxDAIZKJgeaWpt7nl8,1371
|
89
|
+
brainstate/transform/_make_jaxpr.py,sha256=_pSw1oumQ504Emvcgy7eXBTxOw1c1pbnvbcdveKMG8s,80444
|
90
|
+
brainstate/transform/_make_jaxpr_test.py,sha256=EfY-ZKR-u-aLfjOTPY6QiwEzZU-QpIUsB5oCCZi5Z-I,53014
|
91
|
+
brainstate/transform/_mapping.py,sha256=wBTvmXSg8TW-ZjIhKU5nQLLb6BIkLTCaJWwESZ72Hks,22021
|
92
|
+
brainstate/transform/_mapping_test.py,sha256=BkoL9peJGiJvkp7aLErrB6GnDasSviMbBkBqh1yD8LM,3255
|
93
|
+
brainstate/transform/_progress_bar.py,sha256=kZ-mI5hbUQXhqKFVyo0qeKG_LvrR9ZIar7WkXyOeET0,8961
|
94
|
+
brainstate/transform/_unvmap.py,sha256=cW6fjs5Iy1YBB6Nx2mxlM4IzV8U99bEX5QjT8rBRDho,6319
|
95
|
+
brainstate/transform/_util.py,sha256=IYqJj7oyAYzm_m3d9WEsUQRKDdVLQaAWpwM5O8PD4YQ,11304
|
96
|
+
brainstate/util/__init__.py,sha256=anHdG5BIsMqcBQy7gt8lErKInZv1wf2NOLJGUqltyAQ,1154
|
97
|
+
brainstate/util/_others.py,sha256=HSP3ynNv2ocxPz5omQl2rxMTOjGkQNNprsTOCEANCjA,30837
|
98
|
+
brainstate/util/_others_test.py,sha256=gEiUybMxtv11tsD0jge9Y0K8idkWkHVBlTZz1gKXNAU,30535
|
99
|
+
brainstate/util/_pretty_pytree.py,sha256=0fVwW8qHtKrpJU4q2tbn8usSKOlG8lBALn6E6crpf64,46815
|
100
|
+
brainstate/util/_pretty_pytree_test.py,sha256=6DiYX_Rwyvv3XcuQHsU-PuWXmvOqSmiVVN4diVvp76o,22362
|
101
|
+
brainstate/util/_pretty_repr.py,sha256=fafG6SIFoPjaWmQcTHwmnEqbVvcTCoN_lKiBKBk1QTQ,13958
|
102
|
+
brainstate/util/_pretty_repr_test.py,sha256=yfaANMfozlM5E3IZv-LBaNpyvSVB8ftRp0r7_oAr5yA,23122
|
103
|
+
brainstate/util/filter.py,sha256=wY_XUF3OhrXSV1bZkTcVhlEPba4HP1l9N5aRW2zgxqQ,27410
|
104
|
+
brainstate/util/filter_test.py,sha256=ZfrEeOc1yMHYzrcSR3p4jbZGj7c_tXC8VcPq7H13q8E,31653
|
105
|
+
brainstate/util/struct.py,sha256=LYPLGDGfPuw14hhx5k4rb8msSH3yZPdAu_0CvjxPWwE,24505
|
106
|
+
brainstate/util/struct_test.py,sha256=q_fWsUH1ON35DKjUUAMq6VtYglqTDyBvd6WMVGD89EI,16526
|
107
|
+
brainstate-0.2.2.dist-info/licenses/LICENSE,sha256=RJ40fox7u2in2H8wvIS5DsPGlNHaA7JI024thFUlaZE,11348
|
108
|
+
brainstate-0.2.2.dist-info/METADATA,sha256=tbhbVxomU2orW-T3WaEjVtYAyZSn4T4uJxAXzPmU9JY,4421
|
109
|
+
brainstate-0.2.2.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
|
110
|
+
brainstate-0.2.2.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
|
111
|
+
brainstate-0.2.2.dist-info/RECORD,,
|