brainstate 0.1.9__py2.py3-none-any.whl → 0.2.0__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 +130 -19
  2. brainstate/_compatible_import.py +201 -9
  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} +10 -20
  7. brainstate/_state.py +94 -47
  8. brainstate/_state_test.py +1 -1
  9. brainstate/_utils.py +1 -1
  10. brainstate/environ.py +1279 -347
  11. brainstate/environ_test.py +1187 -26
  12. brainstate/graph/__init__.py +6 -13
  13. brainstate/graph/_node.py +240 -0
  14. brainstate/graph/_node_test.py +589 -0
  15. brainstate/graph/{_graph_operation.py → _operation.py} +632 -746
  16. brainstate/graph/_operation_test.py +1147 -0
  17. brainstate/mixin.py +1209 -141
  18. brainstate/mixin_test.py +991 -51
  19. brainstate/nn/__init__.py +74 -72
  20. brainstate/nn/_activations.py +587 -295
  21. brainstate/nn/_activations_test.py +109 -86
  22. brainstate/nn/_collective_ops.py +393 -274
  23. brainstate/nn/_collective_ops_test.py +746 -15
  24. brainstate/nn/_common.py +114 -66
  25. brainstate/nn/_common_test.py +154 -0
  26. brainstate/nn/_conv.py +1652 -143
  27. brainstate/nn/_conv_test.py +838 -227
  28. brainstate/nn/_delay.py +95 -29
  29. brainstate/nn/_delay_test.py +25 -20
  30. brainstate/nn/_dropout.py +359 -167
  31. brainstate/nn/_dropout_test.py +429 -52
  32. brainstate/nn/_dynamics.py +14 -90
  33. brainstate/nn/_dynamics_test.py +1 -12
  34. brainstate/nn/_elementwise.py +492 -313
  35. brainstate/nn/_elementwise_test.py +806 -145
  36. brainstate/nn/_embedding.py +369 -19
  37. brainstate/nn/_embedding_test.py +156 -0
  38. brainstate/nn/{_fixedprob.py → _event_fixedprob.py} +10 -16
  39. brainstate/nn/{_fixedprob_test.py → _event_fixedprob_test.py} +6 -5
  40. brainstate/nn/{_linear_mv.py → _event_linear.py} +2 -2
  41. brainstate/nn/{_linear_mv_test.py → _event_linear_test.py} +6 -5
  42. brainstate/nn/_exp_euler.py +200 -38
  43. brainstate/nn/_exp_euler_test.py +350 -8
  44. brainstate/nn/_linear.py +391 -71
  45. brainstate/nn/_linear_test.py +427 -59
  46. brainstate/nn/_metrics.py +1070 -0
  47. brainstate/nn/_metrics_test.py +611 -0
  48. brainstate/nn/_module.py +10 -3
  49. brainstate/nn/_module_test.py +1 -1
  50. brainstate/nn/_normalizations.py +688 -329
  51. brainstate/nn/_normalizations_test.py +663 -37
  52. brainstate/nn/_paddings.py +1020 -0
  53. brainstate/nn/_paddings_test.py +723 -0
  54. brainstate/nn/_poolings.py +1404 -342
  55. brainstate/nn/_poolings_test.py +828 -92
  56. brainstate/nn/{_rate_rnns.py → _rnns.py} +446 -54
  57. brainstate/nn/_rnns_test.py +593 -0
  58. brainstate/nn/_utils.py +132 -5
  59. brainstate/nn/_utils_test.py +402 -0
  60. brainstate/{init/_random_inits.py → nn/init.py} +301 -45
  61. brainstate/{init/_random_inits_test.py → nn/init_test.py} +51 -20
  62. brainstate/random/__init__.py +247 -1
  63. brainstate/random/_rand_funs.py +668 -346
  64. brainstate/random/_rand_funs_test.py +74 -1
  65. brainstate/random/_rand_seed.py +541 -76
  66. brainstate/random/_rand_seed_test.py +1 -1
  67. brainstate/random/_rand_state.py +601 -393
  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 +1 -1
  72. brainstate/{augment → transform}/_autograd.py +360 -113
  73. brainstate/{augment → transform}/_autograd_test.py +2 -2
  74. brainstate/transform/_conditions.py +316 -0
  75. brainstate/{compile → transform}/_conditions_test.py +11 -11
  76. brainstate/{compile → transform}/_error_if.py +22 -20
  77. brainstate/{compile → transform}/_error_if_test.py +1 -1
  78. brainstate/transform/_eval_shape.py +145 -0
  79. brainstate/{augment → transform}/_eval_shape_test.py +1 -1
  80. brainstate/{compile → transform}/_jit.py +99 -46
  81. brainstate/{compile → transform}/_jit_test.py +3 -3
  82. brainstate/{compile → transform}/_loop_collect_return.py +219 -80
  83. brainstate/{compile → transform}/_loop_collect_return_test.py +1 -1
  84. brainstate/{compile → transform}/_loop_no_collection.py +133 -34
  85. brainstate/{compile → transform}/_loop_no_collection_test.py +2 -2
  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 +78 -25
  91. brainstate/{augment → transform}/_random.py +65 -45
  92. brainstate/{compile → transform}/_unvmap.py +102 -5
  93. brainstate/transform/_util.py +286 -0
  94. brainstate/typing.py +594 -61
  95. brainstate/typing_test.py +780 -0
  96. brainstate/util/__init__.py +9 -32
  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} +161 -27
  102. brainstate/util/_pretty_repr_test.py +696 -0
  103. brainstate/util/filter.py +557 -81
  104. brainstate/util/filter_test.py +912 -0
  105. brainstate/util/struct.py +769 -382
  106. brainstate/util/struct_test.py +602 -0
  107. {brainstate-0.1.9.dist-info → brainstate-0.2.0.dist-info}/METADATA +34 -17
  108. brainstate-0.2.0.dist-info/RECORD +111 -0
  109. brainstate/augment/__init__.py +0 -30
  110. brainstate/augment/_eval_shape.py +0 -99
  111. brainstate/augment/_mapping.py +0 -1060
  112. brainstate/augment/_mapping_test.py +0 -597
  113. brainstate/compile/__init__.py +0 -38
  114. brainstate/compile/_ad_checkpoint.py +0 -204
  115. brainstate/compile/_conditions.py +0 -256
  116. brainstate/compile/_make_jaxpr.py +0 -888
  117. brainstate/compile/_make_jaxpr_test.py +0 -156
  118. brainstate/compile/_util.py +0 -147
  119. brainstate/functional/__init__.py +0 -27
  120. brainstate/graph/_graph_node.py +0 -244
  121. brainstate/graph/_graph_node_test.py +0 -73
  122. brainstate/graph/_graph_operation_test.py +0 -563
  123. brainstate/init/__init__.py +0 -26
  124. brainstate/init/_base.py +0 -52
  125. brainstate/init/_generic.py +0 -244
  126. brainstate/init/_regular_inits.py +0 -105
  127. brainstate/init/_regular_inits_test.py +0 -50
  128. brainstate/nn/_inputs.py +0 -608
  129. brainstate/nn/_ltp.py +0 -28
  130. brainstate/nn/_neuron.py +0 -705
  131. brainstate/nn/_neuron_test.py +0 -161
  132. brainstate/nn/_others.py +0 -46
  133. brainstate/nn/_projection.py +0 -486
  134. brainstate/nn/_rate_rnns_test.py +0 -63
  135. brainstate/nn/_readout.py +0 -209
  136. brainstate/nn/_readout_test.py +0 -53
  137. brainstate/nn/_stp.py +0 -236
  138. brainstate/nn/_synapse.py +0 -505
  139. brainstate/nn/_synapse_test.py +0 -131
  140. brainstate/nn/_synaptic_projection.py +0 -423
  141. brainstate/nn/_synouts.py +0 -162
  142. brainstate/nn/_synouts_test.py +0 -57
  143. brainstate/nn/metrics.py +0 -388
  144. brainstate/optim/__init__.py +0 -38
  145. brainstate/optim/_base.py +0 -64
  146. brainstate/optim/_lr_scheduler.py +0 -448
  147. brainstate/optim/_lr_scheduler_test.py +0 -50
  148. brainstate/optim/_optax_optimizer.py +0 -152
  149. brainstate/optim/_optax_optimizer_test.py +0 -53
  150. brainstate/optim/_sgd_optimizer.py +0 -1104
  151. brainstate/random/_random_for_unit.py +0 -52
  152. brainstate/surrogate.py +0 -1957
  153. brainstate/transform.py +0 -23
  154. brainstate/util/caller.py +0 -98
  155. brainstate/util/others.py +0 -540
  156. brainstate/util/pretty_pytree.py +0 -945
  157. brainstate/util/pretty_pytree_test.py +0 -159
  158. brainstate/util/pretty_table.py +0 -2954
  159. brainstate/util/scaling.py +0 -258
  160. brainstate-0.1.9.dist-info/RECORD +0 -130
  161. {brainstate-0.1.9.dist-info → brainstate-0.2.0.dist-info}/WHEEL +0 -0
  162. {brainstate-0.1.9.dist-info → brainstate-0.2.0.dist-info}/licenses/LICENSE +0 -0
  163. {brainstate-0.1.9.dist-info → brainstate-0.2.0.dist-info}/top_level.txt +0 -0
@@ -1,258 +0,0 @@
1
- # Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
2
- #
3
- # Licensed under the Apache License, Version 2.0 (the "License");
4
- # you may not use this file except in compliance with the License.
5
- # You may obtain a copy of the License at
6
- #
7
- # http://www.apache.org/licenses/LICENSE-2.0
8
- #
9
- # Unless required by applicable law or agreed to in writing, software
10
- # distributed under the License is distributed on an "AS IS" BASIS,
11
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
- # See the License for the specific language governing permissions and
13
- # limitations under the License.
14
- # ==============================================================================
15
-
16
- from typing import Union, Sequence
17
-
18
- __all__ = [
19
- 'MemScaling',
20
- 'IdMemScaling',
21
- ]
22
-
23
-
24
- class MemScaling(object):
25
- """
26
- The scaling object for membrane potential.
27
-
28
- The scaling object is used to transform the membrane potential range to a
29
- standard range. The scaling object can be used to transform the membrane
30
- potential to a standard range, and transform the standard range to the
31
- membrane potential.
32
-
33
- """
34
- __module__ = 'brainstate.util'
35
-
36
- def __init__(self, scale, bias):
37
- self._scale = scale
38
- self._bias = bias
39
-
40
- @classmethod
41
- def transform(
42
- cls,
43
- oring_range: Sequence[Union[float, int]],
44
- target_range: Sequence[Union[float, int]] = (0., 1.)
45
- ) -> 'MemScaling':
46
- """Transform the membrane potential range to a ``Scaling`` instance.
47
-
48
- Args:
49
- oring_range: [V_min, V_max]
50
- target_range: [scaled_V_min, scaled_V_max]
51
-
52
- Returns:
53
- The instanced scaling object.
54
- """
55
- V_min, V_max = oring_range
56
- scaled_V_min, scaled_V_max = target_range
57
- scale = (V_max - V_min) / (scaled_V_max - scaled_V_min)
58
- bias = scaled_V_min * scale - V_min
59
- return cls(scale=scale, bias=bias)
60
-
61
- def scale_offset(self, x, bias=None, scale=None):
62
- """
63
- Transform the membrane potential to the standard range.
64
-
65
- Parameters
66
- ----------
67
- x : array_like
68
- The membrane potential.
69
- bias : float, optional
70
- The bias of the scaling object. If None, the default bias will be used.
71
- scale : float, optional
72
- The scale of the scaling object. If None, the default scale will be used.
73
-
74
- Returns
75
- -------
76
- x : array_like
77
- The standard range of the membrane potential.
78
- """
79
- if bias is None:
80
- bias = self._bias
81
- if scale is None:
82
- scale = self._scale
83
- return (x + bias) / scale
84
-
85
- def scale(self, x, scale=None):
86
- """
87
- Transform the membrane potential to the standard range.
88
-
89
- Parameters
90
- ----------
91
- x : array_like
92
- The membrane potential.
93
- scale : float, optional
94
- The scale of the scaling object. If None, the default scale will be used.
95
-
96
- Returns
97
- -------
98
- x : array_like
99
- The standard range of the membrane potential.
100
- """
101
- if scale is None:
102
- scale = self._scale
103
- return x / scale
104
-
105
- def offset(self, x, bias=None):
106
- """
107
- Transform the membrane potential to the standard range.
108
-
109
- Parameters
110
- ----------
111
- x : array_like
112
- The membrane potential.
113
- bias : float, optional
114
- The bias of the scaling object. If None, the default bias will be used.
115
-
116
- Returns
117
- -------
118
- x : array_like
119
- The standard range of the membrane potential.
120
- """
121
- if bias is None:
122
- bias = self._bias
123
- return x + bias
124
-
125
- def rev_scale(self, x, scale=None):
126
- """
127
- Reversely transform the standard range to the original membrane potential.
128
-
129
- Parameters
130
- ----------
131
- x : array_like
132
- The standard range of the membrane potential.
133
- scale : float, optional
134
- The scale of the scaling object. If None, the default scale will be used.
135
-
136
- Returns
137
- -------
138
- x : array_like
139
- The original membrane potential.
140
- """
141
- if scale is None:
142
- scale = self._scale
143
- return x * scale
144
-
145
- def rev_offset(self, x, bias=None):
146
- """
147
- Reversely transform the standard range to the original membrane potential.
148
-
149
- Parameters
150
- ----------
151
- x : array_like
152
- The standard range of the membrane potential.
153
- bias : float, optional
154
- The bias of the scaling object. If None, the default bias will be used.
155
-
156
- Returns
157
- -------
158
- x : array_like
159
- The original membrane potential.
160
- """
161
- if bias is None:
162
- bias = self._bias
163
- return x - bias
164
-
165
- def rev_scale_offset(self, x, bias=None, scale=None):
166
- """
167
- Reversely transform the standard range to the original membrane potential.
168
-
169
- Parameters
170
- ----------
171
- x : array_like
172
- The standard range of the membrane potential.
173
- bias : float, optional
174
- The bias of the scaling object. If None, the default bias will be used.
175
- scale : float, optional
176
- The scale of the scaling object. If None, the default scale will be used.
177
-
178
- Returns
179
- -------
180
- x : array_like
181
- The original membrane potential.
182
- """
183
- if bias is None:
184
- bias = self._bias
185
- if scale is None:
186
- scale = self._scale
187
- return x * scale - bias
188
-
189
- def clone(self):
190
- """
191
- Clone the scaling object.
192
-
193
- Returns
194
- -------
195
- scaling : MemScaling
196
- The cloned scaling object.
197
- """
198
- return MemScaling(bias=self._bias, scale=self._scale)
199
-
200
-
201
- class IdMemScaling(MemScaling):
202
- """
203
- The identity scaling object.
204
-
205
- The identity scaling object is used to transform the membrane potential to
206
- the standard range, and reversely transform the standard range to the
207
- membrane potential.
208
-
209
- """
210
- __module__ = 'brainstate.util'
211
-
212
- def __init__(self):
213
- super().__init__(scale=1., bias=0.)
214
-
215
- def scale_offset(self, x, bias=None, scale=None):
216
- """
217
- Transform the membrane potential to the standard range.
218
- """
219
- return x
220
-
221
- def scale(self, x, scale=None):
222
- """
223
- Transform the membrane potential to the standard range.
224
- """
225
- return x
226
-
227
- def offset(self, x, bias=None):
228
- """
229
- Transform the membrane potential to the standard range.
230
- """
231
- return x
232
-
233
- def rev_scale(self, x, scale=None):
234
- """
235
- Reversely transform the standard range to the original membrane potential.
236
-
237
- """
238
- return x
239
-
240
- def rev_offset(self, x, bias=None):
241
- """
242
- Reversely transform the standard range to the original membrane potential.
243
-
244
-
245
- """
246
- return x
247
-
248
- def rev_scale_offset(self, x, bias=None, scale=None):
249
- """
250
- Reversely transform the standard range to the original membrane potential.
251
- """
252
- return x
253
-
254
- def clone(self):
255
- """
256
- Clone the scaling object.
257
- """
258
- return IdMemScaling()
@@ -1,130 +0,0 @@
1
- brainstate/__init__.py,sha256=4vCHesk-UDMgifDdr5hm7veKU8Avg94-DC0OLXrQn8g,1450
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=yRIevZvU92sYz6XTqmDLAjCXCnrHW4_FwWKRPNygkf8,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=0B2t6--9I3XOTjdzufAjEYhgZg8zwzduqmmvKA_HpBk,19590
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.9.dist-info/licenses/LICENSE,sha256=RJ40fox7u2in2H8wvIS5DsPGlNHaA7JI024thFUlaZE,11348
127
- brainstate-0.1.9.dist-info/METADATA,sha256=L-hwNoEacz-zOcfjK6SYurldwS31HZjin3KcFCeZmRo,3625
128
- brainstate-0.1.9.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
129
- brainstate-0.1.9.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
130
- brainstate-0.1.9.dist-info/RECORD,,