brainstate 0.2.1__py2.py3-none-any.whl → 0.2.2__py2.py3-none-any.whl

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