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.
Files changed (163) hide show
  1. brainstate/__init__.py +169 -58
  2. brainstate/_compatible_import.py +340 -148
  3. brainstate/_compatible_import_test.py +681 -0
  4. brainstate/_deprecation.py +210 -0
  5. brainstate/_deprecation_test.py +2319 -0
  6. brainstate/{util/error.py → _error.py} +45 -55
  7. brainstate/_state.py +1652 -1605
  8. brainstate/_state_test.py +52 -52
  9. brainstate/_utils.py +47 -47
  10. brainstate/environ.py +1495 -563
  11. brainstate/environ_test.py +1223 -62
  12. brainstate/graph/__init__.py +22 -29
  13. brainstate/graph/_node.py +240 -0
  14. brainstate/graph/_node_test.py +589 -0
  15. brainstate/graph/{_graph_operation.py → _operation.py} +1624 -1738
  16. brainstate/graph/_operation_test.py +1147 -0
  17. brainstate/mixin.py +1433 -365
  18. brainstate/mixin_test.py +1017 -77
  19. brainstate/nn/__init__.py +137 -135
  20. brainstate/nn/_activations.py +1100 -808
  21. brainstate/nn/_activations_test.py +354 -331
  22. brainstate/nn/_collective_ops.py +633 -514
  23. brainstate/nn/_collective_ops_test.py +774 -43
  24. brainstate/nn/_common.py +226 -178
  25. brainstate/nn/_common_test.py +154 -0
  26. brainstate/nn/_conv.py +2010 -501
  27. brainstate/nn/_conv_test.py +849 -238
  28. brainstate/nn/_delay.py +575 -588
  29. brainstate/nn/_delay_test.py +243 -238
  30. brainstate/nn/_dropout.py +618 -426
  31. brainstate/nn/_dropout_test.py +477 -100
  32. brainstate/nn/_dynamics.py +1267 -1343
  33. brainstate/nn/_dynamics_test.py +67 -78
  34. brainstate/nn/_elementwise.py +1298 -1119
  35. brainstate/nn/_elementwise_test.py +830 -169
  36. brainstate/nn/_embedding.py +408 -58
  37. brainstate/nn/_embedding_test.py +156 -0
  38. brainstate/nn/{_fixedprob.py → _event_fixedprob.py} +233 -239
  39. brainstate/nn/{_fixedprob_test.py → _event_fixedprob_test.py} +115 -114
  40. brainstate/nn/{_linear_mv.py → _event_linear.py} +83 -83
  41. brainstate/nn/{_linear_mv_test.py → _event_linear_test.py} +121 -120
  42. brainstate/nn/_exp_euler.py +254 -92
  43. brainstate/nn/_exp_euler_test.py +377 -35
  44. brainstate/nn/_linear.py +744 -424
  45. brainstate/nn/_linear_test.py +475 -107
  46. brainstate/nn/_metrics.py +1070 -0
  47. brainstate/nn/_metrics_test.py +611 -0
  48. brainstate/nn/_module.py +384 -377
  49. brainstate/nn/_module_test.py +40 -40
  50. brainstate/nn/_normalizations.py +1334 -975
  51. brainstate/nn/_normalizations_test.py +699 -73
  52. brainstate/nn/_paddings.py +1020 -0
  53. brainstate/nn/_paddings_test.py +723 -0
  54. brainstate/nn/_poolings.py +2239 -1177
  55. brainstate/nn/_poolings_test.py +953 -217
  56. brainstate/nn/{_rate_rnns.py → _rnns.py} +946 -554
  57. brainstate/nn/_rnns_test.py +593 -0
  58. brainstate/nn/_utils.py +216 -89
  59. brainstate/nn/_utils_test.py +402 -0
  60. brainstate/{init/_random_inits.py → nn/init.py} +809 -553
  61. brainstate/{init/_random_inits_test.py → nn/init_test.py} +180 -149
  62. brainstate/random/__init__.py +270 -24
  63. brainstate/random/_rand_funs.py +3938 -3616
  64. brainstate/random/_rand_funs_test.py +640 -567
  65. brainstate/random/_rand_seed.py +675 -210
  66. brainstate/random/_rand_seed_test.py +48 -48
  67. brainstate/random/_rand_state.py +1617 -1409
  68. brainstate/random/_rand_state_test.py +551 -0
  69. brainstate/transform/__init__.py +59 -0
  70. brainstate/transform/_ad_checkpoint.py +176 -0
  71. brainstate/{compile → transform}/_ad_checkpoint_test.py +49 -49
  72. brainstate/{augment → transform}/_autograd.py +1025 -778
  73. brainstate/{augment → transform}/_autograd_test.py +1289 -1289
  74. brainstate/transform/_conditions.py +316 -0
  75. brainstate/{compile → transform}/_conditions_test.py +220 -220
  76. brainstate/{compile → transform}/_error_if.py +94 -92
  77. brainstate/{compile → transform}/_error_if_test.py +52 -52
  78. brainstate/transform/_eval_shape.py +145 -0
  79. brainstate/{augment → transform}/_eval_shape_test.py +38 -38
  80. brainstate/{compile → transform}/_jit.py +399 -346
  81. brainstate/{compile → transform}/_jit_test.py +143 -143
  82. brainstate/{compile → transform}/_loop_collect_return.py +675 -536
  83. brainstate/{compile → transform}/_loop_collect_return_test.py +58 -58
  84. brainstate/{compile → transform}/_loop_no_collection.py +283 -184
  85. brainstate/{compile → transform}/_loop_no_collection_test.py +50 -50
  86. brainstate/transform/_make_jaxpr.py +2016 -0
  87. brainstate/transform/_make_jaxpr_test.py +1510 -0
  88. brainstate/transform/_mapping.py +529 -0
  89. brainstate/transform/_mapping_test.py +194 -0
  90. brainstate/{compile → transform}/_progress_bar.py +255 -202
  91. brainstate/{augment → transform}/_random.py +171 -151
  92. brainstate/{compile → transform}/_unvmap.py +256 -159
  93. brainstate/transform/_util.py +286 -0
  94. brainstate/typing.py +837 -304
  95. brainstate/typing_test.py +780 -0
  96. brainstate/util/__init__.py +27 -50
  97. brainstate/util/_others.py +1025 -0
  98. brainstate/util/_others_test.py +962 -0
  99. brainstate/util/_pretty_pytree.py +1301 -0
  100. brainstate/util/_pretty_pytree_test.py +675 -0
  101. brainstate/util/{pretty_repr.py → _pretty_repr.py} +462 -328
  102. brainstate/util/_pretty_repr_test.py +696 -0
  103. brainstate/util/filter.py +945 -469
  104. brainstate/util/filter_test.py +912 -0
  105. brainstate/util/struct.py +910 -523
  106. brainstate/util/struct_test.py +602 -0
  107. {brainstate-0.1.10.dist-info → brainstate-0.2.1.dist-info}/METADATA +108 -91
  108. brainstate-0.2.1.dist-info/RECORD +111 -0
  109. {brainstate-0.1.10.dist-info → brainstate-0.2.1.dist-info}/licenses/LICENSE +202 -202
  110. brainstate/augment/__init__.py +0 -30
  111. brainstate/augment/_eval_shape.py +0 -99
  112. brainstate/augment/_mapping.py +0 -1060
  113. brainstate/augment/_mapping_test.py +0 -597
  114. brainstate/compile/__init__.py +0 -38
  115. brainstate/compile/_ad_checkpoint.py +0 -204
  116. brainstate/compile/_conditions.py +0 -256
  117. brainstate/compile/_make_jaxpr.py +0 -888
  118. brainstate/compile/_make_jaxpr_test.py +0 -156
  119. brainstate/compile/_util.py +0 -147
  120. brainstate/functional/__init__.py +0 -27
  121. brainstate/graph/_graph_node.py +0 -244
  122. brainstate/graph/_graph_node_test.py +0 -73
  123. brainstate/graph/_graph_operation_test.py +0 -563
  124. brainstate/init/__init__.py +0 -26
  125. brainstate/init/_base.py +0 -52
  126. brainstate/init/_generic.py +0 -244
  127. brainstate/init/_regular_inits.py +0 -105
  128. brainstate/init/_regular_inits_test.py +0 -50
  129. brainstate/nn/_inputs.py +0 -608
  130. brainstate/nn/_ltp.py +0 -28
  131. brainstate/nn/_neuron.py +0 -705
  132. brainstate/nn/_neuron_test.py +0 -161
  133. brainstate/nn/_others.py +0 -46
  134. brainstate/nn/_projection.py +0 -486
  135. brainstate/nn/_rate_rnns_test.py +0 -63
  136. brainstate/nn/_readout.py +0 -209
  137. brainstate/nn/_readout_test.py +0 -53
  138. brainstate/nn/_stp.py +0 -236
  139. brainstate/nn/_synapse.py +0 -505
  140. brainstate/nn/_synapse_test.py +0 -131
  141. brainstate/nn/_synaptic_projection.py +0 -423
  142. brainstate/nn/_synouts.py +0 -162
  143. brainstate/nn/_synouts_test.py +0 -57
  144. brainstate/nn/metrics.py +0 -388
  145. brainstate/optim/__init__.py +0 -38
  146. brainstate/optim/_base.py +0 -64
  147. brainstate/optim/_lr_scheduler.py +0 -448
  148. brainstate/optim/_lr_scheduler_test.py +0 -50
  149. brainstate/optim/_optax_optimizer.py +0 -152
  150. brainstate/optim/_optax_optimizer_test.py +0 -53
  151. brainstate/optim/_sgd_optimizer.py +0 -1104
  152. brainstate/random/_random_for_unit.py +0 -52
  153. brainstate/surrogate.py +0 -1957
  154. brainstate/transform.py +0 -23
  155. brainstate/util/caller.py +0 -98
  156. brainstate/util/others.py +0 -540
  157. brainstate/util/pretty_pytree.py +0 -945
  158. brainstate/util/pretty_pytree_test.py +0 -159
  159. brainstate/util/pretty_table.py +0 -2954
  160. brainstate/util/scaling.py +0 -258
  161. brainstate-0.1.10.dist-info/RECORD +0 -130
  162. {brainstate-0.1.10.dist-info → brainstate-0.2.1.dist-info}/WHEEL +0 -0
  163. {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.10
4
- Summary: A ``State``-based Transformation System for Program Compilation and Augmentation.
5
- Home-page: https://github.com/chaobrain/brainstate
6
- Author: BrainState Developers
7
- Author-email: BrainState Developers <chao.brain@qq.com>
8
- License: Apache-2.0 license
9
- Project-URL: homepage, http://github.com/chaobrain/brainstate
10
- Project-URL: repository, http://github.com/chaobrain/brainstate
11
- Keywords: computational neuroscience,brain-inspired computation,brain dynamics programming
12
- Classifier: Natural Language :: English
13
- Classifier: Operating System :: OS Independent
14
- Classifier: Development Status :: 4 - Beta
15
- Classifier: Intended Audience :: Developers
16
- Classifier: Intended Audience :: Science/Research
17
- Classifier: Programming Language :: Python :: 3
18
- Classifier: Programming Language :: Python :: 3.10
19
- Classifier: Programming Language :: Python :: 3.11
20
- Classifier: Programming Language :: Python :: 3.12
21
- Classifier: Programming Language :: Python :: 3.13
22
- Classifier: License :: OSI Approved :: Apache Software License
23
- Classifier: Programming Language :: Python
24
- Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
25
- Classifier: Topic :: Scientific/Engineering :: Mathematics
26
- Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
27
- Classifier: Topic :: Software Development :: Libraries
28
- Requires-Python: >=3.10
29
- Description-Content-Type: text/markdown
30
- License-File: LICENSE
31
- Requires-Dist: jax
32
- Requires-Dist: jaxlib
33
- Requires-Dist: numpy
34
- Requires-Dist: brainunit>=0.1.0
35
- Requires-Dist: brainevent
36
- Provides-Extra: testing
37
- Requires-Dist: pytest; extra == "testing"
38
- Dynamic: author
39
- Dynamic: home-page
40
- Dynamic: license-file
41
- Dynamic: requires-python
42
-
43
-
44
- # A ``State``-based Transformation System for Program Compilation and Augmentation
45
-
46
-
47
-
48
- <p align="center">
49
- <img alt="Header image of brainstate." src="https://raw.githubusercontent.com/chaobrain/brainstate/main/docs/_static/brainstate.png" width=40%>
50
- </p>
51
-
52
-
53
-
54
- <p align="center">
55
- <a href="https://pypi.org/project/brainstate/"><img alt="Supported Python Version" src="https://img.shields.io/pypi/pyversions/brainstate"></a>
56
- <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>
57
- <a href='https://brainstate.readthedocs.io/?badge=latest'>
58
- <img src='https://readthedocs.org/projects/brainstate/badge/?version=latest' alt='Documentation Status' />
59
- </a>
60
- <a href="https://badge.fury.io/py/brainstate"><img alt="PyPI version" src="https://badge.fury.io/py/brainstate.svg"></a>
61
- <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>
62
- <a href="https://pepy.tech/projects/brainstate"><img src="https://static.pepy.tech/badge/brainstate" alt="PyPI Downloads"></a>
63
- <a href="https://doi.org/10.5281/zenodo.14970015"><img src="https://zenodo.org/badge/811300394.svg" alt="DOI"></a>
64
- </p>
65
-
66
-
67
- ## Installation
68
-
69
- You can install ``brainstate`` via pip:
70
-
71
- ```bash
72
- pip install brainstate --upgrade
73
- ```
74
-
75
- Alternatively, you can install `BrainX`, which bundles `brainstate` with other compatible packages for a comprehensive brain modeling ecosystem:
76
-
77
- ```bash
78
- pip install BrainX -U
79
- ```
80
-
81
- ## Documentation
82
-
83
- The official documentation is hosted on Read the Docs: [https://brainstate.readthedocs.io/](https://brainstate.readthedocs.io/)
84
-
85
-
86
-
87
- ## See also the brain modeling ecosystem
88
-
89
- We are building the brain modeling ecosystem: https://brainmodeling.readthedocs.io/
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,,