brainstate 0.1.8__py2.py3-none-any.whl → 0.1.9__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 +509 -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.9.dist-info}/METADATA +91 -99
  127. brainstate-0.1.9.dist-info/RECORD +130 -0
  128. {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info}/WHEEL +1 -1
  129. {brainstate-0.1.8.dist-info → brainstate-0.1.9.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.9.dist-info}/top_level.txt +0 -0
@@ -1,204 +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
-
17
- __all__ = [
18
- 'spike_bitwise_or',
19
- 'spike_bitwise_and',
20
- 'spike_bitwise_iand',
21
- 'spike_bitwise_not',
22
- 'spike_bitwise_xor',
23
- 'spike_bitwise_ixor',
24
- 'spike_bitwise',
25
- ]
26
-
27
-
28
- def spike_bitwise_or(x, y):
29
- """
30
- Perform a bitwise OR operation on spike tensors.
31
-
32
- This function computes the OR operation between two spike tensors.
33
- The OR operation is implemented using the formula: x + y - x * y,
34
- which is equivalent to the OR operation for binary values.
35
-
36
- Args:
37
- x (Tensor): The first input spike tensor.
38
- y (Tensor): The second input spike tensor.
39
-
40
- Returns:
41
- Tensor: The result of the bitwise OR operation applied to the input tensors.
42
- The output tensor has the same shape as the input tensors.
43
-
44
- Note:
45
- This operation assumes that the input tensors contain binary (0 or 1) values.
46
- For non-binary inputs, the behavior may not correspond to a true bitwise OR.
47
- """
48
- return x + y - x * y
49
-
50
-
51
- def spike_bitwise_and(x, y):
52
- """
53
- Perform a bitwise AND operation on spike tensors.
54
-
55
- This function computes the AND operation between two spike tensors.
56
- The AND operation is equivalent to element-wise multiplication for binary values.
57
-
58
- Args:
59
- x (Tensor): The first input spike tensor.
60
- y (Tensor): The second input spike tensor.
61
-
62
- Returns:
63
- Tensor: The result of the bitwise AND operation applied to the input tensors.
64
- The output tensor has the same shape as the input tensors.
65
-
66
- Note:
67
- This operation is implemented using element-wise multiplication (x * y),
68
- which is equivalent to the AND operation for binary values.
69
- """
70
- return x * y
71
-
72
-
73
- def spike_bitwise_iand(x, y):
74
- """
75
- Perform a bitwise IAND (Inverse AND) operation on spike tensors.
76
-
77
- This function computes the Inverse AND (IAND) operation between two spike tensors.
78
- IAND is defined as (NOT x) AND y.
79
-
80
- Args:
81
- x (Tensor): The first input spike tensor.
82
- y (Tensor): The second input spike tensor.
83
-
84
- Returns:
85
- Tensor: The result of the bitwise IAND operation applied to the input tensors.
86
- The output tensor has the same shape as the input tensors.
87
-
88
- Note:
89
- This operation is implemented using the formula: (1 - x) * y,
90
- which is equivalent to the IAND operation for binary values.
91
- """
92
- return (1 - x) * y
93
-
94
-
95
- def spike_bitwise_not(x):
96
- """
97
- Perform a bitwise NOT operation on spike tensors.
98
-
99
- This function computes the NOT operation on a spike tensor.
100
- The NOT operation inverts the binary values in the tensor.
101
-
102
- Args:
103
- x (Tensor): The input spike tensor.
104
-
105
- Returns:
106
- Tensor: The result of the bitwise NOT operation applied to the input tensor.
107
- The output tensor has the same shape as the input tensor.
108
-
109
- Note:
110
- This operation is implemented using the formula: 1 - x,
111
- which is equivalent to the NOT operation for binary values.
112
- """
113
- return 1 - x
114
-
115
-
116
- def spike_bitwise_xor(x, y):
117
- """
118
- Perform a bitwise XOR operation on spike tensors.
119
-
120
- This function computes the XOR operation between two spike tensors.
121
- XOR is defined as (x OR y) AND NOT (x AND y).
122
-
123
- Args:
124
- x (Tensor): The first input spike tensor.
125
- y (Tensor): The second input spike tensor.
126
-
127
- Returns:
128
- Tensor: The result of the bitwise XOR operation applied to the input tensors.
129
- The output tensor has the same shape as the input tensors.
130
-
131
- Note:
132
- This operation is implemented using the formula: x + y - 2 * x * y,
133
- which is equivalent to the XOR operation for binary values.
134
- """
135
- return x + y - 2 * x * y
136
-
137
-
138
- def spike_bitwise_ixor(x, y):
139
- """
140
- Perform a bitwise IXOR (Inverse XOR) operation on spike tensors.
141
-
142
- This function computes the Inverse XOR (IXOR) operation between two spike tensors.
143
- IXOR is defined as (x AND NOT y) OR (NOT x AND y).
144
-
145
- Args:
146
- x (Tensor): The first input spike tensor.
147
- y (Tensor): The second input spike tensor.
148
-
149
- Returns:
150
- Tensor: The result of the bitwise IXOR operation applied to the input tensors.
151
- The output tensor has the same shape as the input tensors.
152
-
153
- Note:
154
- This operation is implemented using the formula: x * (1 - y) + (1 - x) * y,
155
- which is equivalent to the IXOR operation for binary values.
156
- """
157
- return x * (1 - y) + (1 - x) * y
158
-
159
-
160
- def spike_bitwise(x, y, op: str):
161
- r"""
162
- Perform bitwise operations on spike tensors.
163
-
164
- This function applies various bitwise operations on spike tensors based on the specified operation.
165
- It supports 'or', 'and', 'iand', 'xor', and 'ixor' operations.
166
-
167
- Args:
168
- x (Tensor): The first input spike tensor.
169
- y (Tensor): The second input spike tensor.
170
- op (str): A string indicating the bitwise operation to perform.
171
- Supported operations are 'or', 'and', 'iand', 'xor', and 'ixor'.
172
-
173
- Returns:
174
- Tensor: The result of the bitwise operation applied to the input tensors.
175
-
176
- Raises:
177
- NotImplementedError: If an unsupported bitwise operation is specified.
178
-
179
- Note:
180
- The function uses the following mathematical expressions for different operations:
181
-
182
- .. math::
183
-
184
- \begin{array}{ccc}
185
- \hline \text { Mode } & \text { Expression for } \mathrm{g}(\mathrm{x}, \mathrm{y}) & \text { Code for } \mathrm{g}(\mathrm{x}, \mathrm{y}) \\
186
- \hline \text { ADD } & x+y & x+y \\
187
- \text { AND } & x \cap y & x \cdot y \\
188
- \text { IAND } & (\neg x) \cap y & (1-x) \cdot y \\
189
- \text { OR } & x \cup y & (x+y)-(x \cdot y) \\
190
- \hline
191
- \end{array}
192
- """
193
- if op == 'or':
194
- return spike_bitwise_or(x, y)
195
- elif op == 'and':
196
- return spike_bitwise_and(x, y)
197
- elif op == 'iand':
198
- return spike_bitwise_iand(x, y)
199
- elif op == 'xor':
200
- return spike_bitwise_xor(x, y)
201
- elif op == 'ixor':
202
- return spike_bitwise_ixor(x, y)
203
- else:
204
- raise NotImplementedError(f"Unsupported bitwise operation: {op}.")
@@ -1,132 +0,0 @@
1
- brainstate/__init__.py,sha256=bM6vJtv8DWVcxOUQ0GPpkCzmRkdTREdTutkSVv0OziQ,1496
2
- brainstate/_compatible_import.py,sha256=LUSZlA0APWozxM8Kf9pZrM2YbwY7X3jzVVHZInaBL7Y,4630
3
- brainstate/_state.py,sha256=PLzoYx13jIgnzyBxnktLTVPCFl6seG__aNIHS9A5Nms,60770
4
- brainstate/_state_test.py,sha256=b6uvZdVRyC4n6-fYzmHNry1b-gJ6zE_kRSxGinqiHaw,1638
5
- brainstate/_utils.py,sha256=j-b239RHfC5BnvhGbSExQpdY21LrMEyWMSHBdNGThOI,1657
6
- brainstate/environ.py,sha256=vxwGyYm4lf9Ym5iD6YHsYy4bTwmeD8yHIuZsqqGfP4s,17793
7
- brainstate/environ_test.py,sha256=QD6sPCKNtqemVCGwkdImjMazatrvvLr6YeAVcfUnVVY,2045
8
- brainstate/mixin.py,sha256=eOOODmESCYITJbvz6dnQLyu8u2m0YrTaEvZsepdAUto,10991
9
- brainstate/mixin_test.py,sha256=Oq_0fwC9vpXDN4t4dTBhWzLdFDNlcYsrcip14F1yECI,3079
10
- brainstate/surrogate.py,sha256=zCNrJG-UMuvH7Le7iBuolDc-Oy8UY-xvw77tNhkGvK8,53861
11
- brainstate/transform.py,sha256=OvshYpPnp3YXPG6riy15Ve7jcX8t0aaATl1xZnsFeic,858
12
- brainstate/typing.py,sha256=cIlcgaWbvsFyEb5GgwZD7NHJh9qAp8l-N-rYN5W9FsU,10519
13
- brainstate/augment/__init__.py,sha256=Q9-JIwQ1FNn8VLS1MA9MrSylbvUjWSw98whrI3NIuKo,1229
14
- brainstate/augment/_autograd.py,sha256=XG0BgWrZ7LPelt2nRBKhpiMqz9jNS8NsK7x_I3DJINs,30884
15
- brainstate/augment/_autograd_test.py,sha256=1gR_38_i3pfH4S744G9x4KfGGtlyY0MBt43R_fndd9A,45396
16
- brainstate/augment/_eval_shape.py,sha256=jyzFzvZnUkbky7lxuchyFNGn0oJP1u10ZH5p79tAYMI,3868
17
- brainstate/augment/_eval_shape_test.py,sha256=WXrmZKmnykmYPveRfZbrDF0sFm2_PTr982yl4JM7Ebg,1390
18
- brainstate/augment/_mapping.py,sha256=R9CYp8JrdWpQcQE-XGr7POXyHqIEAYPxZfWqHF-wc70,43815
19
- brainstate/augment/_mapping_test.py,sha256=tEyXioA4_v4WGRFrpgUInyKyBjAVNLonWdHKVacmphU,21946
20
- brainstate/augment/_random.py,sha256=bkngsIk6Wwi3e46I7YSbBjLCGAz0Q3WuadUH4mqTjbY,5348
21
- brainstate/compile/__init__.py,sha256=fQtG316MLkeeu1Ssp54Kghw1PwbGK5gNq9yRVJu0wjA,1474
22
- brainstate/compile/_ad_checkpoint.py,sha256=HM6L90HU0N84S30uJpX8wqTO0ZcDnctqmwf8qFlkDYo,9358
23
- brainstate/compile/_ad_checkpoint_test.py,sha256=XGqC8kjwKd64RPgBWXHAE3t8NdIXVUqmGDfrg5HqoVw,1743
24
- brainstate/compile/_conditions.py,sha256=Pkvkc1Im7_ythKa9nwrUCNxYwkZri78I91MKbo1lEX8,10183
25
- brainstate/compile/_conditions_test.py,sha256=bh8_D49Fl-s_xFYkO4joZJPgFGhw_vd6Wo8QB1HDKwE,8640
26
- brainstate/compile/_error_if.py,sha256=l8flahWEui9-9IavtZQjyutIeR8DU0jieahRvAXUxuw,2651
27
- brainstate/compile/_error_if_test.py,sha256=PjTIQOflegqnqJb9kO5YqwjW1OfrBmAtWlY_xn8rtDY,1897
28
- brainstate/compile/_jit.py,sha256=rpM0P2t7pKyE2ak-J-TIHL4LH9uu-IsuXnklWoE1NBA,14496
29
- brainstate/compile/_jit_test.py,sha256=zD7kck9SQJGmUDolh9P4luKwQ21fBGje1Z4STTEXIuA,4135
30
- brainstate/compile/_loop_collect_return.py,sha256=qOYGoD2TMZd2Px6y241GWNwSjOLyaKlvzOwJWIPAbxg,23341
31
- brainstate/compile/_loop_collect_return_test.py,sha256=pEJdcOthEM17q5kXYhgR6JfzzqibijD_O1VPXVe_Ml4,1808
32
- brainstate/compile/_loop_no_collection.py,sha256=tPkSxee41VexWEALpN1uuT78BDrX3uT1FHv8WZtov4c,7549
33
- brainstate/compile/_loop_no_collection_test.py,sha256=ivavF59xep_g9bV1SSdXd5E1f6nhc7EUBfcbgHpxbfg,1419
34
- brainstate/compile/_make_jaxpr.py,sha256=BOJa3H1Fa0Esh3Al3CDaS351w45vk8xPDc9R_nk8jh8,37367
35
- brainstate/compile/_make_jaxpr_test.py,sha256=_aRwYt932dXlcntFNYlm5asRRE97oGp4F8hrgNO63SY,5164
36
- brainstate/compile/_progress_bar.py,sha256=AhAyyI_ckzgaj0PSj1ep1hq8rGzQLSyC2aVCYaT-e-o,7502
37
- brainstate/compile/_unvmap.py,sha256=9S42MeTmFJa8nfBI_AjEfrAdUsDmM3KFat59O8zXIEw,4120
38
- brainstate/compile/_util.py,sha256=QD7lvS4Zb3P2HPNIm6mG8Rl_Lk2tQj0znOCPcH95XnI,6275
39
- brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
40
- brainstate/functional/_activations.py,sha256=3ym1oDdftVxYlHIXWdS7XONiDFaDklqokIioi4qrsHo,21688
41
- brainstate/functional/_activations_test.py,sha256=3YR43Jyls2wWYKI-phsf6b1ZdgZa578ob1TfkitroMI,13559
42
- brainstate/functional/_normalization.py,sha256=_F5u9iJm8KJfxlnmV_LwJXtrAZOJj357aXc_sfGe4CQ,2567
43
- brainstate/functional/_others.py,sha256=iAqvLCmXQz6C80EjE0-5bwXCBKHoKqsggneiOAXNmQo,1699
44
- brainstate/functional/_spikes.py,sha256=YJ4pa5LlywximSE6ZAAs9g2k-8cO95mwxdD29HhT-oM,7051
45
- brainstate/graph/__init__.py,sha256=noo4TjBg6iEhjjwk0sAGUhR7Ge-z8Vnc2rLYUvnqttw,1295
46
- brainstate/graph/_graph_node.py,sha256=kl0CDKIcbXd9YzrZIKDhDeT8bboaoR5_OPwtL90eUEo,6835
47
- brainstate/graph/_graph_node_test.py,sha256=g6nrCCI-awfxhZLnjmYhfQvynumWJjz7fPxgIF-VSr4,2832
48
- brainstate/graph/_graph_operation.py,sha256=nIOW0GsA3Yeh8X_ijw5wx4hQgY2sh4knVMt5ouG2Sn0,63975
49
- brainstate/graph/_graph_operation_test.py,sha256=xgHoHXAafEhOqISDKLqYxb-wmMrUpv2PLtH5OLZausI,19729
50
- brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1097
51
- brainstate/init/_base.py,sha256=A5VIJV_xPyCD4Lo4RXe2WPlWvADAy_UPhexuwzp-6EI,1633
52
- brainstate/init/_generic.py,sha256=o60EPJZe7laJgPsgrnorl3Km_Lc_Lr4qQQKmu6g5G4o,7992
53
- brainstate/init/_random_inits.py,sha256=GYovk2h2Lc1jY5ByW4R8OoZ71jbaE3BYGTfDw4-a7yo,18209
54
- brainstate/init/_random_inits_test.py,sha256=c1pUdWXQxL-seEvjO_8fBkaDL6T63b6ALNfQrNuuBbg,5287
55
- brainstate/init/_regular_inits.py,sha256=tcZDTXR8jiRqERuJleg_9JRGGzJj-V3LhTMcr_gt_k0,3149
56
- brainstate/init/_regular_inits_test.py,sha256=bvv0AOfLOEP0BIQIBLztKw3EPyEp7n2fHW8PUrmWuHk,1820
57
- brainstate/nn/__init__.py,sha256=nDIBDlzHaBy-vk4cb7FiPO6zoNe2tI5qvHIK8O7yrOU,3721
58
- brainstate/nn/_collective_ops.py,sha256=Mmg-0ImuGvpv2qggu5wfBrA0MLBaq96D0P5DCZ3yolY,17418
59
- brainstate/nn/_collective_ops_test.py,sha256=bwq0DApcsk0_2xpxMl0_e2cGKT63g5rSngpigCm07ps,1409
60
- brainstate/nn/_common.py,sha256=qHAOID_eeKiPUXk_ION65sbYXyF-ddH5w5BayvH8Thg,6431
61
- brainstate/nn/_conv.py,sha256=Zk-yj34n6CkjntcM9xpMGLTxKNfWdIWsTsoGbtdL0yU,18448
62
- brainstate/nn/_conv_test.py,sha256=2lcUTG7twkyhuyKwuBux-NgU8NU_W4Cp1-G8EyDJ_uk,8862
63
- brainstate/nn/_delay.py,sha256=_FFKQ09-NHLyYPcfuvoHzpL2zc9Twu_p3k33VDbgxKo,19523
64
- brainstate/nn/_delay_test.py,sha256=O5eY-LTuzxLi6hjsC-lC12mhYe8xL36cWzvEwnG7h4g,8187
65
- brainstate/nn/_dropout.py,sha256=Dq3hQrOBT6gODlDbcoag6zLGXTU_p_2MhnmVsV57Hds,17783
66
- brainstate/nn/_dropout_test.py,sha256=L46PvC2OA7EnS4MsRhh_YnvKheHYNafOsKM8uzux_zo,4446
67
- brainstate/nn/_dynamics.py,sha256=QVCQqdglR1p-vMFzXvvEWfK1ltg4PJhUxSEEbbFP95s,48236
68
- brainstate/nn/_dynamics_test.py,sha256=w7AV57LdhbBNYprdFpKq8MFSCbXKVkGgp_NbL3ANX3I,2769
69
- brainstate/nn/_elementwise.py,sha256=_9O4S-gBZiTmsx5lJ3WdQVLcACBmXTwPTY8pnUoK7KY,33460
70
- brainstate/nn/_elementwise_test.py,sha256=_dd9eX2ZJ7p24ahuoapCaRTZ0g1boufXMyqHFx1d4WY,5688
71
- brainstate/nn/_embedding.py,sha256=SaAJbgXmuJ8XlCOX9ob4yvmgh9Fk627wMguRzJMJ1H8,2138
72
- brainstate/nn/_exp_euler.py,sha256=WTpZm-XQmsdMLNazY7wIu8eeO6pK0kRzt2lJnhEgMIk,3293
73
- brainstate/nn/_exp_euler_test.py,sha256=XD--qMbGHrHa3WtcPMmJKk59giDcEhSqZuBOmTNYUr8,1227
74
- brainstate/nn/_fixedprob.py,sha256=KGXohiU0wZnFIQDuwiRUTFsbsr8R0p8zgi5UZDuv1Bk,10004
75
- brainstate/nn/_fixedprob_test.py,sha256=qbRBh-MpMtEOsg492gFu2w9-FOP9z_bXapm-Q0gLLYM,3929
76
- brainstate/nn/_inputs.py,sha256=hMPkx9qDBpJWPshZXLF4H1QiYK1-46wntHUIlG7cT7c,20603
77
- brainstate/nn/_linear.py,sha256=FnPxATdT66DecjTW0tUTyL6clQmN_cG8kPC3qDWUE6A,14500
78
- brainstate/nn/_linear_mv.py,sha256=DyJ0OOHcvuJrBvBhV0CbmVgw-6BCg_ms-Rso9eB5e1E,2635
79
- brainstate/nn/_linear_mv_test.py,sha256=ZCM1Zy6mImQfCfdZOGnTwkiLLPXK5yalv1Ts9sWZuPA,3864
80
- brainstate/nn/_linear_test.py,sha256=eIS-VCR3QmXB_byO1Uexg65Pv48CBRUA_Je-UGrFVTY,2925
81
- brainstate/nn/_ltp.py,sha256=_najNUyfaFYcOlUTm7ThJopInbos3kwJyrm-QUfI-hc,861
82
- brainstate/nn/_module.py,sha256=jlFaoltT2F8a7cxsyu6fXp7mfIkEsk-JD3G1Xh3Ay8I,12783
83
- brainstate/nn/_module_test.py,sha256=v4ZHwjvrEaqHzCgyR8MKIB-iNhdtaCLOc55EORXTjPg,1451
84
- brainstate/nn/_neuron.py,sha256=2walTScvL034LS53pArDASXz6z26SSPbmCvchWWjkUU,27441
85
- brainstate/nn/_neuron_test.py,sha256=QF8pixUqA5Oj7MrNi2NR8VAnfGpAvNpwV2mBc3e_pTY,6393
86
- brainstate/nn/_normalizations.py,sha256=YSC1W7JaexoZ8tKcy5B-dm-_x8GuPzS_6XBfkaKpdXM,37464
87
- brainstate/nn/_normalizations_test.py,sha256=56YfszPQW9y3VoX90-seyLeWDX8akYQTRLNBvrobmS0,2498
88
- brainstate/nn/_poolings.py,sha256=yKG9BLnvDvZb7HMrAt4Y1sLCjFQ8UmWfSSQmICAPWag,47137
89
- brainstate/nn/_poolings_test.py,sha256=qje9PVWvPGiYOv6UlTEWfpqqjpv4Xop5rO0ATcgcF0w,7497
90
- brainstate/nn/_projection.py,sha256=_TW-YmfPqUDkekMMK7tXrdLbjT4Ou2sUAecTNNn7FVM,17498
91
- brainstate/nn/_rate_rnns.py,sha256=OQHMyq9kKd2q44dOHLuTqHKTGuraPMqXVutFP_LYIyU,20676
92
- brainstate/nn/_rate_rnns_test.py,sha256=__hhx7e6LX_1mDLLQyIi4TNCaFAWnOVSTIgwHNjzf2g,2444
93
- brainstate/nn/_readout.py,sha256=OJjSba5Wr7dtUXqYhAv1D7BUGOI-lAmg6urxPBrZe3c,7116
94
- brainstate/nn/_readout_test.py,sha256=L2T0-SkiACxkY_I5Pbnbmy0Zw3tbpV3l5xVzAw42f2g,2136
95
- brainstate/nn/_stp.py,sha256=-ahDEqSp8bQsU_nUK4jks8fjMYKgIbO0v7zpyGVuXtA,8645
96
- brainstate/nn/_synapse.py,sha256=I7npO9rg6bvzCiCNLiTqEOxUjGaDVlkOgmkT9M1V4xY,19860
97
- brainstate/nn/_synapse_test.py,sha256=xmCWFxZUIM2YtmW5otKnADGCCK__4JpXmSYcZ3wzlQM,4994
98
- brainstate/nn/_synaptic_projection.py,sha256=UFgzsMB1VZ9ieumwmaTIC7irLrZ4pmwfiuOYomkwXG4,17917
99
- brainstate/nn/_synouts.py,sha256=jWQP1-qXFpdYgyUSJNFD7_bk4_-67ok36br-OzbcSXY,4524
100
- brainstate/nn/_synouts_test.py,sha256=sfjotlS--4hT22Vb5RfmpfmXABa8z-VdxZYUSocMmlU,2244
101
- brainstate/nn/_utils.py,sha256=nmp6MRQoKJhfxkbbnZJEyhyMQ4DZh8FNN_JQsl55kHE,3130
102
- brainstate/nn/metrics.py,sha256=kFhUZRowtk4IoBIOPbENO4KezA6ZW_AxLAVknKFiL-s,14728
103
- brainstate/optim/__init__.py,sha256=7Ao0LCtDNAoxSRSXiLLKnd1_4mR2GSExizpN38il-Fo,1195
104
- brainstate/optim/_base.py,sha256=P37k8w46iQZZZnFLa5OF83Sb8DLQfYMez9ZRObBqVsE,1835
105
- brainstate/optim/_lr_scheduler.py,sha256=0t7kl35MLmc0UcSu587nI28cI3SohYhy_19Al-5dNTM,15289
106
- brainstate/optim/_lr_scheduler_test.py,sha256=gzzd_rzgtDWzF-93mujg043jv2wO9IceXGKiIYw5_ko,1790
107
- brainstate/optim/_optax_optimizer.py,sha256=eNaHge-9URe82WvIGDGrWrAr0MAkYWe4eCsTwijJlbU,5337
108
- brainstate/optim/_optax_optimizer_test.py,sha256=GVcszbNxBZ4z54bS5r-zDzP8gQZl7s5IsEZEd6G2gLo,1775
109
- brainstate/optim/_sgd_optimizer.py,sha256=Fn4CdGaDfV4qc_4gacCcGSjit5-ZrsBbQP-KB0Ocdac,46110
110
- brainstate/random/__init__.py,sha256=c5q-RC3grRIjx-HBb2IhKZpi_xzbFmUUxzRAzqfREic,1045
111
- brainstate/random/_rand_funs.py,sha256=0OY5kqEVrA2im-uKW2Rown0iIy7F0u54hGXHKxAP-go,137905
112
- brainstate/random/_rand_funs_test.py,sha256=1K6bT6hhOPhomE-AMyPpT6MtG6ewANRSb5bJSU-ysYc,20690
113
- brainstate/random/_rand_seed.py,sha256=ROpPgopGyL2ITNWJAM25NYjTyBizUoC4LzPUX-JpTpk,5998
114
- brainstate/random/_rand_seed_test.py,sha256=waXXfch57X1XE1zDnCRokT6ziZOK0g-lYE80o6epDYM,1536
115
- brainstate/random/_rand_state.py,sha256=hQ_govRxfIgsmtFe_V2B-jftqiMWRWrd9wkviPFOziY,55523
116
- brainstate/random/_random_for_unit.py,sha256=kGp4EUX19MXJ9Govoivbg8N0bddqOldKEI2h_TbdONY,2057
117
- brainstate/util/__init__.py,sha256=6efwr63osmqviNU_6_Nufag19PwxRDFvDAZrq6sH5yo,1555
118
- brainstate/util/caller.py,sha256=RBRwu1HT_Lww7EE3WTjzrwwKqVWg7dn_JXKmO9ojM38,2752
119
- brainstate/util/error.py,sha256=GBLaU_glQlgdrjBYIMiO3Dy5N_ST6-QNsgAOa9kjUjw,1757
120
- brainstate/util/filter.py,sha256=skuq51y0P5-N611fQ-zB2QyaJIFqN_WbROzZHv0RgOc,14086
121
- brainstate/util/others.py,sha256=vGo4uW0vd7wL5DYtxKWqXhoNmI-O_p26HsucoMnhdVA,16407
122
- brainstate/util/pretty_pytree.py,sha256=ROrOO03fxq2NvsZHlz1XG7B4NhAjHXavPjh179GUwG4,33604
123
- brainstate/util/pretty_pytree_test.py,sha256=64x2c9teehF_UoozCuuJ9WQj_287cTOTV1x58jxOTv8,5914
124
- brainstate/util/pretty_repr.py,sha256=7Xp7IFNUeP7cGlpvwwJyBslbQVnXEqC1I6neV1Jx1S8,10527
125
- brainstate/util/pretty_table.py,sha256=uJVaamFGQ4nKP8TkEGPWXHpzjMecDo2q1Ah6XtRjdPY,108117
126
- brainstate/util/scaling.py,sha256=U6DM-afPrLejiGqo1Nla7z4YbTBVicctsBEweurr_mk,7524
127
- brainstate/util/struct.py,sha256=2Y_wuDFQ6ldl_H4_w0IjzAtkbHooVgdsVbnT7Z6_Efc,17528
128
- brainstate-0.1.8.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
129
- brainstate-0.1.8.dist-info/METADATA,sha256=85Nudry-TozOxctIhW8zx7DUgOxPVmML3XCq7YqYDFs,4122
130
- brainstate-0.1.8.dist-info/WHEEL,sha256=AHX6tWk3qWuce7vKLrj7lnulVHEdWoltgauo8bgCXgU,109
131
- brainstate-0.1.8.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
132
- brainstate-0.1.8.dist-info/RECORD,,