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.
- brainstate/__init__.py +58 -51
- brainstate/_compatible_import.py +148 -148
- brainstate/_state.py +1605 -1663
- brainstate/_state_test.py +52 -52
- brainstate/_utils.py +47 -47
- brainstate/augment/__init__.py +30 -30
- brainstate/augment/_autograd.py +778 -778
- brainstate/augment/_autograd_test.py +1289 -1289
- brainstate/augment/_eval_shape.py +99 -99
- brainstate/augment/_eval_shape_test.py +38 -38
- brainstate/augment/_mapping.py +1060 -1060
- brainstate/augment/_mapping_test.py +597 -597
- brainstate/augment/_random.py +151 -151
- brainstate/compile/__init__.py +38 -38
- brainstate/compile/_ad_checkpoint.py +204 -204
- brainstate/compile/_ad_checkpoint_test.py +49 -49
- brainstate/compile/_conditions.py +256 -256
- brainstate/compile/_conditions_test.py +220 -220
- brainstate/compile/_error_if.py +92 -92
- brainstate/compile/_error_if_test.py +52 -52
- brainstate/compile/_jit.py +346 -346
- brainstate/compile/_jit_test.py +143 -143
- brainstate/compile/_loop_collect_return.py +536 -536
- brainstate/compile/_loop_collect_return_test.py +58 -58
- brainstate/compile/_loop_no_collection.py +184 -184
- brainstate/compile/_loop_no_collection_test.py +50 -50
- brainstate/compile/_make_jaxpr.py +888 -888
- brainstate/compile/_make_jaxpr_test.py +156 -156
- brainstate/compile/_progress_bar.py +202 -202
- brainstate/compile/_unvmap.py +159 -159
- brainstate/compile/_util.py +147 -147
- brainstate/environ.py +563 -563
- brainstate/environ_test.py +62 -62
- brainstate/functional/__init__.py +27 -26
- brainstate/graph/__init__.py +29 -29
- brainstate/graph/_graph_node.py +244 -244
- brainstate/graph/_graph_node_test.py +73 -73
- brainstate/graph/_graph_operation.py +1738 -1738
- brainstate/graph/_graph_operation_test.py +563 -563
- brainstate/init/__init__.py +26 -26
- brainstate/init/_base.py +52 -52
- brainstate/init/_generic.py +244 -244
- brainstate/init/_random_inits.py +553 -553
- brainstate/init/_random_inits_test.py +149 -149
- brainstate/init/_regular_inits.py +105 -105
- brainstate/init/_regular_inits_test.py +50 -50
- brainstate/mixin.py +365 -363
- brainstate/mixin_test.py +77 -73
- brainstate/nn/__init__.py +135 -131
- brainstate/{functional → nn}/_activations.py +808 -813
- brainstate/{functional → nn}/_activations_test.py +331 -331
- brainstate/nn/_collective_ops.py +514 -514
- brainstate/nn/_collective_ops_test.py +43 -43
- brainstate/nn/_common.py +178 -178
- brainstate/nn/_conv.py +501 -501
- brainstate/nn/_conv_test.py +238 -238
- brainstate/nn/_delay.py +509 -502
- brainstate/nn/_delay_test.py +238 -184
- brainstate/nn/_dropout.py +426 -426
- brainstate/nn/_dropout_test.py +100 -100
- brainstate/nn/_dynamics.py +1343 -1343
- brainstate/nn/_dynamics_test.py +78 -78
- brainstate/nn/_elementwise.py +1119 -1119
- brainstate/nn/_elementwise_test.py +169 -169
- brainstate/nn/_embedding.py +58 -58
- brainstate/nn/_exp_euler.py +92 -92
- brainstate/nn/_exp_euler_test.py +35 -35
- brainstate/nn/_fixedprob.py +239 -239
- brainstate/nn/_fixedprob_test.py +114 -114
- brainstate/nn/_inputs.py +608 -608
- brainstate/nn/_linear.py +424 -424
- brainstate/nn/_linear_mv.py +83 -83
- brainstate/nn/_linear_mv_test.py +120 -120
- brainstate/nn/_linear_test.py +107 -107
- brainstate/nn/_ltp.py +28 -28
- brainstate/nn/_module.py +377 -377
- brainstate/nn/_module_test.py +40 -40
- brainstate/nn/_neuron.py +705 -705
- brainstate/nn/_neuron_test.py +161 -161
- brainstate/nn/_normalizations.py +975 -918
- brainstate/nn/_normalizations_test.py +73 -73
- brainstate/{functional → nn}/_others.py +46 -46
- brainstate/nn/_poolings.py +1177 -1177
- brainstate/nn/_poolings_test.py +217 -217
- brainstate/nn/_projection.py +486 -486
- brainstate/nn/_rate_rnns.py +554 -554
- brainstate/nn/_rate_rnns_test.py +63 -63
- brainstate/nn/_readout.py +209 -209
- brainstate/nn/_readout_test.py +53 -53
- brainstate/nn/_stp.py +236 -236
- brainstate/nn/_synapse.py +505 -505
- brainstate/nn/_synapse_test.py +131 -131
- brainstate/nn/_synaptic_projection.py +423 -423
- brainstate/nn/_synouts.py +162 -162
- brainstate/nn/_synouts_test.py +57 -57
- brainstate/nn/_utils.py +89 -89
- brainstate/nn/metrics.py +388 -388
- brainstate/optim/__init__.py +38 -38
- brainstate/optim/_base.py +64 -64
- brainstate/optim/_lr_scheduler.py +448 -448
- brainstate/optim/_lr_scheduler_test.py +50 -50
- brainstate/optim/_optax_optimizer.py +152 -152
- brainstate/optim/_optax_optimizer_test.py +53 -53
- brainstate/optim/_sgd_optimizer.py +1104 -1104
- brainstate/random/__init__.py +24 -24
- brainstate/random/_rand_funs.py +3616 -3616
- brainstate/random/_rand_funs_test.py +567 -567
- brainstate/random/_rand_seed.py +210 -210
- brainstate/random/_rand_seed_test.py +48 -48
- brainstate/random/_rand_state.py +1409 -1409
- brainstate/random/_random_for_unit.py +52 -52
- brainstate/surrogate.py +1957 -1957
- brainstate/transform.py +23 -23
- brainstate/typing.py +304 -304
- brainstate/util/__init__.py +50 -50
- brainstate/util/caller.py +98 -98
- brainstate/util/error.py +55 -55
- brainstate/util/filter.py +469 -469
- brainstate/util/others.py +540 -540
- brainstate/util/pretty_pytree.py +945 -945
- brainstate/util/pretty_pytree_test.py +159 -159
- brainstate/util/pretty_repr.py +328 -328
- brainstate/util/pretty_table.py +2954 -2954
- brainstate/util/scaling.py +258 -258
- brainstate/util/struct.py +523 -523
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info}/METADATA +91 -99
- brainstate-0.1.9.dist-info/RECORD +130 -0
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info}/WHEEL +1 -1
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info/licenses}/LICENSE +202 -202
- brainstate/functional/_normalization.py +0 -81
- brainstate/functional/_spikes.py +0 -204
- brainstate-0.1.8.dist-info/RECORD +0 -132
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info}/top_level.txt +0 -0
brainstate/functional/_spikes.py
DELETED
@@ -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,,
|
File without changes
|