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

brainstate/random/_rand_seed_test.py

@@ -13,14 +13,12 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 import unittest
 
 import jax.numpy as jnp
 import jax.random
 
-import brainstate as bst
+import brainstate
 
 
 class TestRandom(unittest.TestCase):
@@ -28,23 +26,23 @@ class TestRandom(unittest.TestCase):
     def test_seed2(self):
         test_seed = 299
         key = jax.random.PRNGKey(test_seed)
-        bst.random.seed(key)
+        brainstate.random.seed(key)
 
         @jax.jit
         def jit_seed(key):
-            bst.random.seed(key)
-            with bst.random.seed_context(key):
-                print(bst.random.DEFAULT.value)
+            brainstate.random.seed(key)
+            with brainstate.random.seed_context(key):
+                print(brainstate.random.DEFAULT.value)
 
         jit_seed(key)
         jit_seed(1)
         jit_seed(None)
-        bst.random.seed(1)
+        brainstate.random.seed(1)
 
     def test_seed(self):
         test_seed = 299
-        bst.random.seed(test_seed)
-        a = bst.random.rand(3)
-        bst.random.seed(test_seed)
-        b = bst.random.rand(3)
+        brainstate.random.seed(test_seed)
+        a = brainstate.random.rand(3)
+        brainstate.random.seed(test_seed)
+        b = brainstate.random.rand(3)
         self.assertTrue(jnp.array_equal(a, b))

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainstate
-Version: 0.1.1
+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.3.dist-info/RECORD

@@ -0,0 +1,131 @@
+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=gt7pPK7VvDR7OO8VjBfxVVKyGQ264cvNHYNdhLpbeX0,10803
+brainstate/mixin_test.py,sha256=Oq_0fwC9vpXDN4t4dTBhWzLdFDNlcYsrcip14F1yECI,3079
+brainstate/surrogate.py,sha256=Iu1-WWJKH5CIjXIHKwEF8ufxVYpFZCltU_6dEH9DYqk,53470
+brainstate/transform.py,sha256=OvshYpPnp3YXPG6riy15Ve7jcX8t0aaATl1xZnsFeic,858
+brainstate/typing.py,sha256=XEOzcvMos8jcVeOf2rZb20M4A89xUEUpEIHu9qit2L8,10485
+brainstate/augment/__init__.py,sha256=Q9-JIwQ1FNn8VLS1MA9MrSylbvUjWSw98whrI3NIuKo,1229
+brainstate/augment/_autograd.py,sha256=zMMLZOidQq2p96wzEOgR-MynV2JH1l1AnoJ28eVwD-s,30100
+brainstate/augment/_autograd_test.py,sha256=UhNd41luca_Kj9a8byL3Dq64Ta55WU-je-LCiK_z0Vc,45060
+brainstate/augment/_eval_shape.py,sha256=mYRB5clSRHdXX0c4c5_WDIIBtRl4a9-xl8Tg60mgLBI,3854
+brainstate/augment/_eval_shape_test.py,sha256=WXrmZKmnykmYPveRfZbrDF0sFm2_PTr982yl4JM7Ebg,1390
+brainstate/augment/_mapping.py,sha256=AvGt8qhOdBgRJyWZn7J5JnOvxZc7XfZ_JKXaECOQjIw,43500
+brainstate/augment/_mapping_test.py,sha256=An5iSqrAXq_OVTvY3GFTjsB9AvFZ4j6AWNZQ_WWfra4,15319
+brainstate/augment/_random.py,sha256=bkngsIk6Wwi3e46I7YSbBjLCGAz0Q3WuadUH4mqTjbY,5348
+brainstate/compile/__init__.py,sha256=fQtG316MLkeeu1Ssp54Kghw1PwbGK5gNq9yRVJu0wjA,1474
+brainstate/compile/_ad_checkpoint.py,sha256=HM6L90HU0N84S30uJpX8wqTO0ZcDnctqmwf8qFlkDYo,9358
+brainstate/compile/_ad_checkpoint_test.py,sha256=XGqC8kjwKd64RPgBWXHAE3t8NdIXVUqmGDfrg5HqoVw,1743
+brainstate/compile/_conditions.py,sha256=fFSJWRFZxriAZFASj2lvlXpxbH323XcfsoB9ni2CYZ0,10183
+brainstate/compile/_conditions_test.py,sha256=bh8_D49Fl-s_xFYkO4joZJPgFGhw_vd6Wo8QB1HDKwE,8640
+brainstate/compile/_error_if.py,sha256=l8flahWEui9-9IavtZQjyutIeR8DU0jieahRvAXUxuw,2651
+brainstate/compile/_error_if_test.py,sha256=PjTIQOflegqnqJb9kO5YqwjW1OfrBmAtWlY_xn8rtDY,1897
+brainstate/compile/_jit.py,sha256=TN6Fk7qTHpZ2rzfBf4uyL9oFoU_IEpaDQhn1R_d_bd4,13858
+brainstate/compile/_jit_test.py,sha256=zD7kck9SQJGmUDolh9P4luKwQ21fBGje1Z4STTEXIuA,4135
+brainstate/compile/_loop_collect_return.py,sha256=qOYGoD2TMZd2Px6y241GWNwSjOLyaKlvzOwJWIPAbxg,23341
+brainstate/compile/_loop_collect_return_test.py,sha256=pEJdcOthEM17q5kXYhgR6JfzzqibijD_O1VPXVe_Ml4,1808
+brainstate/compile/_loop_no_collection.py,sha256=tPkSxee41VexWEALpN1uuT78BDrX3uT1FHv8WZtov4c,7549
+brainstate/compile/_loop_no_collection_test.py,sha256=ivavF59xep_g9bV1SSdXd5E1f6nhc7EUBfcbgHpxbfg,1419
+brainstate/compile/_make_jaxpr.py,sha256=nW5VV9fcqD6bo0GBWxGT2005ANox1ym7ChhGV_q5Aig,33463
+brainstate/compile/_make_jaxpr_test.py,sha256=wvZHl7ecQrmD6C_Nxy8z64l6r3Tj1kQEM54UNIU1Er4,4416
+brainstate/compile/_progress_bar.py,sha256=iTp-CkcK5aMbiA0ZdlSfEXXrc4_anifYhyDzl19dksI,7488
+brainstate/compile/_unvmap.py,sha256=9S42MeTmFJa8nfBI_AjEfrAdUsDmM3KFat59O8zXIEw,4120
+brainstate/compile/_util.py,sha256=QD7lvS4Zb3P2HPNIm6mG8Rl_Lk2tQj0znOCPcH95XnI,6275
+brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
+brainstate/functional/_activations.py,sha256=3ym1oDdftVxYlHIXWdS7XONiDFaDklqokIioi4qrsHo,21688
+brainstate/functional/_activations_test.py,sha256=66PTMI99pfi1hr2rfFixg2OlUN0WME9628T7U2NF0iI,13475
+brainstate/functional/_normalization.py,sha256=_F5u9iJm8KJfxlnmV_LwJXtrAZOJj357aXc_sfGe4CQ,2567
+brainstate/functional/_others.py,sha256=iAqvLCmXQz6C80EjE0-5bwXCBKHoKqsggneiOAXNmQo,1699
+brainstate/functional/_spikes.py,sha256=YJ4pa5LlywximSE6ZAAs9g2k-8cO95mwxdD29HhT-oM,7051
+brainstate/graph/__init__.py,sha256=noo4TjBg6iEhjjwk0sAGUhR7Ge-z8Vnc2rLYUvnqttw,1295
+brainstate/graph/_graph_node.py,sha256=zxtfzv6RXs9ylTerW5scUZ4mHic66df5WTHCrUzHLDA,6836
+brainstate/graph/_graph_node_test.py,sha256=g6nrCCI-awfxhZLnjmYhfQvynumWJjz7fPxgIF-VSr4,2832
+brainstate/graph/_graph_operation.py,sha256=gnLZB5YLgsnwVAZugv_5sryzU9_QYyz2r0-D5y1ZG7Q,63405
+brainstate/graph/_graph_operation_test.py,sha256=nxMwEoaRZpTkIeR3h_Aduuys0Bm1DRlYLd_CXkErTSg,19715
+brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1097
+brainstate/init/_base.py,sha256=A5VIJV_xPyCD4Lo4RXe2WPlWvADAy_UPhexuwzp-6EI,1633
+brainstate/init/_generic.py,sha256=o60EPJZe7laJgPsgrnorl3Km_Lc_Lr4qQQKmu6g5G4o,7992
+brainstate/init/_random_inits.py,sha256=GYovk2h2Lc1jY5ByW4R8OoZ71jbaE3BYGTfDw4-a7yo,18209
+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=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/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
+brainstate/optim/_lr_scheduler_test.py,sha256=gzzd_rzgtDWzF-93mujg043jv2wO9IceXGKiIYw5_ko,1790
+brainstate/optim/_optax_optimizer.py,sha256=RcO5N5JO_U0P4Q82waQuwMS7yWR-oP18eUwgzXJX0EY,5274
+brainstate/optim/_optax_optimizer_test.py,sha256=GVcszbNxBZ4z54bS5r-zDzP8gQZl7s5IsEZEd6G2gLo,1775
+brainstate/optim/_sgd_optimizer.py,sha256=Fn4CdGaDfV4qc_4gacCcGSjit5-ZrsBbQP-KB0Ocdac,46110
+brainstate/random/__init__.py,sha256=c5q-RC3grRIjx-HBb2IhKZpi_xzbFmUUxzRAzqfREic,1045
+brainstate/random/_rand_funs.py,sha256=c4xiY2NeMizSslxbWOa-QJZ3h-LDfsgH4fbvBMSNLWg,137646
+brainstate/random/_rand_funs_test.py,sha256=G8BuxDjBSeE-Mh7KuwVk6mPQhHE0m1R5HDFljEOIdzg,20669
+brainstate/random/_rand_seed.py,sha256=1ZdfFZWyOhpd72EDdEDmpkp3yoLVwdv-sGI9BwiZfzI,5949
+brainstate/random/_rand_seed_test.py,sha256=waXXfch57X1XE1zDnCRokT6ziZOK0g-lYE80o6epDYM,1536
+brainstate/random/_rand_state.py,sha256=dTO7wqmuTYRdPy7ItsrK-7aNt5QQXOsZ4XwiH7mzmy8,55170
+brainstate/random/_random_for_unit.py,sha256=kGp4EUX19MXJ9Govoivbg8N0bddqOldKEI2h_TbdONY,2057
+brainstate/util/__init__.py,sha256=zwdw0253tY2kgWHHPPwaU6JHvjw4rioUnRmSAWF4Po0,1569
+brainstate/util/_caller.py,sha256=RBRwu1HT_Lww7EE3WTjzrwwKqVWg7dn_JXKmO9ojM38,2752
+brainstate/util/_error.py,sha256=GBLaU_glQlgdrjBYIMiO3Dy5N_ST6-QNsgAOa9kjUjw,1757
+brainstate/util/_others.py,sha256=vGo4uW0vd7wL5DYtxKWqXhoNmI-O_p26HsucoMnhdVA,16407
+brainstate/util/_pretty_pytree.py,sha256=PHMiES2vN_bke_GeHaSYHQZUtXtR-HkcMch8HQJ8Cw8,33536
+brainstate/util/_pretty_pytree_test.py,sha256=Dn0TdjX6wLBXaTD4jfYTu6cKfFHwKSxi4_3bX7kB_IA,5621
+brainstate/util/_pretty_repr.py,sha256=7Xp7IFNUeP7cGlpvwwJyBslbQVnXEqC1I6neV1Jx1S8,10527
+brainstate/util/_pretty_table.py,sha256=uJVaamFGQ4nKP8TkEGPWXHpzjMecDo2q1Ah6XtRjdPY,108117
+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.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/_elementwise/_elementwise_test.py

@@ -1,171 +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 __future__ import annotations
-
-from absl.testing import absltest
-from absl.testing import parameterized
-
-import brainstate as bst
-
-
-class Test_Activation(parameterized.TestCase):
-
-    def test_Threshold(self):
-        threshold_layer = bst.nn.Threshold(5, 20)
-        input = bst.random.randn(2)
-        output = threshold_layer(input)
-
-    def test_ReLU(self):
-        ReLU_layer = bst.nn.ReLU()
-        input = bst.random.randn(2)
-        output = ReLU_layer(input)
-
-    def test_RReLU(self):
-        RReLU_layer = bst.nn.RReLU(lower=0, upper=1)
-        input = bst.random.randn(2)
-        output = RReLU_layer(input)
-
-    def test_Hardtanh(self):
-        Hardtanh_layer = bst.nn.Hardtanh(min_val=0, max_val=1, )
-        input = bst.random.randn(2)
-        output = Hardtanh_layer(input)
-
-    def test_ReLU6(self):
-        ReLU6_layer = bst.nn.ReLU6()
-        input = bst.random.randn(2)
-        output = ReLU6_layer(input)
-
-    def test_Sigmoid(self):
-        Sigmoid_layer = bst.nn.Sigmoid()
-        input = bst.random.randn(2)
-        output = Sigmoid_layer(input)
-
-    def test_Hardsigmoid(self):
-        Hardsigmoid_layer = bst.nn.Hardsigmoid()
-        input = bst.random.randn(2)
-        output = Hardsigmoid_layer(input)
-
-    def test_Tanh(self):
-        Tanh_layer = bst.nn.Tanh()
-        input = bst.random.randn(2)
-        output = Tanh_layer(input)
-
-    def test_SiLU(self):
-        SiLU_layer = bst.nn.SiLU()
-        input = bst.random.randn(2)
-        output = SiLU_layer(input)
-
-    def test_Mish(self):
-        Mish_layer = bst.nn.Mish()
-        input = bst.random.randn(2)
-        output = Mish_layer(input)
-
-    def test_Hardswish(self):
-        Hardswish_layer = bst.nn.Hardswish()
-        input = bst.random.randn(2)
-        output = Hardswish_layer(input)
-
-    def test_ELU(self):
-        ELU_layer = bst.nn.ELU(alpha=0.5, )
-        input = bst.random.randn(2)
-        output = ELU_layer(input)
-
-    def test_CELU(self):
-        CELU_layer = bst.nn.CELU(alpha=0.5, )
-        input = bst.random.randn(2)
-        output = CELU_layer(input)
-
-    def test_SELU(self):
-        SELU_layer = bst.nn.SELU()
-        input = bst.random.randn(2)
-        output = SELU_layer(input)
-
-    def test_GLU(self):
-        GLU_layer = bst.nn.GLU()
-        input = bst.random.randn(4, 2)
-        output = GLU_layer(input)
-
-    @parameterized.product(
-        approximate=['tanh', 'none']
-    )
-    def test_GELU(self, approximate):
-        GELU_layer = bst.nn.GELU()
-        input = bst.random.randn(2)
-        output = GELU_layer(input)
-
-    def test_Hardshrink(self):
-        Hardshrink_layer = bst.nn.Hardshrink(lambd=1)
-        input = bst.random.randn(2)
-        output = Hardshrink_layer(input)
-
-    def test_LeakyReLU(self):
-        LeakyReLU_layer = bst.nn.LeakyReLU()
-        input = bst.random.randn(2)
-        output = LeakyReLU_layer(input)
-
-    def test_LogSigmoid(self):
-        LogSigmoid_layer = bst.nn.LogSigmoid()
-        input = bst.random.randn(2)
-        output = LogSigmoid_layer(input)
-
-    def test_Softplus(self):
-        Softplus_layer = bst.nn.Softplus()
-        input = bst.random.randn(2)
-        output = Softplus_layer(input)
-
-    def test_Softshrink(self):
-        Softshrink_layer = bst.nn.Softshrink(lambd=1)
-        input = bst.random.randn(2)
-        output = Softshrink_layer(input)
-
-    def test_PReLU(self):
-        PReLU_layer = bst.nn.PReLU(num_parameters=2, init=0.5)
-        input = bst.random.randn(2)
-        output = PReLU_layer(input)
-
-    def test_Softsign(self):
-        Softsign_layer = bst.nn.Softsign()
-        input = bst.random.randn(2)
-        output = Softsign_layer(input)
-
-    def test_Tanhshrink(self):
-        Tanhshrink_layer = bst.nn.Tanhshrink()
-        input = bst.random.randn(2)
-        output = Tanhshrink_layer(input)
-
-    def test_Softmin(self):
-        Softmin_layer = bst.nn.Softmin(dim=2)
-        input = bst.random.randn(2, 3, 4)
-        output = Softmin_layer(input)
-
-    def test_Softmax(self):
-        Softmax_layer = bst.nn.Softmax(dim=2)
-        input = bst.random.randn(2, 3, 4)
-        output = Softmax_layer(input)
-
-    def test_Softmax2d(self):
-        Softmax2d_layer = bst.nn.Softmax2d()
-        input = bst.random.randn(2, 3, 12, 13)
-        output = Softmax2d_layer(input)
-
-    def test_LogSoftmax(self):
-        LogSoftmax_layer = bst.nn.LogSoftmax(dim=2)
-        input = bst.random.randn(2, 3, 4)
-        output = LogSoftmax_layer(input)
-
-
-if __name__ == '__main__':
-    absltest.main()

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
-)

brainstate-0.1.1.dist-info/RECORD

@@ -1,133 +0,0 @@
-brainstate/__init__.py,sha256=RrDTTcA9N5s3GU2fxXMk4GwGmMuHdkhf4i2Ks98INFY,1496
-brainstate/_compatible_import.py,sha256=MSLUh2xFmboTpwqVnBy5X8dOlNi2fafwFnaU_JJIXQw,4931
-brainstate/_state.py,sha256=bHV6gEFQbYMPiQNcdTxGTMGFaHuQrY9wIR3ftCyeFJk,60749
-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_test.py,sha256=Oq_0fwC9vpXDN4t4dTBhWzLdFDNlcYsrcip14F1yECI,3079
-brainstate/surrogate.py,sha256=Iu1-WWJKH5CIjXIHKwEF8ufxVYpFZCltU_6dEH9DYqk,53470
-brainstate/transform.py,sha256=OvshYpPnp3YXPG6riy15Ve7jcX8t0aaATl1xZnsFeic,858
-brainstate/typing.py,sha256=XEOzcvMos8jcVeOf2rZb20M4A89xUEUpEIHu9qit2L8,10485
-brainstate/augment/__init__.py,sha256=Q9-JIwQ1FNn8VLS1MA9MrSylbvUjWSw98whrI3NIuKo,1229
-brainstate/augment/_autograd.py,sha256=zMMLZOidQq2p96wzEOgR-MynV2JH1l1AnoJ28eVwD-s,30100
-brainstate/augment/_autograd_test.py,sha256=S2eEgrwTzdSi3u2nKE3u37WSThosLwx1WCP9ptJAGKo,44060
-brainstate/augment/_eval_shape.py,sha256=mYRB5clSRHdXX0c4c5_WDIIBtRl4a9-xl8Tg60mgLBI,3854
-brainstate/augment/_eval_shape_test.py,sha256=LFOJx7CWltmRLXdGY175UebLwtEMz2CzJ_gLqMZsJTw,1393
-brainstate/augment/_mapping.py,sha256=AvGt8qhOdBgRJyWZn7J5JnOvxZc7XfZ_JKXaECOQjIw,43500
-brainstate/augment/_mapping_test.py,sha256=OLkd9obwcy5K8jHpiQLT1o5c4rVG7tfg5-CYdMJvJWQ,14711
-brainstate/augment/_random.py,sha256=bkngsIk6Wwi3e46I7YSbBjLCGAz0Q3WuadUH4mqTjbY,5348
-brainstate/compile/__init__.py,sha256=fQtG316MLkeeu1Ssp54Kghw1PwbGK5gNq9yRVJu0wjA,1474
-brainstate/compile/_ad_checkpoint.py,sha256=HM6L90HU0N84S30uJpX8wqTO0ZcDnctqmwf8qFlkDYo,9358
-brainstate/compile/_ad_checkpoint_test.py,sha256=R1I76nG4zIqb6g3M_VxWts7rUC1OHJCjtQhPkcbXodk,1746
-brainstate/compile/_conditions.py,sha256=fFSJWRFZxriAZFASj2lvlXpxbH323XcfsoB9ni2CYZ0,10183
-brainstate/compile/_conditions_test.py,sha256=s9LF6h9LvigvgxUIugTqvgCHBIU8TXS1Ar1OlIxXfrw,8389
-brainstate/compile/_error_if.py,sha256=l8flahWEui9-9IavtZQjyutIeR8DU0jieahRvAXUxuw,2651
-brainstate/compile/_error_if_test.py,sha256=OdJG483IIdOrCHxtHd49OHfOxCSnSkk7GdAUOzSt8bE,2044
-brainstate/compile/_jit.py,sha256=TN6Fk7qTHpZ2rzfBf4uyL9oFoU_IEpaDQhn1R_d_bd4,13858
-brainstate/compile/_jit_test.py,sha256=zD7kck9SQJGmUDolh9P4luKwQ21fBGje1Z4STTEXIuA,4135
-brainstate/compile/_loop_collect_return.py,sha256=qOYGoD2TMZd2Px6y241GWNwSjOLyaKlvzOwJWIPAbxg,23341
-brainstate/compile/_loop_collect_return_test.py,sha256=r-q-D2fYuO9oIGKCu7ZUZDlZj-RlPthdn7fmP04TxbQ,1804
-brainstate/compile/_loop_no_collection.py,sha256=tPkSxee41VexWEALpN1uuT78BDrX3uT1FHv8WZtov4c,7549
-brainstate/compile/_loop_no_collection_test.py,sha256=oStB1CSG_iLp9sHdXd1hJNFvlxbzjck9Iy4sABoJDj4,1419
-brainstate/compile/_make_jaxpr.py,sha256=-864jBQjZCUft_J7m82C9yQSZ4TnTwYliEnOScV9_zM,33120
-brainstate/compile/_make_jaxpr_test.py,sha256=3rfb1oz0u1wHPmnppnmo2BlUBbEE02FlbvD6Rehub9I,4290
-brainstate/compile/_progress_bar.py,sha256=iTp-CkcK5aMbiA0ZdlSfEXXrc4_anifYhyDzl19dksI,7488
-brainstate/compile/_unvmap.py,sha256=9S42MeTmFJa8nfBI_AjEfrAdUsDmM3KFat59O8zXIEw,4120
-brainstate/compile/_util.py,sha256=QD7lvS4Zb3P2HPNIm6mG8Rl_Lk2tQj0znOCPcH95XnI,6275
-brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
-brainstate/functional/_activations.py,sha256=3ym1oDdftVxYlHIXWdS7XONiDFaDklqokIioi4qrsHo,21688
-brainstate/functional/_activations_test.py,sha256=T___RlSrIfXwlkw8dg5A9EZMTZGDzv3a2evUwq_nYFg,13034
-brainstate/functional/_normalization.py,sha256=_F5u9iJm8KJfxlnmV_LwJXtrAZOJj357aXc_sfGe4CQ,2567
-brainstate/functional/_others.py,sha256=iAqvLCmXQz6C80EjE0-5bwXCBKHoKqsggneiOAXNmQo,1699
-brainstate/functional/_spikes.py,sha256=YJ4pa5LlywximSE6ZAAs9g2k-8cO95mwxdD29HhT-oM,7051
-brainstate/graph/__init__.py,sha256=noo4TjBg6iEhjjwk0sAGUhR7Ge-z8Vnc2rLYUvnqttw,1295
-brainstate/graph/_graph_node.py,sha256=zxtfzv6RXs9ylTerW5scUZ4mHic66df5WTHCrUzHLDA,6836
-brainstate/graph/_graph_node_test.py,sha256=BFGfdzZFDHI0XK7hHotSVWKt3em1taGvn8FHF9NCXx8,2702
-brainstate/graph/_graph_operation.py,sha256=gnLZB5YLgsnwVAZugv_5sryzU9_QYyz2r0-D5y1ZG7Q,63405
-brainstate/graph/_graph_operation_test.py,sha256=zjvpKjQAFWtw8YZuqOk_jmlZNb_-E8oPyNx57dyc8jI,18556
-brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1097
-brainstate/init/_base.py,sha256=A5VIJV_xPyCD4Lo4RXe2WPlWvADAy_UPhexuwzp-6EI,1633
-brainstate/init/_generic.py,sha256=o60EPJZe7laJgPsgrnorl3Km_Lc_Lr4qQQKmu6g5G4o,7992
-brainstate/init/_random_inits.py,sha256=GYovk2h2Lc1jY5ByW4R8OoZ71jbaE3BYGTfDw4-a7yo,18209
-brainstate/init/_random_inits_test.py,sha256=lBL2RQdBSZ88Zqz4IMdbHJMvDi7ooZq6caCpHfNtIRk,5197
-brainstate/init/_regular_inits.py,sha256=tcZDTXR8jiRqERuJleg_9JRGGzJj-V3LhTMcr_gt_k0,3149
-brainstate/init/_regular_inits_test.py,sha256=tJl4aOkclllJIfKzJTbc0cfYCw2SoBsx8_G123RnqbU,1842
-brainstate/nn/__init__.py,sha256=dUEH_S5jRTq0nY1xNkpN_4MQrmHXHGSmGp3xgGenqNc,1924
-brainstate/nn/_collective_ops.py,sha256=rWcjqaP0rW6sXdhE0fjtDi4twmq5VvhNEg8AoiQ-tDU,17390
-brainstate/nn/_collective_ops_test.py,sha256=vvyXNPluFa0Ii_sxVUTwsdMwiEITSXvRA1o7-HoPANk,1367
-brainstate/nn/_common.py,sha256=Pt8P1LgE0qW3QnfX8CQQGH3yZ78RL7NkIqqYEw_z8xs,7096
-brainstate/nn/_exp_euler.py,sha256=ndDB43PM4jsZKu_zdLTZ2-ojnuNrg55LZap23oBTtdA,3493
-brainstate/nn/_exp_euler_test.py,sha256=clwRD8QR71k1jn6NrACMDEUcFMh0J9RTosoPnlYWUkw,1242
-brainstate/nn/_module.py,sha256=ksbmO_-bZzaP8XTRAcN2oi2ZMdSuSv-XVeWc2VJncVc,12808
-brainstate/nn/_module_test.py,sha256=V4ZhiY_zYPvArkB2eeOTtZcgQrtlRyXKMbS1AJH4vC8,8893
-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=Tfzrzu7udGrLJGnqItiLWe5WT0dgduvYOgzGCnaPJQg,6317
-brainstate/nn/_dyn_impl/_dynamics_synapse.py,sha256=b2QMEzasbXTp7k1XWNg7nRt4-WddCym0CSEErjPRIWA,27449
-brainstate/nn/_dyn_impl/_dynamics_synapse_test.py,sha256=t5i-HV0ii9sUNzWTEv04o26QVtQ-mCdMJcFq2MD755A,4981
-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=gNgtr-a4ZiU1XF9wFG1HiJ9fLosfWchVR9Zn1x39xt4,2452
-brainstate/nn/_dyn_impl/_readout.py,sha256=UzaJ6WIMpNjqHolhjHTpyCZ5E2nSDucvuV5PBW7rOPs,7151
-brainstate/nn/_dyn_impl/_readout_test.py,sha256=R9JJPRvy3mAHSv8n1Hzjk2kBSDjBzJNbS83ystll86s,2109
-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=gXMwENqqSvyZbMpLP0QtYndJ_h39dF5gIeiiSbMAjTk,2721
-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=V_jDswRN4VvEXD-2yJO3VA1TALgX0HK6oPBQiUntOWc,2266
-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=k6aB5v8RYMoV5w8UV9UNSFhaQTV7woS6jx3SNESuCRs,4383
-brainstate/nn/_elementwise/_elementwise.py,sha256=yO6BukJFLfnICPELEFp2kfutFM8Jeg7TaERXFp-aNrk,33304
-brainstate/nn/_elementwise/_elementwise_test.py,sha256=SZI9jB39sZ5SO1dpWGW-PhodthwN0GU9FY1nqf2fWcs,5341
-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=AAtzAY9K3YS-n_5oEOTR5EUnEtWZXGnOw1xc4oOLXW8,3965
-brainstate/nn/_event/_linear_mv.py,sha256=KslIdse7KZ7FWm5qcOMC-COM__1rB-sUJmXvlXY11y8,2683
-brainstate/nn/_event/_linear_mv_test.py,sha256=XMffPFMXwSryAW1gLxIEXE8eC7DJSC3aWRMDuDrLem8,3900
-brainstate/nn/_interaction/__init__.py,sha256=TTY_SeNrdx4VnUSw6vdyl02OHdS9Qs15cWBp6kjsyNQ,1289
-brainstate/nn/_interaction/_conv.py,sha256=QMZR4HRAYe8Nzr3qExp0PWr6wuUp2rh5YxmPq1r1vcY,18461
-brainstate/nn/_interaction/_conv_test.py,sha256=VJacu1afSmxFpgKerjMA2r3kwXAhf7aVhP6QdgJbqyY,8697
-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=QfCR8SBBed9OnSY-AmQ0kJqoggDA3Xem0dRJ0BusxLU,2872
-brainstate/nn/_interaction/_normalizations.py,sha256=4yoU04wK21_zk6Egpwb9YN_pSWUH5Qa6hJ1bLxX7wQw,37400
-brainstate/nn/_interaction/_normalizations_test.py,sha256=2p1Jf8nA999VYGWbvOZfKYlKk6UmL0vaEB76xkXxkXw,2438
-brainstate/nn/_interaction/_poolings.py,sha256=2hJDACuHv3CkO-nyyAqVZp7Sa3UFa_ROFGiYmhXWqKk,46982
-brainstate/nn/_interaction/_poolings_test.py,sha256=wmd5PngZ3E9tNyF3s0xk-DoDR5yFqpTi9A6nbNoIqn4,7429
-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
-brainstate/optim/_lr_scheduler_test.py,sha256=QJSYfvd7llHOGWY6C2Jc_D3QbH-yZoRK8mSn5osmSes,1776
-brainstate/optim/_optax_optimizer.py,sha256=RcO5N5JO_U0P4Q82waQuwMS7yWR-oP18eUwgzXJX0EY,5274
-brainstate/optim/_optax_optimizer_test.py,sha256=DAomE8Eu3dn4gh1S3EZ_u4pW4rhcl16vWPbnDcN3Rs4,1762
-brainstate/optim/_sgd_optimizer.py,sha256=Fn4CdGaDfV4qc_4gacCcGSjit5-ZrsBbQP-KB0Ocdac,46110
-brainstate/random/__init__.py,sha256=c5q-RC3grRIjx-HBb2IhKZpi_xzbFmUUxzRAzqfREic,1045
-brainstate/random/_rand_funs.py,sha256=c4xiY2NeMizSslxbWOa-QJZ3h-LDfsgH4fbvBMSNLWg,137646
-brainstate/random/_rand_funs_test.py,sha256=abO5lSoPBgBcg6ecFE1qnCg98__QGa68GSYC5pQW5QI,19438
-brainstate/random/_rand_seed.py,sha256=1ZdfFZWyOhpd72EDdEDmpkp3yoLVwdv-sGI9BwiZfzI,5949
-brainstate/random/_rand_seed_test.py,sha256=Qibcs-ZqCvj1LuucmQ8H00B_HBNhf2f6un0aUdNZNTw,1518
-brainstate/random/_rand_state.py,sha256=dTO7wqmuTYRdPy7ItsrK-7aNt5QQXOsZ4XwiH7mzmy8,55170
-brainstate/random/_random_for_unit.py,sha256=kGp4EUX19MXJ9Govoivbg8N0bddqOldKEI2h_TbdONY,2057
-brainstate/util/__init__.py,sha256=zwdw0253tY2kgWHHPPwaU6JHvjw4rioUnRmSAWF4Po0,1569
-brainstate/util/_caller.py,sha256=RBRwu1HT_Lww7EE3WTjzrwwKqVWg7dn_JXKmO9ojM38,2752
-brainstate/util/_error.py,sha256=GBLaU_glQlgdrjBYIMiO3Dy5N_ST6-QNsgAOa9kjUjw,1757
-brainstate/util/_others.py,sha256=vGo4uW0vd7wL5DYtxKWqXhoNmI-O_p26HsucoMnhdVA,16407
-brainstate/util/_pretty_pytree.py,sha256=PHMiES2vN_bke_GeHaSYHQZUtXtR-HkcMch8HQJ8Cw8,33536
-brainstate/util/_pretty_pytree_test.py,sha256=Dn0TdjX6wLBXaTD4jfYTu6cKfFHwKSxi4_3bX7kB_IA,5621
-brainstate/util/_pretty_repr.py,sha256=7Xp7IFNUeP7cGlpvwwJyBslbQVnXEqC1I6neV1Jx1S8,10527
-brainstate/util/_pretty_table.py,sha256=uJVaamFGQ4nKP8TkEGPWXHpzjMecDo2q1Ah6XtRjdPY,108117
-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.1.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
-brainstate-0.1.1.dist-info/METADATA,sha256=_vaI9i2ZItieDpBQMXDI0yodxtyV_Fi4F2pb6riQPR4,4135
-brainstate-0.1.1.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-brainstate-0.1.1.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
-brainstate-0.1.1.dist-info/RECORD,,