brainstate 0.1.2__py2.py3-none-any.whl → 0.1.3__py2.py3-none-any.whl

brainstate/nn/_synaptic_projection.py

@@ -0,0 +1,133 @@
+# Copyright 2025 BDP Ecosystem Limited. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+# -*- coding: utf-8 -*-
+
+
+from typing import Callable, Union
+
+import brainunit as u
+
+from brainstate._compatible_import import brainevent
+from brainstate.mixin import ParamDescriber, AlignPost, UpdateReturn
+from ._dynamics import Dynamics, Projection
+from ._projection import AlignPostProj, RawProj
+from ._stp import ShortTermPlasticity
+from ._synapse import Synapse
+from ._synouts import SynOut
+
+__all__ = [
+    'align_pre_projection',
+    'align_post_projection',
+]
+
+
+class align_pre_projection(Projection):
+    """
+    Represents a pre-synaptic alignment projection mechanism.
+
+    This class inherits from the `Projection` base class and is designed to
+    manage the pre-synaptic alignment process in neural network simulations.
+    It takes into account pre-synaptic dynamics, synaptic properties, delays,
+    communication functions, synaptic outputs, post-synaptic dynamics, and
+    short-term plasticity.
+
+    Attributes:
+        pre (Dynamics): The pre-synaptic dynamics object.
+        syn (Synapse): The synaptic object after pre-synaptic alignment.
+        delay (u.Quantity[u.second]): The output delay from the synapse.
+        projection (RawProj): The raw projection object handling communication,
+            output, and post-synaptic dynamics.
+        stp (ShortTermPlasticity, optional): The short-term plasticity object,
+            defaults to None.
+    """
+
+    def __init__(
+        self,
+        pre: Dynamics,
+        syn: Union[Synapse, ParamDescriber[Synapse]],
+        delay: u.Quantity[u.second] | None,
+        comm: Callable,
+        out: SynOut,
+        post: Dynamics,
+        stp: ShortTermPlasticity = None,
+    ):
+        super().__init__()
+        pre = pre
+        syn: Synapse = pre.align_pre(syn)
+        assert isinstance(syn, UpdateReturn), "Synapse must implement UpdateReturn interface"
+        # require "syn" implement the "update_return()" function
+        self.delay = syn.output_delay(delay)
+        self.projection = RawProj(comm=comm, out=out, post=post)
+        self.stp = stp
+
+    def update(self):
+        x = self.delay()
+        if self.stp is not None:
+            x = self.stp(x)
+        return self.projection(x)
+
+
+class align_post_projection(Projection):
+    """
+    Represents a post-synaptic alignment projection mechanism.
+
+    This class inherits from the `Projection` base class and is designed to
+    manage the post-synaptic alignment process in neural network simulations.
+    It takes into account spike generators, communication functions, synaptic
+    properties, synaptic outputs, post-synaptic dynamics, and short-term plasticity.
+
+    Args:
+        *spike_generator: Callable(s) that generate spike events or transform input spikes.
+        comm (Callable): Communication function for the projection.
+        syn (Union[AlignPost, ParamDescriber[AlignPost]]): The post-synaptic alignment object or its parameter describer.
+        out (Union[SynOut, ParamDescriber[SynOut]]): The synaptic output object or its parameter describer.
+        post (Dynamics): The post-synaptic dynamics object.
+        stp (ShortTermPlasticity, optional): The short-term plasticity object, defaults to None.
+
+    """
+    def __init__(
+        self,
+        *spike_generator,
+        comm: Callable,
+        syn: Union[AlignPost, ParamDescriber[AlignPost]],
+        out: Union[SynOut, ParamDescriber[SynOut]],
+        post: Dynamics,
+        stp: ShortTermPlasticity = None,
+    ):
+        super().__init__()
+        self.spike_generator = spike_generator
+        self.projection = AlignPostProj(comm=comm, syn=syn, out=out, post=post)
+        self.stp = stp
+
+    def update(self, *x):
+        for fun in self.spike_generator:
+            x = fun(*x)
+            if isinstance(x, (tuple, list)):
+                x = tuple(x)
+            else:
+                x = (x,)
+        assert len(x) == 1, "Spike generator must return a single value or a tuple/list of values"
+        x = brainevent.BinaryArray(x[0])  # Ensure input is a BinaryFloat for spike generation
+        if self.stp is not None:
+            x = brainevent.MaskedFloat(self.stp(x))  # Ensure STP output is a MaskedFloat
+        return self.projection(x)
+
+
+class align_pre_ltp(Projection):
+    pass
+
+
+class align_post_ltp(Projection):
+    pass

brainstate/nn/{_dynamics/_synouts.py → _synouts.py}

@@ -19,8 +19,8 @@ import brainunit as u
 import jax.numpy as jnp
 
 from brainstate.mixin import BindCondData
-from brainstate.nn._module import Module
 from brainstate.typing import ArrayLike
+from ._module import Module
 
 __all__ = [
     'SynOut', 'COBA', 'CUBA', 'MgBlock',
@@ -47,6 +47,9 @@ class SynOut(Module, BindCondData):
         ret = self.update(self._conductance, *args, **kwargs)
         return ret
 
+    def update(self, conductance, potential):
+        raise NotImplementedError
+
 
 class COBA(SynOut):
     r"""

{brainstate-0.1.2.dist-info → brainstate-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainstate
-Version: 0.1.2
+Version: 0.1.3
 Summary: A ``State``-based Transformation System for Program Compilation and Augmentation.
 Home-page: https://github.com/chaobrain/brainstate
 Author: BrainState Developers

{brainstate-0.1.2.dist-info → brainstate-0.1.3.dist-info}/RECORD

@@ -1,11 +1,11 @@
-brainstate/__init__.py,sha256=50znZ0wsJBdgX9Wvlv4xvsuNGp4XdgR60LTFrplmBz0,1496
-brainstate/_compatible_import.py,sha256=96CR40jbcrvaBXO0Hrlm7Cpde-4nRj-6z-P753bitFc,5001
+brainstate/__init__.py,sha256=cJBE8xL38olR6vA35aozQE1oJHTkgRFUKGUaiwsU19g,1496
+brainstate/_compatible_import.py,sha256=E0KTFLltv84mA6QuMzV-0KLjN5UdTP-sbhfolzjMVUA,5061
 brainstate/_state.py,sha256=o5Kk4bGwVz6Dfj9dlmZqdh6zUXcz6Tvc6WOjH9ajlIU,60716
 brainstate/_state_test.py,sha256=b6uvZdVRyC4n6-fYzmHNry1b-gJ6zE_kRSxGinqiHaw,1638
 brainstate/_utils.py,sha256=j-b239RHfC5BnvhGbSExQpdY21LrMEyWMSHBdNGThOI,1657
 brainstate/environ.py,sha256=VgtG0S_aR1g_1gplRWg_v2ZrcS-F6LZk35BPCBgsIIA,17660
 brainstate/environ_test.py,sha256=QD6sPCKNtqemVCGwkdImjMazatrvvLr6YeAVcfUnVVY,2045
-brainstate/mixin.py,sha256=1PPifp91eKEeMJO8cLjZPmXehPBE5tIoRhMVpvysquo,11272
+brainstate/mixin.py,sha256=gt7pPK7VvDR7OO8VjBfxVVKyGQ264cvNHYNdhLpbeX0,10803
 brainstate/mixin_test.py,sha256=Oq_0fwC9vpXDN4t4dTBhWzLdFDNlcYsrcip14F1yECI,3079
 brainstate/surrogate.py,sha256=Iu1-WWJKH5CIjXIHKwEF8ufxVYpFZCltU_6dEH9DYqk,53470
 brainstate/transform.py,sha256=OvshYpPnp3YXPG6riy15Ve7jcX8t0aaATl1xZnsFeic,858
@@ -54,53 +54,51 @@ brainstate/init/_random_inits.py,sha256=GYovk2h2Lc1jY5ByW4R8OoZ71jbaE3BYGTfDw4-a
 brainstate/init/_random_inits_test.py,sha256=c1pUdWXQxL-seEvjO_8fBkaDL6T63b6ALNfQrNuuBbg,5287
 brainstate/init/_regular_inits.py,sha256=tcZDTXR8jiRqERuJleg_9JRGGzJj-V3LhTMcr_gt_k0,3149
 brainstate/init/_regular_inits_test.py,sha256=bvv0AOfLOEP0BIQIBLztKw3EPyEp7n2fHW8PUrmWuHk,1820
-brainstate/nn/__init__.py,sha256=dUEH_S5jRTq0nY1xNkpN_4MQrmHXHGSmGp3xgGenqNc,1924
+brainstate/nn/__init__.py,sha256=NfZMwitNb9nT-UPMt4j3frIiBNHcsY1D9LWxDTNwSY0,3666
 brainstate/nn/_collective_ops.py,sha256=rWcjqaP0rW6sXdhE0fjtDi4twmq5VvhNEg8AoiQ-tDU,17390
 brainstate/nn/_collective_ops_test.py,sha256=bwq0DApcsk0_2xpxMl0_e2cGKT63g5rSngpigCm07ps,1409
 brainstate/nn/_common.py,sha256=Pt8P1LgE0qW3QnfX8CQQGH3yZ78RL7NkIqqYEw_z8xs,7096
+brainstate/nn/_conv.py,sha256=Zk-yj34n6CkjntcM9xpMGLTxKNfWdIWsTsoGbtdL0yU,18448
+brainstate/nn/_conv_test.py,sha256=2lcUTG7twkyhuyKwuBux-NgU8NU_W4Cp1-G8EyDJ_uk,8862
+brainstate/nn/_delay.py,sha256=AbTGEVrZIG9TjwvUXxLRcNVQPPiNxgPPYpFKqJ8pfUM,16957
+brainstate/nn/_dropout.py,sha256=WX9aYBG4KkPPog5qaSqujxATeb4hVQl3EorXIDmM1hg,17720
+brainstate/nn/_dropout_test.py,sha256=9i2ZW5by0S9zfVHmZKLN0WpI1bXH6h9-QLXamG6lVXA,4432
+brainstate/nn/_dynamics.py,sha256=l4Z8v2uGSiwDP-kXUcWPGvgc0UDyy7w6kPTAgi6p7is,48125
+brainstate/nn/_dynamics_test.py,sha256=w7AV57LdhbBNYprdFpKq8MFSCbXKVkGgp_NbL3ANX3I,2769
+brainstate/nn/_elementwise.py,sha256=4czeJWGQopV49iZo8DuN_WzAbXoMC1gtqaGjlON6e7c,33291
+brainstate/nn/_elementwise_test.py,sha256=_dd9eX2ZJ7p24ahuoapCaRTZ0g1boufXMyqHFx1d4WY,5688
+brainstate/nn/_embedding.py,sha256=SaAJbgXmuJ8XlCOX9ob4yvmgh9Fk627wMguRzJMJ1H8,2138
 brainstate/nn/_exp_euler.py,sha256=ndDB43PM4jsZKu_zdLTZ2-ojnuNrg55LZap23oBTtdA,3493
 brainstate/nn/_exp_euler_test.py,sha256=XD--qMbGHrHa3WtcPMmJKk59giDcEhSqZuBOmTNYUr8,1227
+brainstate/nn/_fixedprob_mv.py,sha256=8GRp4MALSIAGnIFqVZpnEAKoWTEsyeYAiIyu6fcmLLM,6606
+brainstate/nn/_fixedprob_mv_test.py,sha256=qbRBh-MpMtEOsg492gFu2w9-FOP9z_bXapm-Q0gLLYM,3929
+brainstate/nn/_inputs.py,sha256=wPOfPE4IesNoDmxZJxqR0siBlJioEX-_1IZ2cltAIpM,20605
+brainstate/nn/_linear.py,sha256=5WuhcqU-uBUC91vnwezQYMHPKmlZPDgIJ5UpffxoX1I,14472
+brainstate/nn/_linear_mv.py,sha256=vgO7Xi7WlGULzgvvbWK5zefymn-YAdAhdAQ0tUMC2Ls,2670
+brainstate/nn/_linear_mv_test.py,sha256=ZCM1Zy6mImQfCfdZOGnTwkiLLPXK5yalv1Ts9sWZuPA,3864
+brainstate/nn/_linear_test.py,sha256=eIS-VCR3QmXB_byO1Uexg65Pv48CBRUA_Je-UGrFVTY,2925
+brainstate/nn/_ltp.py,sha256=_najNUyfaFYcOlUTm7ThJopInbos3kwJyrm-QUfI-hc,861
 brainstate/nn/_module.py,sha256=ksbmO_-bZzaP8XTRAcN2oi2ZMdSuSv-XVeWc2VJncVc,12808
 brainstate/nn/_module_test.py,sha256=_RKHY8LHg6nbFHkvPAsEPVpp9rW6XfrcF8YD68u4o4w,8956
+brainstate/nn/_neuron.py,sha256=2walTScvL034LS53pArDASXz6z26SSPbmCvchWWjkUU,27441
+brainstate/nn/_neuron_test.py,sha256=QF8pixUqA5Oj7MrNi2NR8VAnfGpAvNpwV2mBc3e_pTY,6393
+brainstate/nn/_normalizations.py,sha256=4B1YVI6AZyJLHseJK9eiSeMOIq--0ZuDG58iC9mrekE,37387
+brainstate/nn/_normalizations_test.py,sha256=JTqpH265GT0eKw3hXmT0qC8ZM5NkUe98of0uYS7_5Us,2456
+brainstate/nn/_poolings.py,sha256=h5aIj_K3X2HSMjrOtkgwiCpvLxfXXEUvFFa57Cxp5bI,46969
+brainstate/nn/_poolings_test.py,sha256=qje9PVWvPGiYOv6UlTEWfpqqjpv4Xop5rO0ATcgcF0w,7497
+brainstate/nn/_projection.py,sha256=h-UinIlkmJ4ITtsW-K4TpyPP-xDH3zHb7LjTWFOu4eI,13668
+brainstate/nn/_rate_rnns.py,sha256=OQHMyq9kKd2q44dOHLuTqHKTGuraPMqXVutFP_LYIyU,20676
+brainstate/nn/_rate_rnns_test.py,sha256=__hhx7e6LX_1mDLLQyIi4TNCaFAWnOVSTIgwHNjzf2g,2444
+brainstate/nn/_readout.py,sha256=OJjSba5Wr7dtUXqYhAv1D7BUGOI-lAmg6urxPBrZe3c,7116
+brainstate/nn/_readout_test.py,sha256=L2T0-SkiACxkY_I5Pbnbmy0Zw3tbpV3l5xVzAw42f2g,2136
+brainstate/nn/_stp.py,sha256=-ahDEqSp8bQsU_nUK4jks8fjMYKgIbO0v7zpyGVuXtA,8645
+brainstate/nn/_synapse.py,sha256=5AP_UpY5LvsAVAu0yExi7pXGpeqnlZW8J8r-Gw-26AA,20239
+brainstate/nn/_synapse_test.py,sha256=xmCWFxZUIM2YtmW5otKnADGCCK__4JpXmSYcZ3wzlQM,4994
+brainstate/nn/_synaptic_projection.py,sha256=iGw5NWhpvOJ62M7EzQpD1oVcHgd_L6XkPNs0VM3ognk,5083
+brainstate/nn/_synouts.py,sha256=jWQP1-qXFpdYgyUSJNFD7_bk4_-67ok36br-OzbcSXY,4524
+brainstate/nn/_synouts_test.py,sha256=sfjotlS--4hT22Vb5RfmpfmXABa8z-VdxZYUSocMmlU,2244
 brainstate/nn/_utils.py,sha256=epfELIy1COgdS9z5be-fmbFhagNugcIHpw4ww-HlkSY,3123
 brainstate/nn/metrics.py,sha256=TXCB_yGQzamklJCI5FGOAZ5dihtY-gjYlSi2SCLC3LA,14700
-brainstate/nn/_dyn_impl/__init__.py,sha256=J1hMqNey1921tZppCjuHApB6oc-5aGhWXyOXh96CeWM,1322
-brainstate/nn/_dyn_impl/_dynamics_neuron.py,sha256=6xsGF2Oy_oD6bcTdfwqnHBtfdAJREIl5qvNUZKZRlnQ,27482
-brainstate/nn/_dyn_impl/_dynamics_neuron_test.py,sha256=QF8pixUqA5Oj7MrNi2NR8VAnfGpAvNpwV2mBc3e_pTY,6393
-brainstate/nn/_dyn_impl/_dynamics_synapse.py,sha256=b2QMEzasbXTp7k1XWNg7nRt4-WddCym0CSEErjPRIWA,27449
-brainstate/nn/_dyn_impl/_dynamics_synapse_test.py,sha256=xmCWFxZUIM2YtmW5otKnADGCCK__4JpXmSYcZ3wzlQM,4994
-brainstate/nn/_dyn_impl/_inputs.py,sha256=Fe3tgSwJMrq2HfZJq9CTgNy24Ji6tnLCqB4hkhaY_K0,20675
-brainstate/nn/_dyn_impl/_rate_rnns.py,sha256=SXlrHZhih4LD99Jg8OwIWur5UQbd6E_zotkHvkFFstc,20715
-brainstate/nn/_dyn_impl/_rate_rnns_test.py,sha256=__hhx7e6LX_1mDLLQyIi4TNCaFAWnOVSTIgwHNjzf2g,2444
-brainstate/nn/_dyn_impl/_readout.py,sha256=UzaJ6WIMpNjqHolhjHTpyCZ5E2nSDucvuV5PBW7rOPs,7151
-brainstate/nn/_dyn_impl/_readout_test.py,sha256=L2T0-SkiACxkY_I5Pbnbmy0Zw3tbpV3l5xVzAw42f2g,2136
-brainstate/nn/_dynamics/__init__.py,sha256=j1HSWu01wf5-KjSaNhBC9utVGDALOhUsFPrLPcPPDsM,1208
-brainstate/nn/_dynamics/_dynamics_base.py,sha256=stBAkb8f973vJe66p0tve-0AO_j008BCpAA96FIPnTI,43363
-brainstate/nn/_dynamics/_dynamics_base_test.py,sha256=w7AV57LdhbBNYprdFpKq8MFSCbXKVkGgp_NbL3ANX3I,2769
-brainstate/nn/_dynamics/_projection_base.py,sha256=O86VfZkd8Qd5_8CDK-4eCGS6Jkp1CKyW6J9__bQnmo0,12893
-brainstate/nn/_dynamics/_state_delay.py,sha256=kSgel1OSFB92WcfGTsIjgb-KW389ydXTcLbTUjylU-A,16867
-brainstate/nn/_dynamics/_synouts.py,sha256=Py9DM3o-JmkoqYyQM5vl89-SBqTBRdk2CUYgYHYu3No,4453
-brainstate/nn/_dynamics/_synouts_test.py,sha256=sfjotlS--4hT22Vb5RfmpfmXABa8z-VdxZYUSocMmlU,2244
-brainstate/nn/_elementwise/__init__.py,sha256=PK8oq1K_EG2941AiUyLxCWoRdWvMO3yt8ZJbw3Lkhu8,935
-brainstate/nn/_elementwise/_dropout.py,sha256=ymr87YYA5x5C_c9HzNLGPJK_ebJ22Y-KsK-gZJY_2MU,17733
-brainstate/nn/_elementwise/_dropout_test.py,sha256=9i2ZW5by0S9zfVHmZKLN0WpI1bXH6h9-QLXamG6lVXA,4432
-brainstate/nn/_elementwise/_elementwise.py,sha256=yO6BukJFLfnICPELEFp2kfutFM8Jeg7TaERXFp-aNrk,33304
-brainstate/nn/_elementwise/_elementwise_test.py,sha256=_dd9eX2ZJ7p24ahuoapCaRTZ0g1boufXMyqHFx1d4WY,5688
-brainstate/nn/_event/__init__.py,sha256=OLteeA3d1Kq8yi3XETA-rFYtYpojZRpY1J_h5MpH3rY,919
-brainstate/nn/_event/_fixedprob_mv.py,sha256=MpVCiYGiDC8GAuJWtYXKrpzP9yDtnYHolHlwyzoQ1r4,6619
-brainstate/nn/_event/_fixedprob_mv_test.py,sha256=qbRBh-MpMtEOsg492gFu2w9-FOP9z_bXapm-Q0gLLYM,3929
-brainstate/nn/_event/_linear_mv.py,sha256=KslIdse7KZ7FWm5qcOMC-COM__1rB-sUJmXvlXY11y8,2683
-brainstate/nn/_event/_linear_mv_test.py,sha256=ZCM1Zy6mImQfCfdZOGnTwkiLLPXK5yalv1Ts9sWZuPA,3864
-brainstate/nn/_interaction/__init__.py,sha256=TTY_SeNrdx4VnUSw6vdyl02OHdS9Qs15cWBp6kjsyNQ,1289
-brainstate/nn/_interaction/_conv.py,sha256=QMZR4HRAYe8Nzr3qExp0PWr6wuUp2rh5YxmPq1r1vcY,18461
-brainstate/nn/_interaction/_conv_test.py,sha256=2lcUTG7twkyhuyKwuBux-NgU8NU_W4Cp1-G8EyDJ_uk,8862
-brainstate/nn/_interaction/_embedding.py,sha256=8HC-J3RhvGGrvqA0-BBaFDk6tmlunPP6bNohjRSQzYo,2151
-brainstate/nn/_interaction/_linear.py,sha256=jk765kASgwVG1r-xjoM49Thfa1Cu32mscy7ci79bjDI,14729
-brainstate/nn/_interaction/_linear_test.py,sha256=eIS-VCR3QmXB_byO1Uexg65Pv48CBRUA_Je-UGrFVTY,2925
-brainstate/nn/_interaction/_normalizations.py,sha256=4yoU04wK21_zk6Egpwb9YN_pSWUH5Qa6hJ1bLxX7wQw,37400
-brainstate/nn/_interaction/_normalizations_test.py,sha256=JTqpH265GT0eKw3hXmT0qC8ZM5NkUe98of0uYS7_5Us,2456
-brainstate/nn/_interaction/_poolings.py,sha256=2hJDACuHv3CkO-nyyAqVZp7Sa3UFa_ROFGiYmhXWqKk,46982
-brainstate/nn/_interaction/_poolings_test.py,sha256=7-jEs_UcMNWm8q6NbrRRX3bTn14jBjGiAX0KBaROng0,7510
 brainstate/optim/__init__.py,sha256=7Ao0LCtDNAoxSRSXiLLKnd1_4mR2GSExizpN38il-Fo,1195
 brainstate/optim/_base.py,sha256=P37k8w46iQZZZnFLa5OF83Sb8DLQfYMez9ZRObBqVsE,1835
 brainstate/optim/_lr_scheduler.py,sha256=0t7kl35MLmc0UcSu587nI28cI3SohYhy_19Al-5dNTM,15289
@@ -126,8 +124,8 @@ brainstate/util/_pretty_table.py,sha256=uJVaamFGQ4nKP8TkEGPWXHpzjMecDo2q1Ah6XtRj
 brainstate/util/_scaling.py,sha256=U6DM-afPrLejiGqo1Nla7z4YbTBVicctsBEweurr_mk,7524
 brainstate/util/_struct.py,sha256=7HbbQNrZ3zxYw93MU1bUZ9ZPBKftYOVKuXEochLSErw,17479
 brainstate/util/filter.py,sha256=skuq51y0P5-N611fQ-zB2QyaJIFqN_WbROzZHv0RgOc,14086
-brainstate-0.1.2.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
-brainstate-0.1.2.dist-info/METADATA,sha256=Mv44muheQeXJ67RF5GQrkSxVWfj4VAZnkse6_7z5OTY,4135
-brainstate-0.1.2.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-brainstate-0.1.2.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
-brainstate-0.1.2.dist-info/RECORD,,
+brainstate-0.1.3.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
+brainstate-0.1.3.dist-info/METADATA,sha256=PaBKgarOn04Zgm7OY8tI8onrfX3bh0jACHKuarxZpas,4135
+brainstate-0.1.3.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+brainstate-0.1.3.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
+brainstate-0.1.3.dist-info/RECORD,,

brainstate/nn/_dyn_impl/__init__.py

@@ -1,42 +0,0 @@
-# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==============================================================================
-
-
-from ._dynamics_neuron import *
-from ._dynamics_neuron import __all__ as dyn_neuron_all
-from ._dynamics_synapse import *
-from ._dynamics_synapse import __all__ as dyn_synapse_all
-from ._inputs import *
-from ._inputs import __all__ as inputs_all
-from ._rate_rnns import *
-from ._rate_rnns import __all__ as rate_rnns
-from ._readout import *
-from ._readout import __all__ as readout_all
-
-__all__ = (
-    dyn_neuron_all
-    + dyn_synapse_all
-    + inputs_all
-    + rate_rnns
-    + readout_all
-)
-
-del (
-    dyn_neuron_all,
-    dyn_synapse_all,
-    inputs_all,
-    readout_all,
-    rate_rnns,
-)

brainstate/nn/_dynamics/__init__.py

@@ -1,37 +0,0 @@
-# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==============================================================================
-
-from ._dynamics_base import *
-from ._dynamics_base import __all__ as dyn_all
-from ._projection_base import *
-from ._projection_base import __all__ as projection_all
-from ._state_delay import *
-from ._state_delay import __all__ as state_delay_all
-from ._synouts import *
-from ._synouts import __all__ as synouts_all
-
-__all__ = (
-    dyn_all
-    + projection_all
-    + state_delay_all
-    + synouts_all
-)
-
-del (
-    dyn_all,
-    projection_all,
-    state_delay_all,
-    synouts_all
-)

brainstate/nn/_elementwise/__init__.py

@@ -1,22 +0,0 @@
-# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==============================================================================
-
-from ._dropout import *
-from ._dropout import __all__ as dropout_all
-from ._elementwise import *
-from ._elementwise import __all__ as elementwise_all
-
-__all__ = dropout_all + elementwise_all
-del dropout_all, elementwise_all

brainstate/nn/_interaction/__init__.py

@@ -1,41 +0,0 @@
-# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==============================================================================
-
-from ._conv import *
-from ._conv import __all__ as conv_all
-from ._embedding import *
-from ._embedding import __all__ as embed_all
-from ._linear import *
-from ._linear import __all__ as linear_all
-from ._normalizations import *
-from ._normalizations import __all__ as normalizations_all
-from ._poolings import *
-from ._poolings import __all__ as poolings_all
-
-__all__ = (
-    conv_all +
-    linear_all +
-    normalizations_all +
-    poolings_all +
-    embed_all
-)
-
-del (
-    conv_all,
-    linear_all,
-    normalizations_all,
-    poolings_all,
-    embed_all
-)