brainstate 0.1.10__py2.py3-none-any.whl → 0.2.1__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 +169 -58
- brainstate/_compatible_import.py +340 -148
- brainstate/_compatible_import_test.py +681 -0
- brainstate/_deprecation.py +210 -0
- brainstate/_deprecation_test.py +2319 -0
- brainstate/{util/error.py → _error.py} +45 -55
- brainstate/_state.py +1652 -1605
- brainstate/_state_test.py +52 -52
- brainstate/_utils.py +47 -47
- brainstate/environ.py +1495 -563
- brainstate/environ_test.py +1223 -62
- brainstate/graph/__init__.py +22 -29
- brainstate/graph/_node.py +240 -0
- brainstate/graph/_node_test.py +589 -0
- brainstate/graph/{_graph_operation.py → _operation.py} +1624 -1738
- brainstate/graph/_operation_test.py +1147 -0
- brainstate/mixin.py +1433 -365
- brainstate/mixin_test.py +1017 -77
- brainstate/nn/__init__.py +137 -135
- brainstate/nn/_activations.py +1100 -808
- brainstate/nn/_activations_test.py +354 -331
- brainstate/nn/_collective_ops.py +633 -514
- brainstate/nn/_collective_ops_test.py +774 -43
- brainstate/nn/_common.py +226 -178
- brainstate/nn/_common_test.py +154 -0
- brainstate/nn/_conv.py +2010 -501
- brainstate/nn/_conv_test.py +849 -238
- brainstate/nn/_delay.py +575 -588
- brainstate/nn/_delay_test.py +243 -238
- brainstate/nn/_dropout.py +618 -426
- brainstate/nn/_dropout_test.py +477 -100
- brainstate/nn/_dynamics.py +1267 -1343
- brainstate/nn/_dynamics_test.py +67 -78
- brainstate/nn/_elementwise.py +1298 -1119
- brainstate/nn/_elementwise_test.py +830 -169
- brainstate/nn/_embedding.py +408 -58
- brainstate/nn/_embedding_test.py +156 -0
- brainstate/nn/{_fixedprob.py → _event_fixedprob.py} +233 -239
- brainstate/nn/{_fixedprob_test.py → _event_fixedprob_test.py} +115 -114
- brainstate/nn/{_linear_mv.py → _event_linear.py} +83 -83
- brainstate/nn/{_linear_mv_test.py → _event_linear_test.py} +121 -120
- brainstate/nn/_exp_euler.py +254 -92
- brainstate/nn/_exp_euler_test.py +377 -35
- brainstate/nn/_linear.py +744 -424
- brainstate/nn/_linear_test.py +475 -107
- brainstate/nn/_metrics.py +1070 -0
- brainstate/nn/_metrics_test.py +611 -0
- brainstate/nn/_module.py +384 -377
- brainstate/nn/_module_test.py +40 -40
- brainstate/nn/_normalizations.py +1334 -975
- brainstate/nn/_normalizations_test.py +699 -73
- brainstate/nn/_paddings.py +1020 -0
- brainstate/nn/_paddings_test.py +723 -0
- brainstate/nn/_poolings.py +2239 -1177
- brainstate/nn/_poolings_test.py +953 -217
- brainstate/nn/{_rate_rnns.py → _rnns.py} +946 -554
- brainstate/nn/_rnns_test.py +593 -0
- brainstate/nn/_utils.py +216 -89
- brainstate/nn/_utils_test.py +402 -0
- brainstate/{init/_random_inits.py → nn/init.py} +809 -553
- brainstate/{init/_random_inits_test.py → nn/init_test.py} +180 -149
- brainstate/random/__init__.py +270 -24
- brainstate/random/_rand_funs.py +3938 -3616
- brainstate/random/_rand_funs_test.py +640 -567
- brainstate/random/_rand_seed.py +675 -210
- brainstate/random/_rand_seed_test.py +48 -48
- brainstate/random/_rand_state.py +1617 -1409
- brainstate/random/_rand_state_test.py +551 -0
- brainstate/transform/__init__.py +59 -0
- brainstate/transform/_ad_checkpoint.py +176 -0
- brainstate/{compile → transform}/_ad_checkpoint_test.py +49 -49
- brainstate/{augment → transform}/_autograd.py +1025 -778
- brainstate/{augment → transform}/_autograd_test.py +1289 -1289
- brainstate/transform/_conditions.py +316 -0
- brainstate/{compile → transform}/_conditions_test.py +220 -220
- brainstate/{compile → transform}/_error_if.py +94 -92
- brainstate/{compile → transform}/_error_if_test.py +52 -52
- brainstate/transform/_eval_shape.py +145 -0
- brainstate/{augment → transform}/_eval_shape_test.py +38 -38
- brainstate/{compile → transform}/_jit.py +399 -346
- brainstate/{compile → transform}/_jit_test.py +143 -143
- brainstate/{compile → transform}/_loop_collect_return.py +675 -536
- brainstate/{compile → transform}/_loop_collect_return_test.py +58 -58
- brainstate/{compile → transform}/_loop_no_collection.py +283 -184
- brainstate/{compile → transform}/_loop_no_collection_test.py +50 -50
- brainstate/transform/_make_jaxpr.py +2016 -0
- brainstate/transform/_make_jaxpr_test.py +1510 -0
- brainstate/transform/_mapping.py +529 -0
- brainstate/transform/_mapping_test.py +194 -0
- brainstate/{compile → transform}/_progress_bar.py +255 -202
- brainstate/{augment → transform}/_random.py +171 -151
- brainstate/{compile → transform}/_unvmap.py +256 -159
- brainstate/transform/_util.py +286 -0
- brainstate/typing.py +837 -304
- brainstate/typing_test.py +780 -0
- brainstate/util/__init__.py +27 -50
- brainstate/util/_others.py +1025 -0
- brainstate/util/_others_test.py +962 -0
- brainstate/util/_pretty_pytree.py +1301 -0
- brainstate/util/_pretty_pytree_test.py +675 -0
- brainstate/util/{pretty_repr.py → _pretty_repr.py} +462 -328
- brainstate/util/_pretty_repr_test.py +696 -0
- brainstate/util/filter.py +945 -469
- brainstate/util/filter_test.py +912 -0
- brainstate/util/struct.py +910 -523
- brainstate/util/struct_test.py +602 -0
- {brainstate-0.1.10.dist-info → brainstate-0.2.1.dist-info}/METADATA +108 -91
- brainstate-0.2.1.dist-info/RECORD +111 -0
- {brainstate-0.1.10.dist-info → brainstate-0.2.1.dist-info}/licenses/LICENSE +202 -202
- brainstate/augment/__init__.py +0 -30
- brainstate/augment/_eval_shape.py +0 -99
- brainstate/augment/_mapping.py +0 -1060
- brainstate/augment/_mapping_test.py +0 -597
- brainstate/compile/__init__.py +0 -38
- brainstate/compile/_ad_checkpoint.py +0 -204
- brainstate/compile/_conditions.py +0 -256
- brainstate/compile/_make_jaxpr.py +0 -888
- brainstate/compile/_make_jaxpr_test.py +0 -156
- brainstate/compile/_util.py +0 -147
- brainstate/functional/__init__.py +0 -27
- brainstate/graph/_graph_node.py +0 -244
- brainstate/graph/_graph_node_test.py +0 -73
- brainstate/graph/_graph_operation_test.py +0 -563
- brainstate/init/__init__.py +0 -26
- brainstate/init/_base.py +0 -52
- brainstate/init/_generic.py +0 -244
- brainstate/init/_regular_inits.py +0 -105
- brainstate/init/_regular_inits_test.py +0 -50
- brainstate/nn/_inputs.py +0 -608
- brainstate/nn/_ltp.py +0 -28
- brainstate/nn/_neuron.py +0 -705
- brainstate/nn/_neuron_test.py +0 -161
- brainstate/nn/_others.py +0 -46
- brainstate/nn/_projection.py +0 -486
- brainstate/nn/_rate_rnns_test.py +0 -63
- brainstate/nn/_readout.py +0 -209
- brainstate/nn/_readout_test.py +0 -53
- brainstate/nn/_stp.py +0 -236
- brainstate/nn/_synapse.py +0 -505
- brainstate/nn/_synapse_test.py +0 -131
- brainstate/nn/_synaptic_projection.py +0 -423
- brainstate/nn/_synouts.py +0 -162
- brainstate/nn/_synouts_test.py +0 -57
- brainstate/nn/metrics.py +0 -388
- brainstate/optim/__init__.py +0 -38
- brainstate/optim/_base.py +0 -64
- brainstate/optim/_lr_scheduler.py +0 -448
- brainstate/optim/_lr_scheduler_test.py +0 -50
- brainstate/optim/_optax_optimizer.py +0 -152
- brainstate/optim/_optax_optimizer_test.py +0 -53
- brainstate/optim/_sgd_optimizer.py +0 -1104
- brainstate/random/_random_for_unit.py +0 -52
- brainstate/surrogate.py +0 -1957
- brainstate/transform.py +0 -23
- brainstate/util/caller.py +0 -98
- brainstate/util/others.py +0 -540
- brainstate/util/pretty_pytree.py +0 -945
- brainstate/util/pretty_pytree_test.py +0 -159
- brainstate/util/pretty_table.py +0 -2954
- brainstate/util/scaling.py +0 -258
- brainstate-0.1.10.dist-info/RECORD +0 -130
- {brainstate-0.1.10.dist-info → brainstate-0.2.1.dist-info}/WHEEL +0 -0
- {brainstate-0.1.10.dist-info → brainstate-0.2.1.dist-info}/top_level.txt +0 -0
@@ -1,91 +1,108 @@
|
|
1
|
-
Metadata-Version: 2.4
|
2
|
-
Name: brainstate
|
3
|
-
Version: 0.1
|
4
|
-
Summary: A
|
5
|
-
|
6
|
-
|
7
|
-
|
8
|
-
|
9
|
-
Project-URL:
|
10
|
-
Project-URL:
|
11
|
-
|
12
|
-
|
13
|
-
Classifier:
|
14
|
-
Classifier:
|
15
|
-
Classifier:
|
16
|
-
Classifier: Intended Audience ::
|
17
|
-
Classifier:
|
18
|
-
Classifier: Programming Language :: Python :: 3
|
19
|
-
Classifier: Programming Language :: Python :: 3.
|
20
|
-
Classifier: Programming Language :: Python :: 3.
|
21
|
-
Classifier: Programming Language :: Python :: 3.
|
22
|
-
Classifier:
|
23
|
-
Classifier:
|
24
|
-
Classifier:
|
25
|
-
Classifier: Topic :: Scientific/Engineering ::
|
26
|
-
Classifier: Topic :: Scientific/Engineering ::
|
27
|
-
Classifier: Topic ::
|
28
|
-
|
29
|
-
|
30
|
-
|
31
|
-
|
32
|
-
Requires-Dist:
|
33
|
-
Requires-Dist:
|
34
|
-
Requires-Dist: brainunit
|
35
|
-
Requires-Dist: brainevent
|
36
|
-
Provides-Extra:
|
37
|
-
Requires-Dist:
|
38
|
-
|
39
|
-
|
40
|
-
|
41
|
-
|
42
|
-
|
43
|
-
|
44
|
-
|
45
|
-
|
46
|
-
|
47
|
-
|
48
|
-
|
49
|
-
|
50
|
-
|
51
|
-
|
52
|
-
|
53
|
-
|
54
|
-
|
55
|
-
|
56
|
-
|
57
|
-
|
58
|
-
|
59
|
-
|
60
|
-
|
61
|
-
|
62
|
-
|
63
|
-
|
64
|
-
|
65
|
-
|
66
|
-
|
67
|
-
|
68
|
-
|
69
|
-
|
70
|
-
|
71
|
-
|
72
|
-
|
73
|
-
|
74
|
-
|
75
|
-
|
76
|
-
|
77
|
-
|
78
|
-
|
79
|
-
|
80
|
-
|
81
|
-
|
82
|
-
|
83
|
-
|
84
|
-
|
85
|
-
|
86
|
-
|
87
|
-
|
88
|
-
|
89
|
-
|
90
|
-
|
91
|
-
|
1
|
+
Metadata-Version: 2.4
|
2
|
+
Name: brainstate
|
3
|
+
Version: 0.2.1
|
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=QXLW5UOeT0Q25j6AyLkejFrGkBAQWA2I9O7TVycjV_w,6193
|
2
|
+
brainstate/_compatible_import.py,sha256=JUAHEeZBFTN6KW9ZZoHW5M1KtxQA-XyJJB6v_f83ndM,11377
|
3
|
+
brainstate/_compatible_import_test.py,sha256=ynjx_ErAzE7jZJkJe-mnBjVVuBSURiCV3HzZ-RYAT1E,23443
|
4
|
+
brainstate/_deprecation.py,sha256=sg5xBy15QQd7wo9Ew7QH89vH6ZSs96ayCi5cEdyrpqE,8591
|
5
|
+
brainstate/_deprecation_test.py,sha256=5KPrCC8o5xqLg9BhmvcLn-UT97YPLSsSaXhla2jVQQA,90644
|
6
|
+
brainstate/_error.py,sha256=d7aaHVHLfh8aynhduvBmZSB8aK69aGHGm17D55xhTEY,1567
|
7
|
+
brainstate/_state.py,sha256=ngz227ueaFV1E48LvlHrzjz7cRPKXntPyjYCSlQN2T8,59748
|
8
|
+
brainstate/_state_test.py,sha256=LzfGfM2efKMjlX5EpIwFsi_tbBWM9HQkCPzeiSZ7PMg,1641
|
9
|
+
brainstate/_utils.py,sha256=jgxW7Ouj_r6nKcG7UHjX4UYtyOX4QOMD5Q4CYrnswtU,1660
|
10
|
+
brainstate/environ.py,sha256=WxCkG_9-RORunpVHU2zJebyxShf_XotD2HHzatFeUos,43664
|
11
|
+
brainstate/environ_test.py,sha256=vgEcFtZXLYBnVwOPGLQJehz1qODcJc_C2esLL9ORD2M,43641
|
12
|
+
brainstate/mixin.py,sha256=Iv149vfepDmYYFp8_wBVXc41JwP6WvPpTIG1p70N7Uk,46068
|
13
|
+
brainstate/mixin_test.py,sha256=Vi9p3BVU6wrNVsJvQfooS3riJTml8tYNq1UM-wO4N8Q,35735
|
14
|
+
brainstate/typing.py,sha256=cKFeLRa8Aowjq_-eWXkt1xO9cWX6zvyVn_IWDcm9fTA,27138
|
15
|
+
brainstate/typing_test.py,sha256=nJc325IwKrB1DaxJX1fq4VIdLpmvT7x1vYtNcGWnZmA,27020
|
16
|
+
brainstate/graph/__init__.py,sha256=KZOn3B0vuzYl55XQm6N5pCNnDTyHdrEggfp_h_-Ml7Q,868
|
17
|
+
brainstate/graph/_node.py,sha256=ybMR1I-5ztLeNJ9GgxuiB-kgSSisYTp4xJ6XUt-qv40,6647
|
18
|
+
brainstate/graph/_node_test.py,sha256=BIUOuXRYLj0KVPC5Slb98wRcO158T63T53zHnPnhpZk,19276
|
19
|
+
brainstate/graph/_operation.py,sha256=UAQe2z73rOHX45PNFAC2X0ltkh9NVQ3daIk395Jxn70,55720
|
20
|
+
brainstate/graph/_operation_test.py,sha256=_Z7ZJLdOsZAjorvc4PCu38hDQ7-2DQnJsgmwgHdePrg,40266
|
21
|
+
brainstate/nn/__init__.py,sha256=3l3uT6UBQCta6MUMgIAyZIOtjhfi6KI6zT7gy2GiCX0,4896
|
22
|
+
brainstate/nn/_activations.py,sha256=6GriLZzgEFcv_13f0EiaG7r3xXWun7X749vriPYHdLM,28017
|
23
|
+
brainstate/nn/_activations_test.py,sha256=rjDx7Pn2gg5vk1RyMNfnK47kB2YDilY4J7ygmOEgutU,13792
|
24
|
+
brainstate/nn/_collective_ops.py,sha256=db988H8e5B8PR8_BQJvspeOoKl-pSdb-hP2ZBRTbbR8,21913
|
25
|
+
brainstate/nn/_collective_ops_test.py,sha256=82V4egN9GdQeph9T3BG4YH4I43qPmB98hgkBU--UAlk,26374
|
26
|
+
brainstate/nn/_common.py,sha256=-uBCsn0B5Z1LMwJI4K9FuK_8l7UiU88H2S84Nv6jveA,7410
|
27
|
+
brainstate/nn/_common_test.py,sha256=bqngm_1syVwozgDLhzkdmEaKtEExKghCo2Nqdq6R5MM,6120
|
28
|
+
brainstate/nn/_conv.py,sha256=lpX75tqVDnA_AJRV7Mp0OpX662wTUzdiLhTjfTKSONw,84116
|
29
|
+
brainstate/nn/_conv_test.py,sha256=srZDdlG4TiYQfhDwdQNOEyb_fH2FT79Loe8voZ48ods,31022
|
30
|
+
brainstate/nn/_delay.py,sha256=SLeH0Qv953tVJwZv5GhRIwIVX1WkasaN4dDTTNFqEkE,22906
|
31
|
+
brainstate/nn/_delay_test.py,sha256=RX1Ih0p2PzOOqEbVTePLITobKBN8AveUJ5shXGCfdgk,10619
|
32
|
+
brainstate/nn/_dropout.py,sha256=kWOMzv6peNYc4K5sczennI10-dEPCMXiDicAWEBcUdA,23173
|
33
|
+
brainstate/nn/_dropout_test.py,sha256=_Ym1iYz5XSjMfGk_YRIK7qZ7s2dFwJkP-8JIWizF1Jo,21938
|
34
|
+
brainstate/nn/_dynamics.py,sha256=1UO9k077gePxnxMyZMMdB-2N0LalvxaPIaqII3H5E1w,45377
|
35
|
+
brainstate/nn/_dynamics_test.py,sha256=Qz4sl9TpZygGzYxGnIbtYXNO8aJf-PHvWnZ_dByBYp4,2423
|
36
|
+
brainstate/nn/_elementwise.py,sha256=I8FW5oWagFZzI-LhRMxxUf334Ai5t8elAsaeZKx1Lcc,36156
|
37
|
+
brainstate/nn/_elementwise_test.py,sha256=2_lpkkJjAo-GxKNkxPo4sbSvsDbs0qbRMlHSXlFVVpk,28125
|
38
|
+
brainstate/nn/_embedding.py,sha256=rEahcqYYq1oWpKL-chxzQMIectQdbFu_weVEULynVMg,15350
|
39
|
+
brainstate/nn/_embedding_test.py,sha256=FnXhmYWtbzZvoyIFoJkOTsvP3hTOtfxB6LCHyaSbqQI,6398
|
40
|
+
brainstate/nn/_event_fixedprob.py,sha256=ngR9LKqxf-YDMtHAm14eHrORRFagc4ez4qcpQJFgpBE,9628
|
41
|
+
brainstate/nn/_event_fixedprob_test.py,sha256=XM-4XfqSkFQoSTChm6EIRThOHBPVRtdb-vJfoNp72PI,3951
|
42
|
+
brainstate/nn/_event_linear.py,sha256=LfccbsMwaYMDqcKQVREvTTmKtuVEqGBllMjeTDd-lNs,2637
|
43
|
+
brainstate/nn/_event_linear_test.py,sha256=g7-FqtJL32Me_WVA02XyoFZoJWsc_W5OX1sMqFt7zXM,3886
|
44
|
+
brainstate/nn/_exp_euler.py,sha256=-o1gBs2bPOraWu83v-cSBG86PFeqKu5EC3pZVAC0GMM,8889
|
45
|
+
brainstate/nn/_exp_euler_test.py,sha256=iLFgSDSF8sfke4GOmxZpzhH-OhwbiOiqbDa75yWAKH8,13391
|
46
|
+
brainstate/nn/_linear.py,sha256=m5hggIYbgLYPbbfpQYgBrb0g8scu5ue7wnu3VW1djGc,24791
|
47
|
+
brainstate/nn/_linear_test.py,sha256=YTVpbVP6ZzuwcheUuq9kHLK0JRQVIL9mvUg2mVkk12k,18351
|
48
|
+
brainstate/nn/_metrics.py,sha256=-jl7fVYGmhGS4VMQUuf3iLzMboRRj9AfprZ1EfBXAxQ,37551
|
49
|
+
brainstate/nn/_metrics_test.py,sha256=z5Z8BpHh6qTsS2qZkUupcZt5sRQ9QR1C-8E24SfBkr0,24715
|
50
|
+
brainstate/nn/_module.py,sha256=1e8ZMdq4Nl2im_ZpDDr2TZu9Xi-4sO4BJedBtXg9Aj8,13159
|
51
|
+
brainstate/nn/_module_test.py,sha256=XH1i8R15_uKkKkotfOAZK_2WNcXj-gLL6gmSgnFfHks,1454
|
52
|
+
brainstate/nn/_normalizations.py,sha256=udyWdudXg5ZlOxwzsILocRDoHq2tv_SwK0heTvqEAoM,51519
|
53
|
+
brainstate/nn/_normalizations_test.py,sha256=XPW5icjqiMyfC398-0TRelg9ZyFu1SD-wJJLVjSXPUU,24049
|
54
|
+
brainstate/nn/_paddings.py,sha256=i7cLWHwwx9y5EOkxeDilbxr84xW4U8ZwBeECCGcS14E,33435
|
55
|
+
brainstate/nn/_paddings_test.py,sha256=z0Tyz0rE4I1Y0C4wbZAqb3v8YUQl5Y8vBalzTm-YSlU,28131
|
56
|
+
brainstate/nn/_poolings.py,sha256=WbA8helFy-PbBYfUphVXDrXwcYymhxp9vbosU7PZpjU,88284
|
57
|
+
brainstate/nn/_poolings_test.py,sha256=bouz1sMq0xz5EmwdLuFFX2c26apSW1NW8dDxiWruFpc,35116
|
58
|
+
brainstate/nn/_rnns.py,sha256=iKsJT7KmNYsVQGN23JvnrwvhGBAzSpg1yfJ8P18nwBY,33893
|
59
|
+
brainstate/nn/_rnns_test.py,sha256=VGOtLh-C5iMUDQUppYWPDXyFCGy74sdberdeMIjUOP4,22704
|
60
|
+
brainstate/nn/_utils.py,sha256=1iRQEBCXwel0E72c7XPw_SQolSRk7BZsEm6_A2b4ZkQ,7619
|
61
|
+
brainstate/nn/_utils_test.py,sha256=fqxfbJM03N8K4qmjtoHiG6h8UlzwdDrd7YiEK13HWSA,14667
|
62
|
+
brainstate/nn/init.py,sha256=UenFiGcB7KDa0f67eabsHZHl38-AYM0eoblrXAxiB4I,26465
|
63
|
+
brainstate/nn/init_test.py,sha256=HkjWcNkLP9qVy4qLTazEaKgULsmBo-wyaBvHj77vbuI,6380
|
64
|
+
brainstate/random/__init__.py,sha256=K_JLJNW0H3RZKW1G3sGnlZ9pK9l6ANx5DOxw5P0RlyY,8653
|
65
|
+
brainstate/random/_rand_funs.py,sha256=ac2rA2hWNfZX6QznG8endpRcJv7umiq61b36kbQx0Xg,139277
|
66
|
+
brainstate/random/_rand_funs_test.py,sha256=QETjwkq_PlVE2mOptQeNCNbfZ4pQ_8MsDVJLFW9N5GA,23680
|
67
|
+
brainstate/random/_rand_seed.py,sha256=LBi4rCtR5iQ-GilU-3E3eHPc8d1rlHA8147Fa3I6R60,25602
|
68
|
+
brainstate/random/_rand_seed_test.py,sha256=BIQdvb9TsfIHs4UF4N9qvU3wJKdnkqnDK5CkWYCo3QA,1539
|
69
|
+
brainstate/random/_rand_state.py,sha256=NbnihT3BupYQOmiU9njHXN2HSKjugAa2xnxDI_0rj7k,55208
|
70
|
+
brainstate/random/_rand_state_test.py,sha256=XB1xq3WXAFLSGmxo-vBJulw9WoeBnJq5AFuL5Xty5SQ,19778
|
71
|
+
brainstate/transform/__init__.py,sha256=CjHzuRQHh0EDYIxugZ0T819X1P2IXIC-89Q_gQ9izCw,2300
|
72
|
+
brainstate/transform/_ad_checkpoint.py,sha256=248gdnHJHoa0HMIyqC2MiS5dNVC3xVI5ViCxskneTtc,7088
|
73
|
+
brainstate/transform/_ad_checkpoint_test.py,sha256=KLfpBozd6Doc3ck6an3VGunViKpzHEc2f-3gcT189C4,1746
|
74
|
+
brainstate/transform/_autograd.py,sha256=rC_1zUVLgZMDTA3EAYpZYsifcMl3P2tgWf-2VDsFuSk,41016
|
75
|
+
brainstate/transform/_autograd_test.py,sha256=jXX_w-iXjOrETKvSvRau6aTP98SOGEaOpqMcqPAEVGU,45401
|
76
|
+
brainstate/transform/_conditions.py,sha256=eMIPgg5N_Olq-iaek4ELkFGG0zBkThNJC9aZLC5sdIk,11715
|
77
|
+
brainstate/transform/_conditions_test.py,sha256=IpAzs8W5zmXcC_wYFwI4WFoNRbR9VmGdHUg8zw-S8BA,8713
|
78
|
+
brainstate/transform/_error_if.py,sha256=Pe5h2sMeDmZqfuhBsmiaNQrFgbj0m6LLP2y8VYso5PY,2774
|
79
|
+
brainstate/transform/_error_if_test.py,sha256=H4SUYIPwAVA9xMZsBYr_Kj79sndQzT26ypRck4iG7qk,1900
|
80
|
+
brainstate/transform/_eval_shape.py,sha256=g00ue2Ijo3-jTvMjHzhJql0DR_u337nLKsDaLAov8R0,5500
|
81
|
+
brainstate/transform/_eval_shape_test.py,sha256=BEmtqyudsKEz0Mg7SdXSm55dalL5W6ZXwhvs4oUJZEY,1393
|
82
|
+
brainstate/transform/_jit.py,sha256=msAGXVHIBx5h0jdnfYXhDDYIqCDJBQuKe-iR56ELCd4,15820
|
83
|
+
brainstate/transform/_jit_test.py,sha256=g97Kc5eYfe05opPTlr0N1JXetM6WilbwMfbH1t4xBjo,4156
|
84
|
+
brainstate/transform/_loop_collect_return.py,sha256=UtAlwjD2NAD7Wd71SNqKKRf6tNEmsywXy64nixcgzGU,26574
|
85
|
+
brainstate/transform/_loop_collect_return_test.py,sha256=vMptPxrsxWeVSBD8tcOhsO6B9aCjYMcz1QBEgccHNqQ,1811
|
86
|
+
brainstate/transform/_loop_no_collection.py,sha256=bmxd2YWc63lVZhDq12mZPDgGghYqB6ObN63m0IzjqL0,10439
|
87
|
+
brainstate/transform/_loop_no_collection_test.py,sha256=OlYGQEIXrHTr0ZylLBBx02f4FGZCjXYRixkvGCwH-oo,1421
|
88
|
+
brainstate/transform/_make_jaxpr.py,sha256=dHE4TRBALtzVsCg8DwTfTKWJTpim5OMKVp8SdTGPxZw,75343
|
89
|
+
brainstate/transform/_make_jaxpr_test.py,sha256=neK2iu1HK0dAk_XPERE6DUquuEkXBFnbsKiltXJBThE,50638
|
90
|
+
brainstate/transform/_mapping.py,sha256=EtpGI_aABQNk5i9cKuEDr7pCG-YN97cBL_XkvsPH9zs,22099
|
91
|
+
brainstate/transform/_mapping_test.py,sha256=RALLjERRZarsRt-XgCD8_vEvnbSbNnz25H-u9fuvumo,7841
|
92
|
+
brainstate/transform/_progress_bar.py,sha256=z5SIRj3R8IUVMHPnOsXHQDFsEe3wYXIBM_b9-Zk9Weg,9216
|
93
|
+
brainstate/transform/_random.py,sha256=kYa1Xiu_LGHofoPcV9JVBcobTNoOeC9zygJZ6UMeLcI,5430
|
94
|
+
brainstate/transform/_unvmap.py,sha256=DLalab8FY6tzFiha4M6bZ4i4DRDJsaQpzmOxj0SkvhI,6575
|
95
|
+
brainstate/transform/_util.py,sha256=6Dm4QE1zPS9qwxVop8gzzKKOWKXTt_FqjQY1rl6GtKo,11590
|
96
|
+
brainstate/util/__init__.py,sha256=2lKyFfMScZjRh2Zl6PheaOnAJBZPNUPj8yZ82c3hiV0,1181
|
97
|
+
brainstate/util/_others.py,sha256=giUSIzIspCAIuyZL9g6ZFRo4DH6lk0aslTt2ADagAzc,31861
|
98
|
+
brainstate/util/_others_test.py,sha256=0oCdhvFjEoc5de2SkjSmmyoMweJEDABz3pnFzy_hoI0,31497
|
99
|
+
brainstate/util/_pretty_pytree.py,sha256=aACAe6gEwO5dFia_O-cg96biflcxONVakvFrlz99Pyk,48116
|
100
|
+
brainstate/util/_pretty_pytree_test.py,sha256=b0Zg-0p3oaaVSrsdvZZXDWbiPejrQMtIlppQYUFbAtQ,23037
|
101
|
+
brainstate/util/_pretty_repr.py,sha256=QE8BzOhCCMmpJIHsbtdx9p-unfVdya_eVDAin8-7dQU,14420
|
102
|
+
brainstate/util/_pretty_repr_test.py,sha256=dpIYgjtVT3Bp8U7RyHCFKFsvQ2LYpJfdLNrjgzgL4jw,23818
|
103
|
+
brainstate/util/filter.py,sha256=94NfApx2qCMVhoW1Pw0opTCkgTWjPx3pkWT7QA2C9h8,28355
|
104
|
+
brainstate/util/filter_test.py,sha256=vLnxNrCEK77GNo0IOMKaNRP63TkbB9MwpjOZxIPqRxU,32564
|
105
|
+
brainstate/util/struct.py,sha256=wmIvV-QF3qRrr6YP-7uw0ONhDJnXXHlkABY00DTo8_A,25415
|
106
|
+
brainstate/util/struct_test.py,sha256=UCBqTzmsZowHHqy5Dc1mcoJfO6RMzHtp8zTV7nf-_DI,17128
|
107
|
+
brainstate-0.2.1.dist-info/licenses/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
|
108
|
+
brainstate-0.2.1.dist-info/METADATA,sha256=fIjy8AFu5OWak7iKP83FBotT9u-sKDCq8EGwuDi_qLw,4529
|
109
|
+
brainstate-0.2.1.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
|
110
|
+
brainstate-0.2.1.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
|
111
|
+
brainstate-0.2.1.dist-info/RECORD,,
|