brainstate 0.2.0__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 (112) hide show
  1. brainstate/__init__.py +169 -169
  2. brainstate/_compatible_import.py +340 -340
  3. brainstate/_compatible_import_test.py +681 -681
  4. brainstate/_deprecation.py +210 -210
  5. brainstate/_deprecation_test.py +2319 -2319
  6. brainstate/_error.py +45 -45
  7. brainstate/_state.py +1652 -1652
  8. brainstate/_state_test.py +52 -52
  9. brainstate/_utils.py +47 -47
  10. brainstate/environ.py +1495 -1495
  11. brainstate/environ_test.py +1223 -1223
  12. brainstate/graph/__init__.py +22 -22
  13. brainstate/graph/_node.py +240 -240
  14. brainstate/graph/_node_test.py +589 -589
  15. brainstate/graph/_operation.py +1624 -1624
  16. brainstate/graph/_operation_test.py +1147 -1147
  17. brainstate/mixin.py +1433 -1433
  18. brainstate/mixin_test.py +1017 -1017
  19. brainstate/nn/__init__.py +137 -137
  20. brainstate/nn/_activations.py +1100 -1100
  21. brainstate/nn/_activations_test.py +354 -354
  22. brainstate/nn/_collective_ops.py +633 -633
  23. brainstate/nn/_collective_ops_test.py +774 -774
  24. brainstate/nn/_common.py +226 -226
  25. brainstate/nn/_common_test.py +154 -154
  26. brainstate/nn/_conv.py +2010 -2010
  27. brainstate/nn/_conv_test.py +849 -849
  28. brainstate/nn/_delay.py +575 -575
  29. brainstate/nn/_delay_test.py +243 -243
  30. brainstate/nn/_dropout.py +618 -618
  31. brainstate/nn/_dropout_test.py +477 -477
  32. brainstate/nn/_dynamics.py +1267 -1267
  33. brainstate/nn/_dynamics_test.py +67 -67
  34. brainstate/nn/_elementwise.py +1298 -1298
  35. brainstate/nn/_elementwise_test.py +829 -829
  36. brainstate/nn/_embedding.py +408 -408
  37. brainstate/nn/_embedding_test.py +156 -156
  38. brainstate/nn/_event_fixedprob.py +233 -233
  39. brainstate/nn/_event_fixedprob_test.py +115 -115
  40. brainstate/nn/_event_linear.py +83 -83
  41. brainstate/nn/_event_linear_test.py +121 -121
  42. brainstate/nn/_exp_euler.py +254 -254
  43. brainstate/nn/_exp_euler_test.py +377 -377
  44. brainstate/nn/_linear.py +744 -744
  45. brainstate/nn/_linear_test.py +475 -475
  46. brainstate/nn/_metrics.py +1070 -1070
  47. brainstate/nn/_metrics_test.py +611 -611
  48. brainstate/nn/_module.py +384 -384
  49. brainstate/nn/_module_test.py +40 -40
  50. brainstate/nn/_normalizations.py +1334 -1334
  51. brainstate/nn/_normalizations_test.py +699 -699
  52. brainstate/nn/_paddings.py +1020 -1020
  53. brainstate/nn/_paddings_test.py +722 -722
  54. brainstate/nn/_poolings.py +2239 -2239
  55. brainstate/nn/_poolings_test.py +952 -952
  56. brainstate/nn/_rnns.py +946 -946
  57. brainstate/nn/_rnns_test.py +592 -592
  58. brainstate/nn/_utils.py +216 -216
  59. brainstate/nn/_utils_test.py +401 -401
  60. brainstate/nn/init.py +809 -809
  61. brainstate/nn/init_test.py +180 -180
  62. brainstate/random/__init__.py +270 -270
  63. brainstate/random/_rand_funs.py +3938 -3938
  64. brainstate/random/_rand_funs_test.py +640 -640
  65. brainstate/random/_rand_seed.py +675 -675
  66. brainstate/random/_rand_seed_test.py +48 -48
  67. brainstate/random/_rand_state.py +1617 -1617
  68. brainstate/random/_rand_state_test.py +551 -551
  69. brainstate/transform/__init__.py +59 -59
  70. brainstate/transform/_ad_checkpoint.py +176 -176
  71. brainstate/transform/_ad_checkpoint_test.py +49 -49
  72. brainstate/transform/_autograd.py +1025 -1025
  73. brainstate/transform/_autograd_test.py +1289 -1289
  74. brainstate/transform/_conditions.py +316 -316
  75. brainstate/transform/_conditions_test.py +220 -220
  76. brainstate/transform/_error_if.py +94 -94
  77. brainstate/transform/_error_if_test.py +52 -52
  78. brainstate/transform/_eval_shape.py +145 -145
  79. brainstate/transform/_eval_shape_test.py +38 -38
  80. brainstate/transform/_jit.py +399 -399
  81. brainstate/transform/_jit_test.py +143 -143
  82. brainstate/transform/_loop_collect_return.py +675 -675
  83. brainstate/transform/_loop_collect_return_test.py +58 -58
  84. brainstate/transform/_loop_no_collection.py +283 -283
  85. brainstate/transform/_loop_no_collection_test.py +50 -50
  86. brainstate/transform/_make_jaxpr.py +2016 -2016
  87. brainstate/transform/_make_jaxpr_test.py +1510 -1510
  88. brainstate/transform/_mapping.py +529 -529
  89. brainstate/transform/_mapping_test.py +194 -194
  90. brainstate/transform/_progress_bar.py +255 -255
  91. brainstate/transform/_random.py +171 -171
  92. brainstate/transform/_unvmap.py +256 -256
  93. brainstate/transform/_util.py +286 -286
  94. brainstate/typing.py +837 -837
  95. brainstate/typing_test.py +780 -780
  96. brainstate/util/__init__.py +27 -27
  97. brainstate/util/_others.py +1024 -1024
  98. brainstate/util/_others_test.py +962 -962
  99. brainstate/util/_pretty_pytree.py +1301 -1301
  100. brainstate/util/_pretty_pytree_test.py +675 -675
  101. brainstate/util/_pretty_repr.py +462 -462
  102. brainstate/util/_pretty_repr_test.py +696 -696
  103. brainstate/util/filter.py +945 -945
  104. brainstate/util/filter_test.py +911 -911
  105. brainstate/util/struct.py +910 -910
  106. brainstate/util/struct_test.py +602 -602
  107. {brainstate-0.2.0.dist-info → brainstate-0.2.1.dist-info}/METADATA +108 -108
  108. brainstate-0.2.1.dist-info/RECORD +111 -0
  109. {brainstate-0.2.0.dist-info → brainstate-0.2.1.dist-info}/licenses/LICENSE +202 -202
  110. brainstate-0.2.0.dist-info/RECORD +0 -111
  111. {brainstate-0.2.0.dist-info → brainstate-0.2.1.dist-info}/WHEEL +0 -0
  112. {brainstate-0.2.0.dist-info → brainstate-0.2.1.dist-info}/top_level.txt +0 -0
@@ -1,108 +1,108 @@
1
- Metadata-Version: 2.4
2
- Name: brainstate
3
- Version: 0.2.0
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.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,,