brainstate 0.1.8__py2.py3-none-any.whl → 0.1.10__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 (133) hide show
  1. brainstate/__init__.py +58 -51
  2. brainstate/_compatible_import.py +148 -148
  3. brainstate/_state.py +1605 -1663
  4. brainstate/_state_test.py +52 -52
  5. brainstate/_utils.py +47 -47
  6. brainstate/augment/__init__.py +30 -30
  7. brainstate/augment/_autograd.py +778 -778
  8. brainstate/augment/_autograd_test.py +1289 -1289
  9. brainstate/augment/_eval_shape.py +99 -99
  10. brainstate/augment/_eval_shape_test.py +38 -38
  11. brainstate/augment/_mapping.py +1060 -1060
  12. brainstate/augment/_mapping_test.py +597 -597
  13. brainstate/augment/_random.py +151 -151
  14. brainstate/compile/__init__.py +38 -38
  15. brainstate/compile/_ad_checkpoint.py +204 -204
  16. brainstate/compile/_ad_checkpoint_test.py +49 -49
  17. brainstate/compile/_conditions.py +256 -256
  18. brainstate/compile/_conditions_test.py +220 -220
  19. brainstate/compile/_error_if.py +92 -92
  20. brainstate/compile/_error_if_test.py +52 -52
  21. brainstate/compile/_jit.py +346 -346
  22. brainstate/compile/_jit_test.py +143 -143
  23. brainstate/compile/_loop_collect_return.py +536 -536
  24. brainstate/compile/_loop_collect_return_test.py +58 -58
  25. brainstate/compile/_loop_no_collection.py +184 -184
  26. brainstate/compile/_loop_no_collection_test.py +50 -50
  27. brainstate/compile/_make_jaxpr.py +888 -888
  28. brainstate/compile/_make_jaxpr_test.py +156 -156
  29. brainstate/compile/_progress_bar.py +202 -202
  30. brainstate/compile/_unvmap.py +159 -159
  31. brainstate/compile/_util.py +147 -147
  32. brainstate/environ.py +563 -563
  33. brainstate/environ_test.py +62 -62
  34. brainstate/functional/__init__.py +27 -26
  35. brainstate/graph/__init__.py +29 -29
  36. brainstate/graph/_graph_node.py +244 -244
  37. brainstate/graph/_graph_node_test.py +73 -73
  38. brainstate/graph/_graph_operation.py +1738 -1738
  39. brainstate/graph/_graph_operation_test.py +563 -563
  40. brainstate/init/__init__.py +26 -26
  41. brainstate/init/_base.py +52 -52
  42. brainstate/init/_generic.py +244 -244
  43. brainstate/init/_random_inits.py +553 -553
  44. brainstate/init/_random_inits_test.py +149 -149
  45. brainstate/init/_regular_inits.py +105 -105
  46. brainstate/init/_regular_inits_test.py +50 -50
  47. brainstate/mixin.py +365 -363
  48. brainstate/mixin_test.py +77 -73
  49. brainstate/nn/__init__.py +135 -131
  50. brainstate/{functional → nn}/_activations.py +808 -813
  51. brainstate/{functional → nn}/_activations_test.py +331 -331
  52. brainstate/nn/_collective_ops.py +514 -514
  53. brainstate/nn/_collective_ops_test.py +43 -43
  54. brainstate/nn/_common.py +178 -178
  55. brainstate/nn/_conv.py +501 -501
  56. brainstate/nn/_conv_test.py +238 -238
  57. brainstate/nn/_delay.py +588 -502
  58. brainstate/nn/_delay_test.py +238 -184
  59. brainstate/nn/_dropout.py +426 -426
  60. brainstate/nn/_dropout_test.py +100 -100
  61. brainstate/nn/_dynamics.py +1343 -1343
  62. brainstate/nn/_dynamics_test.py +78 -78
  63. brainstate/nn/_elementwise.py +1119 -1119
  64. brainstate/nn/_elementwise_test.py +169 -169
  65. brainstate/nn/_embedding.py +58 -58
  66. brainstate/nn/_exp_euler.py +92 -92
  67. brainstate/nn/_exp_euler_test.py +35 -35
  68. brainstate/nn/_fixedprob.py +239 -239
  69. brainstate/nn/_fixedprob_test.py +114 -114
  70. brainstate/nn/_inputs.py +608 -608
  71. brainstate/nn/_linear.py +424 -424
  72. brainstate/nn/_linear_mv.py +83 -83
  73. brainstate/nn/_linear_mv_test.py +120 -120
  74. brainstate/nn/_linear_test.py +107 -107
  75. brainstate/nn/_ltp.py +28 -28
  76. brainstate/nn/_module.py +377 -377
  77. brainstate/nn/_module_test.py +40 -40
  78. brainstate/nn/_neuron.py +705 -705
  79. brainstate/nn/_neuron_test.py +161 -161
  80. brainstate/nn/_normalizations.py +975 -918
  81. brainstate/nn/_normalizations_test.py +73 -73
  82. brainstate/{functional → nn}/_others.py +46 -46
  83. brainstate/nn/_poolings.py +1177 -1177
  84. brainstate/nn/_poolings_test.py +217 -217
  85. brainstate/nn/_projection.py +486 -486
  86. brainstate/nn/_rate_rnns.py +554 -554
  87. brainstate/nn/_rate_rnns_test.py +63 -63
  88. brainstate/nn/_readout.py +209 -209
  89. brainstate/nn/_readout_test.py +53 -53
  90. brainstate/nn/_stp.py +236 -236
  91. brainstate/nn/_synapse.py +505 -505
  92. brainstate/nn/_synapse_test.py +131 -131
  93. brainstate/nn/_synaptic_projection.py +423 -423
  94. brainstate/nn/_synouts.py +162 -162
  95. brainstate/nn/_synouts_test.py +57 -57
  96. brainstate/nn/_utils.py +89 -89
  97. brainstate/nn/metrics.py +388 -388
  98. brainstate/optim/__init__.py +38 -38
  99. brainstate/optim/_base.py +64 -64
  100. brainstate/optim/_lr_scheduler.py +448 -448
  101. brainstate/optim/_lr_scheduler_test.py +50 -50
  102. brainstate/optim/_optax_optimizer.py +152 -152
  103. brainstate/optim/_optax_optimizer_test.py +53 -53
  104. brainstate/optim/_sgd_optimizer.py +1104 -1104
  105. brainstate/random/__init__.py +24 -24
  106. brainstate/random/_rand_funs.py +3616 -3616
  107. brainstate/random/_rand_funs_test.py +567 -567
  108. brainstate/random/_rand_seed.py +210 -210
  109. brainstate/random/_rand_seed_test.py +48 -48
  110. brainstate/random/_rand_state.py +1409 -1409
  111. brainstate/random/_random_for_unit.py +52 -52
  112. brainstate/surrogate.py +1957 -1957
  113. brainstate/transform.py +23 -23
  114. brainstate/typing.py +304 -304
  115. brainstate/util/__init__.py +50 -50
  116. brainstate/util/caller.py +98 -98
  117. brainstate/util/error.py +55 -55
  118. brainstate/util/filter.py +469 -469
  119. brainstate/util/others.py +540 -540
  120. brainstate/util/pretty_pytree.py +945 -945
  121. brainstate/util/pretty_pytree_test.py +159 -159
  122. brainstate/util/pretty_repr.py +328 -328
  123. brainstate/util/pretty_table.py +2954 -2954
  124. brainstate/util/scaling.py +258 -258
  125. brainstate/util/struct.py +523 -523
  126. {brainstate-0.1.8.dist-info → brainstate-0.1.10.dist-info}/METADATA +91 -99
  127. brainstate-0.1.10.dist-info/RECORD +130 -0
  128. {brainstate-0.1.8.dist-info → brainstate-0.1.10.dist-info}/WHEEL +1 -1
  129. {brainstate-0.1.8.dist-info → brainstate-0.1.10.dist-info/licenses}/LICENSE +202 -202
  130. brainstate/functional/_normalization.py +0 -81
  131. brainstate/functional/_spikes.py +0 -204
  132. brainstate-0.1.8.dist-info/RECORD +0 -132
  133. {brainstate-0.1.8.dist-info → brainstate-0.1.10.dist-info}/top_level.txt +0 -0
@@ -1,99 +1,91 @@
1
- Metadata-Version: 2.1
2
- Name: brainstate
3
- Version: 0.1.8
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.0.4
35
- Requires-Dist: brainevent
36
- Provides-Extra: cpu
37
- Requires-Dist: jax[cpu]; extra == "cpu"
38
- Requires-Dist: brainunit[cpu]; extra == "cpu"
39
- Requires-Dist: brainevent[cpu]; extra == "cpu"
40
- Provides-Extra: cuda12
41
- Requires-Dist: jax[cuda12]; extra == "cuda12"
42
- Requires-Dist: brainunit[cuda12]; extra == "cuda12"
43
- Requires-Dist: brainevent[cuda12]; extra == "cuda12"
44
- Provides-Extra: testing
45
- Requires-Dist: pytest; extra == "testing"
46
- Provides-Extra: tpu
47
- Requires-Dist: jax[tpu]; extra == "tpu"
48
- Requires-Dist: brainunit[tpu]; extra == "tpu"
49
- Requires-Dist: brainevent[tpu]; extra == "tpu"
50
-
51
-
52
- # A ``State``-based Transformation System for Program Compilation and Augmentation
53
-
54
-
55
-
56
- <p align="center">
57
- <img alt="Header image of brainstate." src="https://raw.githubusercontent.com/chaobrain/brainstate/main/docs/_static/brainstate.png" width=40%>
58
- </p>
59
-
60
-
61
-
62
- <p align="center">
63
- <a href="https://pypi.org/project/brainstate/"><img alt="Supported Python Version" src="https://img.shields.io/pypi/pyversions/brainstate"></a>
64
- <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>
65
- <a href='https://brainstate.readthedocs.io/?badge=latest'>
66
- <img src='https://readthedocs.org/projects/brainstate/badge/?version=latest' alt='Documentation Status' />
67
- </a>
68
- <a href="https://badge.fury.io/py/brainstate"><img alt="PyPI version" src="https://badge.fury.io/py/brainstate.svg"></a>
69
- <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>
70
- <a href="https://pepy.tech/projects/brainstate"><img src="https://static.pepy.tech/badge/brainstate" alt="PyPI Downloads"></a>
71
- <a href="https://doi.org/10.5281/zenodo.14970015"><img src="https://zenodo.org/badge/811300394.svg" alt="DOI"></a>
72
- </p>
73
-
74
-
75
- ## Installation
76
-
77
- You can install ``brainstate`` via pip:
78
-
79
- ```bash
80
- pip install brainstate --upgrade
81
- ```
82
-
83
- Alternatively, you can install `BrainX`, which bundles `brainstate` with other compatible packages for a comprehensive brain modeling ecosystem:
84
-
85
- ```bash
86
- pip install BrainX -U
87
- ```
88
-
89
- ## Documentation
90
-
91
- The official documentation is hosted on Read the Docs: [https://brainstate.readthedocs.io/](https://brainstate.readthedocs.io/)
92
-
93
-
94
-
95
- ## See also the brain modeling ecosystem
96
-
97
- We are building the brain modeling ecosystem: https://brainmodeling.readthedocs.io/
98
-
99
-
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
+
@@ -0,0 +1,130 @@
1
+ brainstate/__init__.py,sha256=lG0hJsHre06CcJMm_Em25hkKKFH9J8ZqnyGPq3iixZI,1451
2
+ brainstate/_compatible_import.py,sha256=kCJy0OIr2VAJHqyk2bdAZ1s2VAFKeEgOpfFosy9wi7A,4482
3
+ brainstate/_state.py,sha256=iZeLyjaCG39yXHh7y5LkOC2_MwkLgEnnKKr6puHkeME,56966
4
+ brainstate/_state_test.py,sha256=UqbXHhJNAHLehvVMnxnOUfWT1PVRWFawOvoyjJvGIRo,1586
5
+ brainstate/_utils.py,sha256=44Yy4cBlsSfZln7CEF8XfNCsZiIIDPPmC9LT17OOi5s,1610
6
+ brainstate/environ.py,sha256=mVIb_4KUP7s26tBMIuEfoo47Q6lcnY_XlCIE5rRO5m8,17230
7
+ brainstate/environ_test.py,sha256=KYnPRv-ZgNDrvoTyNGuyO5ElLBoSFR4AN-B1hi2tp4k,1983
8
+ brainstate/mixin.py,sha256=88obor0L9qhoj4Kt_Q-dID9ZyPO-P7pasgbbykdWeoA,10651
9
+ brainstate/mixin_test.py,sha256=xhKEQMvpHQE74o_hQ88EID-mL_tSF7eStCSCVUfY5Tk,3463
10
+ brainstate/surrogate.py,sha256=liSc2ndjrSToxdNDbAqSJ_PsxzynyLFesUS6vYJp3jE,51904
11
+ brainstate/transform.py,sha256=s1CuEIIWPswuUst7UN4vw5tevg0w2iMiTR-WRp_benY,835
12
+ brainstate/typing.py,sha256=CCwCLtVC0k7w5j--KrJn5klp2s4_Tm63-cDZDhScubs,10215
13
+ brainstate/augment/__init__.py,sha256=TmJdG7lTwsuWykzEoQMhBcCCLhjHFzwb8XjLL3ZzRbE,1199
14
+ brainstate/augment/_autograd.py,sha256=YYsBi2WANGc68sF9nC7Td8wp-aD1i3UtkhNaZ_9igVc,30106
15
+ brainstate/augment/_autograd_test.py,sha256=-zFRq-PD0-IfLaMjWrqGFK8JY_bXiPARaQjMTN01mUs,44107
16
+ brainstate/augment/_eval_shape.py,sha256=ohupDUjA4NuRpopOK8adxk-J3_-HGAt-lxT4IdZC9gc,3769
17
+ brainstate/augment/_eval_shape_test.py,sha256=A_nwiXkY-pAFuNzpg28E0_Kb4HFSuy89CBWCbQZs1wY,1352
18
+ brainstate/augment/_mapping.py,sha256=B-31rDUsgN1TBhox2Zt7ldjy3LId6CGS7LzDUAM_D54,42755
19
+ brainstate/augment/_mapping_test.py,sha256=XnnLMByzoJlm1CEXDE03jABtkXo-mR-6WPqpRjIMnA4,21349
20
+ brainstate/augment/_random.py,sha256=4LN6hYb0xJgRdjHIp7b1jbINlAapdtxoc2TtfWaEuNY,5197
21
+ brainstate/compile/__init__.py,sha256=X_8ixcOTfwNBj9UrHRsQbPF87d-8DUmqMGjAtXCd23U,1436
22
+ brainstate/compile/_ad_checkpoint.py,sha256=nEvixC-0YmqG30mVPS1AXlYMy7W6mVVGUq_xQGMtQAw,9154
23
+ brainstate/compile/_ad_checkpoint_test.py,sha256=rXBzX6BHGE2Ygg2tEHbHiTt8w8xHrfu0ykxhE-WcELM,1694
24
+ brainstate/compile/_conditions.py,sha256=l80YEX6Usjdbe1gpUp8EiKJdShUmoCzdVEo4xz5PBs0,9927
25
+ brainstate/compile/_conditions_test.py,sha256=xQCJfoVUfYe_YZSHEPrN2Vs2MJmWvUcVMaGIml8iNUc,8420
26
+ brainstate/compile/_error_if.py,sha256=ZPEnv-cvP1FIO5RThm-6F_UY5u19XSLHtf6ccyHxKV0,2559
27
+ brainstate/compile/_error_if_test.py,sha256=RyDLasCACaFFSpeApixelQ4s3xo39fLuMfWDukUPE5M,1845
28
+ brainstate/compile/_jit.py,sha256=LMMJjowymimr50iMR-YFvmvIej1o252lsi8xpxf3K7w,14150
29
+ brainstate/compile/_jit_test.py,sha256=4ODCh1ime6iBmb_2C8RNOSfECWXNrDx5ojwy-n0m5LU,3992
30
+ brainstate/compile/_loop_collect_return.py,sha256=K5W-bIcMDgH4k7UM_Gem5Y5VLkdnYosQFfJ4afID5Yc,22805
31
+ brainstate/compile/_loop_collect_return_test.py,sha256=0IfYFqFR8cMSh5LE8HciMVREcF4IN2g-7Yuaizr314A,1750
32
+ brainstate/compile/_loop_no_collection.py,sha256=IlJx0wcvQaTKWYM7zgsxHaW9CoG5c1nXqqxBlX0EIsk,7365
33
+ brainstate/compile/_loop_no_collection_test.py,sha256=r75XGnTH8Q52h7VVKSbwNPWblCc7-a7PlJU8welPn9E,1369
34
+ brainstate/compile/_make_jaxpr.py,sha256=VxOgwnxg52lD0U9AoGukkqhdgcGsGoIyxp6eAC_K7mQ,36479
35
+ brainstate/compile/_make_jaxpr_test.py,sha256=oDQunz5OQUVtkJiM8QKkppQCRUknnGz_nTJLCxB02tM,5008
36
+ brainstate/compile/_progress_bar.py,sha256=HfjEOyVGlWWNh47wcQDWL0Nd1826QYJlc_gKtw8qsEM,7300
37
+ brainstate/compile/_unvmap.py,sha256=6Hgw_8wxkjmVSp_DNDZGmuA4AvcHjsg0IjwcYDwXSAk,3961
38
+ brainstate/compile/_util.py,sha256=O_STS1cjjZR3X85r_5j1iACD9ey_xozGeuvVmpKNiFM,6128
39
+ brainstate/functional/__init__.py,sha256=pfq1O-Q2yX4qU-lhRX61yLTbuWOVej5tEGyBrDEVEUg,1063
40
+ brainstate/graph/__init__.py,sha256=uM1Me3laclHcPbGoV8jrlqf2wB09po9SLPgcaqWVFwA,1266
41
+ brainstate/graph/_graph_node.py,sha256=XuKcr6rCVVuIUbKEIOHTDwH_ad1Xji4hISfdcGGP3qo,6591
42
+ brainstate/graph/_graph_node_test.py,sha256=9Y4OeMnPH1mH_VBH1Rtlxrmm75eJG3ZCBZ-E_20-XSw,2759
43
+ brainstate/graph/_graph_operation.py,sha256=1Ls_ZIUQRvsPyJ7f0VcAqRMclZ6NUVBwmHDU6J8y8r4,62237
44
+ brainstate/graph/_graph_operation_test.py,sha256=NRTBxXmSDpN5aQFPN6exi_4DzLvY_yh4wCteyw92x4k,19166
45
+ brainstate/init/__init__.py,sha256=MSBAzAXh6Us4fV5nkOtAWq891UhTj2f2Rmfmjm30-po,1071
46
+ brainstate/init/_base.py,sha256=cnqFmsy9SMRnwcL1RHZ1x7Ao14-_SVEsjbqNhIM-dXw,1581
47
+ brainstate/init/_generic.py,sha256=-H7pXrAcDJv_-0ktkJqs7sfXf0nKSap_PU16K5NtHSE,7748
48
+ brainstate/init/_random_inits.py,sha256=sX3gtLVUN3yoYkAvFwq4ddf5KsD7kDDXHNHlm4wvoz4,17656
49
+ brainstate/init/_random_inits_test.py,sha256=q-NxvjQpyKxeAJ4biI3nrHBoLFticIc9-lq9T_IEQi4,5138
50
+ brainstate/init/_regular_inits.py,sha256=A3YMrieLbWSCAtIHxcXxIZLja1hGL-9zrfbBuagBAN8,3044
51
+ brainstate/init/_regular_inits_test.py,sha256=9FE0vQpJsoMOLsD5cHtlb8cXKHSlvELNzDkLqmG3kfY,1770
52
+ brainstate/nn/__init__.py,sha256=k2MANfL0vgiqOBFT1j1dRLUCSRZdPCBXi5RF6WMHOxE,3693
53
+ brainstate/nn/_activations.py,sha256=itBWscfrTWY4fu2o_s0q2OIzrpiaQr5IPy6jouLCXVY,20580
54
+ brainstate/nn/_activations_test.py,sha256=rcaI-t_sTyhjx-WkiUxPIpfjFe3PR9s3cIkbqO4fLCg,13228
55
+ brainstate/nn/_collective_ops.py,sha256=z6gaox3CNrfjpiFdWx3G_iXNtdVBToGqaAwEEeQuFcY,16904
56
+ brainstate/nn/_collective_ops_test.py,sha256=bZlWQDldsdaX89exbhnXPTP-sONrk6riii3hvhNy6S0,1366
57
+ brainstate/nn/_common.py,sha256=N3AaSqRLDOtdynGE_xO03EJXvyh55G6O1q1ZrHKkhf8,6253
58
+ brainstate/nn/_conv.py,sha256=r77NK9xAUy4l6bh680wAlV69FzzQzLE56gcxlLKn5as,17947
59
+ brainstate/nn/_conv_test.py,sha256=Ekkra0Inn0aiBTJcF7orrrlpWXWs1Vv0xjcJxQ9GiDY,8624
60
+ brainstate/nn/_delay.py,sha256=jMCvnYrivXJwAmZQBfLxM29wSlkgmW2FudXGT_yt8kk,23051
61
+ brainstate/nn/_delay_test.py,sha256=p0YbiqpqRe_ulhFr9qDmMt2qihXg5En9O_hWxAlHFVc,10158
62
+ brainstate/nn/_dropout.py,sha256=xiKeTsLdq5a3kCmhjf9XzuTc2VMjF6NEqJFLUqqKfKw,17357
63
+ brainstate/nn/_dropout_test.py,sha256=J0Ju3weOWWk6ZYkG7_3At8Kq-xnE5WK8_svfZBJAq04,4346
64
+ brainstate/nn/_dynamics.py,sha256=TA_SKKqso5dR27wE2paaEbLdnnoF9BBmSUW0Yacm6c0,46893
65
+ brainstate/nn/_dynamics_test.py,sha256=7A9SHyc5K2VpPWmjcMwuiIBzcubBfzzybC018fMfORE,2691
66
+ brainstate/nn/_elementwise.py,sha256=vWP_UMfSXbHLhd8HkuuBwW3YdYyye1IWGCWkoKIL3LM,32334
67
+ brainstate/nn/_elementwise_test.py,sha256=u2X6K48xGN9P1IVrugl9GSQzxYb1QczUnawERjcnulo,5519
68
+ brainstate/nn/_embedding.py,sha256=xK4m6puPNkLHYluK59AL28zb4l2OJunr9H74l703p_s,2080
69
+ brainstate/nn/_exp_euler.py,sha256=Aoc_oiWeNVkTrS8R5ryazXZ5ZonnYuC1DkDFn_sAGNs,3201
70
+ brainstate/nn/_exp_euler_test.py,sha256=768KS8iVMXogQ56p1lL4rLAA9VFj27yNSBRcpDMgRcs,1192
71
+ brainstate/nn/_fixedprob.py,sha256=ovpu2sjiGwe6OUwhsd4eH7v7moYSUFD3WYGAlhi97Rc,9765
72
+ brainstate/nn/_fixedprob_test.py,sha256=QcIrRQ7fc6_ZhTDBJVB42xP41yc2LYwWyCmZ_46apKI,3815
73
+ brainstate/nn/_inputs.py,sha256=cGzVogXQ76U3bylV_Dxx7E0tNykU54wVZTQ-9_V1UyI,19995
74
+ brainstate/nn/_linear.py,sha256=06IOcqG4yilyUzIUYLnhPYVSlqj-Rz8dklJuGj9wXyM,14076
75
+ brainstate/nn/_linear_mv.py,sha256=jCDj_daTwHvy1_YkGsN70ReCaywjUXpZQGJObjBfP1w,2552
76
+ brainstate/nn/_linear_mv_test.py,sha256=DpO9tItPP2veLnPdVUu_GI4bORbcKOAzR1jsUPUf6uo,3744
77
+ brainstate/nn/_linear_test.py,sha256=RtpN1T6a86Fy5sJA7IzrXZDs9utXg9CFPmiQ_n7w1FI,2818
78
+ brainstate/nn/_ltp.py,sha256=yvVWEsFRP6CmDxkRA6HT1yHqkhd2D1AdDD-M9soTorw,833
79
+ brainstate/nn/_module.py,sha256=Fe6iyHyvOPqYzU9oyxbHHSnFN4YeKBkvb3y26SbakCo,12406
80
+ brainstate/nn/_module_test.py,sha256=gQ6WHjNg15NN6R_W1wc47o1Jme648XI5oDf4bnU9IWo,1411
81
+ brainstate/nn/_neuron.py,sha256=AhGalDvfsIzH_V-nbf9Q-I1DtfvGUrkbryK8Pefqz7Y,26736
82
+ brainstate/nn/_neuron_test.py,sha256=b--unQkOpUF_hTdTFN2_0NuzK14aJbuYpQLdSe28rcw,6232
83
+ brainstate/nn/_normalizations.py,sha256=0CfnZHLuGwj00VetaF9jAVe1jgvquflYzp6cY6wPuik,38159
84
+ brainstate/nn/_normalizations_test.py,sha256=D2zXErQJ7CYeAI2HHCIckl7JuWaVvpLBo5l38wTLQl4,2425
85
+ brainstate/nn/_others.py,sha256=nAjzEh2t_hoXWKTwwqiKlcuD6Vi0Meo276V-YoJAzLk,1653
86
+ brainstate/nn/_poolings.py,sha256=4ETe0CcRXk6_ECHdCM_RE6ZzCc5916t2L6KJtK2EIyA,45960
87
+ brainstate/nn/_poolings_test.py,sha256=lFR2RNNLegSBLBzAzqLTxgZSBGRWGGZoTjmJAKVIHK8,7280
88
+ brainstate/nn/_projection.py,sha256=7Gu75KKw5cyixZc850jo8vhyvZzwBvyIFzP2eNtziNM,17012
89
+ brainstate/nn/_rate_rnns.py,sha256=YZ6ishu0vuK6G4i6Rzt6kEG7sFTI2MqkSzbDRbZc6XU,20122
90
+ brainstate/nn/_rate_rnns_test.py,sha256=i9fDoGs-rDfXqMdKXAiPZ5GgsqBa4m7gEOp5HWV4WCo,2381
91
+ brainstate/nn/_readout.py,sha256=A6fbztI_aG_AXXHAnwzRFcEfKyTVaB3jZHOL9GZqwRw,6907
92
+ brainstate/nn/_readout_test.py,sha256=fwypIlbjwFLfLNgfCPTNf9gJJmKv64VEuI9-zFYHCt0,2083
93
+ brainstate/nn/_stp.py,sha256=90pHMBC50P7aIf-5u8bxz1-zdFLJ0mSdzCrN3IV1GDM,8409
94
+ brainstate/nn/_synapse.py,sha256=3ozES8SM53vARpWvRs9_5bjK0uClQzHGPt6_Pnbtrao,19355
95
+ brainstate/nn/_synapse_test.py,sha256=bQCLVFHbmXCc7apoIOejN2VnMqxDHx7l-nHoPrCQSGg,4863
96
+ brainstate/nn/_synaptic_projection.py,sha256=b23QxIQR7_Wot8eg_dHjtbnTxOFOpf3wUOh479q2v7Y,17494
97
+ brainstate/nn/_synouts.py,sha256=d33Ey0AiVUoGmu9LZ83Uvu078C-vdRSVEWwfQaxlzp0,4362
98
+ brainstate/nn/_synouts_test.py,sha256=6m0CLaal96PnZoYw2sFYlM1yrfQ1K_tsvpfTCO4KUhk,2187
99
+ brainstate/nn/_utils.py,sha256=m4aDCM3yGe43jZZKP1a11GKl18vZ_AeWSJg7muNI0AE,3041
100
+ brainstate/nn/metrics.py,sha256=gd7fEReCA3BdkqWHCqmR-7tIPtnnHJ_G9iydcG64rO4,14340
101
+ brainstate/optim/__init__.py,sha256=LReU1WjMJBndmiokuVvLVWlqbqnqpUBg7zw_fuSCfVg,1157
102
+ brainstate/optim/_base.py,sha256=ozskxALXAn0CmFFtc0Zx5r9UxixK9E7DO0tuZCSp1_Q,1771
103
+ brainstate/optim/_lr_scheduler.py,sha256=XOm-43bvrFb30ULJ76O47UYr9YSYaMrq5NKPJzb-gqA,14841
104
+ brainstate/optim/_lr_scheduler_test.py,sha256=Z0Eyzyq2oojOByl5V5aP34d4Z_CbPB3OD3icrPin8z8,1740
105
+ brainstate/optim/_optax_optimizer.py,sha256=QmQk-Nmy3Qm_33p88IBgSo8a51l5B2d_0Gboj9OljoE,5185
106
+ brainstate/optim/_optax_optimizer_test.py,sha256=h0XOC01O_-8wDy6mOjnMo2E9BEklmaVruITHaRpw5lU,1722
107
+ brainstate/optim/_sgd_optimizer.py,sha256=qmgkH5zu2YzaMplCcUL2LMB81XY2GqzhRnqIQrUPs0U,45006
108
+ brainstate/random/__init__.py,sha256=TqrGj5dthOtTdcUReyZZZAv2IIJJjcGAaV9IVYwWaPQ,1021
109
+ brainstate/random/_rand_funs.py,sha256=VVfi5WfJWNtZjiQYV9FUzjZoqw8pLIuewEiJIDUwmFk,134289
110
+ brainstate/random/_rand_funs_test.py,sha256=WFS1H87XgOS8xas7DXcE_0Dx1p4oJz1J6AxLGqX-SEI,20123
111
+ brainstate/random/_rand_seed.py,sha256=LzrAypJX-b0krDv2S6I3z64WllnrBZFUIZMDO__TCm8,5788
112
+ brainstate/random/_rand_seed_test.py,sha256=mNYMmtXfuvMzq70p7NOPTbdGfKhecQZMUEggDSmohY0,1488
113
+ brainstate/random/_rand_state.py,sha256=ymPotAvoGmBQI8Ja5PaOJRYJaZQya3d91OFcs6hmXP8,54114
114
+ brainstate/random/_random_for_unit.py,sha256=S7LjwssN9PiujAy2tsua8NFWkhmeUwYY4ifrVFGk_a0,2005
115
+ brainstate/util/__init__.py,sha256=d3C7zRypxfAA_baWhbNPo9vWCwkUY61mjx1TFUy8g54,1505
116
+ brainstate/util/caller.py,sha256=k0MdMB-PkIHccAFCYsgCZgrvu8XIHgix_cT8df0DMWY,2654
117
+ brainstate/util/error.py,sha256=Rq4q_41jrJ8GwOmlfSO5rSKNdGHlPAhSxHDHpHXY-bQ,1702
118
+ brainstate/util/filter.py,sha256=aB2CTgMC3dsnITsWlbCFU281thsrlpC8yYdxlmy8Z2E,13617
119
+ brainstate/util/others.py,sha256=8GtJvB-OJA4Jh-SmZ9nP2joIug_r_ElBJmuNq13yK8E,15867
120
+ brainstate/util/pretty_pytree.py,sha256=Xc4CpbET0swi1tkzzPUEbHFjTxeS96NpTA9mx2l8AaA,32659
121
+ brainstate/util/pretty_pytree_test.py,sha256=SIBnUegbyZL-KyBIusmfG41ZCq1rOKp6VeJnP3jORe4,5755
122
+ brainstate/util/pretty_repr.py,sha256=DVDwNRfXFrYe_j2HPJ9-1rur5tJ_YXMduImUOwjf7cU,10199
123
+ brainstate/util/pretty_table.py,sha256=qiEA3lSd9l4-Evia2KSu_KSw-F8p2BmyB7XepLpDVlk,105163
124
+ brainstate/util/scaling.py,sha256=Y4kYCr3dDmn808VUCY3i_L1XFLrUSOkRJykJukohCBA,7266
125
+ brainstate/util/struct.py,sha256=RBvB_LZyAWt6-3SHJnUfS3pE7D4pcEdrYEPbjw_fGfU,17005
126
+ brainstate-0.1.10.dist-info/licenses/LICENSE,sha256=RJ40fox7u2in2H8wvIS5DsPGlNHaA7JI024thFUlaZE,11348
127
+ brainstate-0.1.10.dist-info/METADATA,sha256=uZXs-VisW59P3ZC1GALwziVeF-XctLimfSfEbsIvvvI,3626
128
+ brainstate-0.1.10.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
129
+ brainstate-0.1.10.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
130
+ brainstate-0.1.10.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (75.1.0)
2
+ Generator: setuptools (80.9.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py2-none-any
5
5
  Tag: py3-none-any