brainstate 0.0.2.post20240913__py2.py3-none-any.whl → 0.0.2.post20241010__py2.py3-none-any.whl

brainstate/transform/_jit.py

@@ -76,7 +76,7 @@ def _get_jitted_fun(
   def jitted_fun(*args, **params):
     if jax.config.jax_disable_jit:
       return fun.fun(*args, **params)
-    states = fun.compile_and_get_states_by_static_args(*args, **kwargs)
+    states = fun.compile_and_get_states_by_static_args(*args, **params)
     state_vals, outs = jit_fun([st.value for st in states], *args, **params)
     _assign_state_values(states, state_vals)
     return outs

brainstate/transform/_jit_test.py

@@ -46,6 +46,25 @@ class TestJIT(unittest.TestCase):
     key = fun1.stateful_fun.get_arg_cache_key(x)
     self.assertTrue(len(fun1.stateful_fun.get_states(key)) == 2)
 
+  def test_kwargs(self):
+    a = bc.State(bc.random.randn(10))
+
+    @bc.transform.jit
+    def fun1(inp):
+      a.value += inp
+
+      b = bc.State(bc.random.randn(1))
+
+      def inner_fun(x):
+        b.value += x
+
+      bc.transform.for_loop(inner_fun, bc.random.randn(100))
+
+      return a.value + b.value
+
+    # test kwargs
+    print(fun1(inp=bc.random.randn(10)))
+
   def test_jit_compile_sensitive_to_input_shape(self):
     global_data = [0]
 

brainstate/transform/_make_jaxpr.py

@@ -727,7 +727,7 @@ def _make_jaxpr(
     closed_jaxpr = jax.core.ClosedJaxpr(jaxpr, consts)
     if return_shape:
       out_avals, _ = jax.util.unzip2(out_type)
-      out_shapes_flat = [jax.ShapeDtypeStruct(a.shape, a.dtype, a.named_shape) for a in out_avals]
+      out_shapes_flat = [jax.ShapeDtypeStruct(a.shape, a.dtype) for a in out_avals]
       return closed_jaxpr, jax.tree.unflatten(out_tree(), out_shapes_flat)
     return closed_jaxpr
 

{brainstate-0.0.2.post20240913.dist-info → brainstate-0.0.2.post20241010.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainstate
-Version: 0.0.2.post20240913
+Version: 0.0.2.post20241010
 Summary: A State-based Transformation System for Brain Dynamics Programming.
 Home-page: https://github.com/brainpy/brainstate
 Author: BDP

brainstate-0.0.2.post20241010.dist-info/RECORD

@@ -0,0 +1,87 @@
+brainstate/__init__.py,sha256=mr3AKWhoIpRW-h7YO6M6JR9SMlmQ5NyROJ1eChvlIHI,1539
+brainstate/_module.py,sha256=7UompqgCC80iJZA3a5XgL1ZtFlCWIBQoq9qUiylgYxg,53655
+brainstate/_module_test.py,sha256=oQaoaZBTo1o3wHrMEJTInQCc7RdcVs1gcfQGvdSb1SI,7843
+brainstate/_state.py,sha256=W6GCjxL0YigLXrUOben4OcZiHeyk-hbqZTNNrbb5MpE,12182
+brainstate/_state_test.py,sha256=HDdipndRLhEHWEdTmyT1ayEBkbv6qJKykfCWKI6yJ_E,1253
+brainstate/_utils.py,sha256=RLorgGJkt2BhbX4C-ygd-PPG0wfcGCghjSP93sRvzqM,833
+brainstate/_visualization.py,sha256=iE4UbAr6ejZQ806zJGRM7WXMUfxfIlK7HQ3PCBZBV-c,1445
+brainstate/environ.py,sha256=imgalvuA6iXQUFHzPQJhqos7QBbRyS26pULfZ2gLm3c,15674
+brainstate/environ_test.py,sha256=bW5-V-IvlKa6ci3DWSA0Sr_Q7681ZyO_eNygWzuefJc,1742
+brainstate/mixin.py,sha256=UU56tB_44oZNCWADAaXPAOUDf5fz9Hfmz7gCV1wqfws,10768
+brainstate/mixin_test.py,sha256=-Ej9oUOu8O1M4oy37SVMj7xNRYhHHyAHwrjS_aISayo,2923
+brainstate/surrogate.py,sha256=6AO79JOOs-X5x0FT0EDqO9lNtjJZAs26H4mljgpTvAw,45197
+brainstate/typing.py,sha256=szCYee9R15YQfsEAQOx95_LqfrD9AYuE5dfTBTPd8sg,9165
+brainstate/util.py,sha256=y-6eX1z3EMyg6pfZt4YdDalOnJ3HDAT1IPBCJDp-gQI,19876
+brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
+brainstate/functional/_activations.py,sha256=Q4M3vfe3ogDIFzyJPNaSkzwtbsI0ETUnOnjCAyWDa-I,20773
+brainstate/functional/_activations_test.py,sha256=cF7kLad1vDq0QudBAr8hU5pHaYBbwlvDb4Vz__wpw2s,12061
+brainstate/functional/_normalization.py,sha256=3uiZ5pBm5jYccb6qqQMHy9mqYg1_mvD51i3PDgewmUY,2485
+brainstate/functional/_others.py,sha256=1Epp75RkGYobMc2kHISZuS-_xnTFk3zHb1UHadwugCo,1711
+brainstate/functional/_spikes.py,sha256=70qGvo4B--QtxfJMjLwGmk9pVsf2x2YNEEgjT-il_Jw,2574
+brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1097
+brainstate/init/_base.py,sha256=jRTmfoUsH_315vW9YMZzyIn2KDAAsv56SplBnvOyBW0,1148
+brainstate/init/_generic.py,sha256=I6ZYT4t2Cx1L5o4yGvuLXHCZJcp00D-uGXyMAx6hpwQ,7383
+brainstate/init/_random_inits.py,sha256=vNUVDdUOCXTx2i3i1enzxgg1USCzugYd56r0-2lBL-0,15919
+brainstate/init/_regular_inits.py,sha256=2jdIvII_L0FzyRX_Xv81xHjreTt69rXqJ1QPVpMm_Ww,3015
+brainstate/nn/__init__.py,sha256=VQcAV6AR-HwtC392q0Mf4fano0-zB-lopHV6l79WXw0,2279
+brainstate/nn/_base.py,sha256=GNn0W3Q1TqsVhe9FjNRS6_9UFuzNe-QunBRHhczeieA,8499
+brainstate/nn/_connections.py,sha256=Ws9XA-kbYNi8jMCU-lnetfM0iZgz64uGqhWZoKoedCo,23005
+brainstate/nn/_dynamics.py,sha256=K1KjCb5lbBqr-7rKoJimqDEocAOGhWybH9Al-4Fzy8M,13249
+brainstate/nn/_elementwise.py,sha256=w5_TVwF5l8Ea6XCviUvYcHf04zLIOE-LLs_95G3t-Po,43639
+brainstate/nn/_embedding.py,sha256=qsFXgh9aC_r0ouynXEjQobadeC-MMVUgLNnlYpEPDKQ,2276
+brainstate/nn/_misc.py,sha256=Xc4U4NLmvfnKdBNDayFrRBPAy3p0beS6T9C59rIDP00,3790
+brainstate/nn/_normalizations.py,sha256=nh7A2lwZFDmfXH0fup0FqWZKIMdMzWsSvRpDM1go1BE,14356
+brainstate/nn/_others.py,sha256=_ODJpRkavxTQ7rehPycj1YNWj_DdnJX4ZiEJ5r__yuw,4530
+brainstate/nn/_poolings.py,sha256=jjYf23Yx8k09HpA2ds0YsKwlYSJJYCRd4QUQ0sRGBlU,45758
+brainstate/nn/_poolings_test.py,sha256=Mj4gO86Xl4JS5hHNR_CgeUdZQIqAxUoeBldS-eoZoBg,7264
+brainstate/nn/_rate_rnns.py,sha256=0WSbTXzKnKIsJxjTaUPogri0c6qpgFhC2ezdyFpqrN4,14466
+brainstate/nn/_readout.py,sha256=cjKYmUr2a1aOgZNeJ1h5eacWGaNp-kpHS-Of-LB_zyo,4249
+brainstate/nn/_synouts.py,sha256=Ta89Fp2zURTMsLRZq3501EyFhJ6r9pe2SDO80fb68To,4388
+brainstate/nn/metrics.py,sha256=cXctMcN9Wur4u0QlNbkgWafivxWtQs5Pud-XJ3j9J6c,13854
+brainstate/nn/_projection/__init__.py,sha256=L6svNHTb8BDh2rdX2eYmcx_NdscSdKykkQbzpdCSkTA,1207
+brainstate/nn/_projection/_align_post.py,sha256=S1huNBq3NkOfwrr7SXgTU6JvQk7KPVv86XwJ5iyvaBI,21106
+brainstate/nn/_projection/_align_pre.py,sha256=_wjdj8muuv2_fSW9m3KBUVjNkBg28BUmz3qZ9IA1rUM,24597
+brainstate/nn/_projection/_delta.py,sha256=KT8ySo3n_Q_7swzOH-ISDf0x9rjMkiv99H-vqeQZDR8,7122
+brainstate/nn/_projection/_utils.py,sha256=UcmELOqsINgqJr7eC5BSNNteyZ--1lyGjhUTJfxyMmA,813
+brainstate/nn/_projection/_vanilla.py,sha256=_bh_DLtF0o33SBtj6IGL8CTanFEtJwfjBrgxBEAmIlg,3397
+brainstate/nn/event/__init__.py,sha256=fALgJES2kkdEoabxeJ277s0RsEVrlEhD2e4XeW2SsbI,1052
+brainstate/nn/event/_misc.py,sha256=WJfEcFFk-NQQK_82mCYvSK3VfNh6DVNvSB0ZyNzLufM,994
+brainstate/nn/event/csr.py,sha256=9faOY8HXMTG1bK1KAhYsCe8HqE96Gs0eZzUHfYMw2t8,9915
+brainstate/nn/event/csr_test.py,sha256=yn2IMRi-Stv4y58jOx53nCmfmjFR1Q8W2wsjvyM9dUw,3547
+brainstate/nn/event/fixed_probability.py,sha256=T_cb1o9YFcHa1SMrAsdajttQIUjQgPrRTI6Yxik38Qo,9260
+brainstate/nn/event/fixed_probability_test.py,sha256=_MAfDPNj2Ww4ZYR4uQ49b1dAntjmal-NG7BdkCg4bzk,3740
+brainstate/nn/event/linear.py,sha256=-50mBVcYH6g_bFEJx5Bn5nWMStop7FRHmmacMAsaqIg,6719
+brainstate/nn/event/linear_test.py,sha256=joIVIOnqhQLbFcdkT7YsbjesI8ZxoPiPT-85hag16Gg,3243
+brainstate/optim/__init__.py,sha256=xDTUXKVCA9j9O4faieZqHz6HGpGrmnsVjlWqtbt0BUA,1061
+brainstate/optim/_lr_scheduler.py,sha256=emKnA52UVqOfUcX7LJqwP-FVDVlGGzTQi2djYmbCWUo,15627
+brainstate/optim/_lr_scheduler_test.py,sha256=OwF8Iz-PorEbO0gO--A7IIgQEytqEfYWbPucAgzqL90,1598
+brainstate/optim/_optax_optimizer.py,sha256=WxpXvm1JPoHq4Jk_xH97ixrpMb5GeSxElWphQJfX3tE,7324
+brainstate/optim/_optax_optimizer_test.py,sha256=Hn20tlRwq9_XnJ_R2TVRr3YNoRTDoFO9T8PF5cGOwQ0,596
+brainstate/optim/_sgd_optimizer.py,sha256=JiK_AVGregL0wn8uHhRQvK9Qq7Qja7dEyLW6Aan7b70,45826
+brainstate/random/__init__.py,sha256=c5q-RC3grRIjx-HBb2IhKZpi_xzbFmUUxzRAzqfREic,1045
+brainstate/random/_rand_funs.py,sha256=mIoENR3iEVeVR-qCQ2UQVP0SEosWry4xhzhYr0UXPAI,132072
+brainstate/random/_rand_seed.py,sha256=jRXP4zsQde1XyiOdx4aWpHXDvz5PAB1pogqQw0ywYnk,4633
+brainstate/random/_rand_state.py,sha256=LDt3p7JCEidg7tBQ8GNUJ8aAIlj-PWUh21T5-yutmlI,53887
+brainstate/random/_random_for_unit.py,sha256=Nm02GmMFiPx5LdfEDmbdK9hqLsebPdVd0bAQduGhASI,2017
+brainstate/random/random_test.py,sha256=vw45gTn-39tu2WUuamR6rr0gb2h1eeClvcaJDj16Vz8,18402
+brainstate/transform/__init__.py,sha256=hqef3a4sLQ_Oihuqs8E5IghSLr9o2bS7CWmwRL8jX6E,1887
+brainstate/transform/_autograd.py,sha256=Pj_YxpU52guaxQs1NcB6qDtXgkvaPcoJbuvIF8T-Wmk,23964
+brainstate/transform/_autograd_test.py,sha256=RWriMemIF9FVFUjQh4IHzLhT9LGyd1JXpjXfFZKHn10,38654
+brainstate/transform/_conditions.py,sha256=bdqdHCPCJIpRJxNr0enO2u81924YoIuA8kS8GUGY98g,12970
+brainstate/transform/_conditions_test.py,sha256=hg3gyOk4jn88F_ZYYqqwf6m87N3GlOUE9dC2V3BnMTA,7691
+brainstate/transform/_error_if.py,sha256=0JThfFqt9B3K3H6mS84qecBS22yTi3-FPzviaYacaMY,2597
+brainstate/transform/_error_if_test.py,sha256=kQZujlgr9bYnL-Vf7x4Zfc7jJk7rCLNVu-bsiry40dQ,1874
+brainstate/transform/_jit.py,sha256=SHMnenMP14u6ZyTf8k_2oS-tx4Nd7f1cboa2rE79lZk,11471
+brainstate/transform/_jit_test.py,sha256=KriQcRPi3Up1TI26Md1NH2xqlGn6aE2nvKWEPsTV1Ow,3254
+brainstate/transform/_loop_collect_return.py,sha256=8X6-3T3YoL_Buph9LiGASdrqPnRhsgsH9GQg1wcRos0,20800
+brainstate/transform/_loop_no_collection.py,sha256=p2vHoNNesDH2cM7b5LgLzSv90M8iNQPkRZEl0jhf7yA,6476
+brainstate/transform/_make_jaxpr.py,sha256=a66xJJ27S5hhwkAHSLtrfvAWxATFCoTQ3xvXXgG0-9Y,30435
+brainstate/transform/_make_jaxpr_test.py,sha256=K3vRUBroDTCCx0lnmhgHtgrlWvWglJO2f1K2phTvU70,3819
+brainstate/transform/_mapping.py,sha256=G9XUsD1xKLCprwwE0wv0gSXS0NYZ-ZIsv-PKKRlOoTA,3821
+brainstate/transform/_progress_bar.py,sha256=VGoRZPRBmB8ELNwLc6c7S8QhUUTvn0FY46IbBm9cuYM,3502
+brainstate/transform/_unvmap.py,sha256=8Se_23QrwDdcJpFcUnnMgD6EP-4XylbhP9K5TDhW358,3311
+brainstate-0.0.2.post20241010.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
+brainstate-0.0.2.post20241010.dist-info/METADATA,sha256=yfpoOk0ZcT8Pd3_kC6AGm4CfGb6IrMbIgZKBgAxgBsQ,3311
+brainstate-0.0.2.post20241010.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+brainstate-0.0.2.post20241010.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
+brainstate-0.0.2.post20241010.dist-info/RECORD,,

brainstate-0.0.2.post20240913.dist-info/RECORD

@@ -1,70 +0,0 @@
-brainstate/__init__.py,sha256=zipNSih9Tyvi4-5cXqNPGsDF7VeestkLp-lcjJ4-dA0,1408
-brainstate/_module.py,sha256=ULfItiqiQoIK1YUYfkasmyh8Rj4PoYJP7cxyuphEnIo,52463
-brainstate/_module_test.py,sha256=oQaoaZBTo1o3wHrMEJTInQCc7RdcVs1gcfQGvdSb1SI,7843
-brainstate/_random_for_unit.py,sha256=1rHr7gfH_bYrJfpxbDhQUk_j00Yosx-GzyZCXrLxsd0,2007
-brainstate/_state.py,sha256=J5j5NujvqU3Ftd_m_u_3mz4xWw81mdYgdzltrdJSy8o,12162
-brainstate/_state_test.py,sha256=HDdipndRLhEHWEdTmyT1ayEBkbv6qJKykfCWKI6yJ_E,1253
-brainstate/_utils.py,sha256=RLorgGJkt2BhbX4C-ygd-PPG0wfcGCghjSP93sRvzqM,833
-brainstate/environ.py,sha256=k0p1oyi9jbsPfuvqrPL-_zgSd7VW3LRs0LboxlaaIfc,11806
-brainstate/mixin.py,sha256=OumTTSVyYSbtudjfS_MRThsBaeVJ_0JggeMClY7xtBA,10758
-brainstate/mixin_test.py,sha256=-Ej9oUOu8O1M4oy37SVMj7xNRYhHHyAHwrjS_aISayo,2923
-brainstate/random.py,sha256=pTZvTH06hv08_TpwzAWCqAjy-8oNGmB6-Jp6MKfkLaY,188087
-brainstate/random_test.py,sha256=cCeuYvlZkCS2_RgG0vipZFNSHG8b-uJ7SXM9SZDCYQM,17866
-brainstate/surrogate.py,sha256=6AO79JOOs-X5x0FT0EDqO9lNtjJZAs26H4mljgpTvAw,45197
-brainstate/typing.py,sha256=szCYee9R15YQfsEAQOx95_LqfrD9AYuE5dfTBTPd8sg,9165
-brainstate/util.py,sha256=y-6eX1z3EMyg6pfZt4YdDalOnJ3HDAT1IPBCJDp-gQI,19876
-brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
-brainstate/functional/_activations.py,sha256=gvZ9E1-TsEUlyO7Om0eYzlM9DF-14_A32-gta1mjGo4,17798
-brainstate/functional/_normalization.py,sha256=IxE580waloZylZVXcpUUK4bWQdlE6oSPfafaKYfDkbg,2169
-brainstate/functional/_others.py,sha256=1Epp75RkGYobMc2kHISZuS-_xnTFk3zHb1UHadwugCo,1711
-brainstate/functional/_spikes.py,sha256=70qGvo4B--QtxfJMjLwGmk9pVsf2x2YNEEgjT-il_Jw,2574
-brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1097
-brainstate/init/_base.py,sha256=jRTmfoUsH_315vW9YMZzyIn2KDAAsv56SplBnvOyBW0,1148
-brainstate/init/_generic.py,sha256=LB7IQfswOG6X-q0QX5N8T5vZmxdygetsSBQ6iXlZ0oU,7324
-brainstate/init/_random_inits.py,sha256=vNUVDdUOCXTx2i3i1enzxgg1USCzugYd56r0-2lBL-0,15919
-brainstate/init/_regular_inits.py,sha256=2jdIvII_L0FzyRX_Xv81xHjreTt69rXqJ1QPVpMm_Ww,3015
-brainstate/nn/__init__.py,sha256=YJHoI8cXKVRS8f2vUl3Zegp5wm0svMz3qo9JmQJiMQk,2162
-brainstate/nn/_base.py,sha256=lzbZpku3Q2arH6ZaAwjs6bhbV0RcFChxo2UcpnX5t84,8481
-brainstate/nn/_connections.py,sha256=GSOW2IbpJRHdPyF4nFJ2RPgO8y6SVHT1Gn-pbri9pMk,22970
-brainstate/nn/_dynamics.py,sha256=OeYYXv1dqjUDcCsRhZo1XS7SP2li1vlH9uhME_PE9v0,13205
-brainstate/nn/_elementwise.py,sha256=Br2yd1kdr06iWGSvpoebWWO6suXFDiF8PQv_hOX9kZQ,43599
-brainstate/nn/_embedding.py,sha256=WbgrIaM_14abN8zBDr0xipBOsFc8dXP2m7Z_aRLAfmU,2249
-brainstate/nn/_misc.py,sha256=Xc4U4NLmvfnKdBNDayFrRBPAy3p0beS6T9C59rIDP00,3790
-brainstate/nn/_normalizations.py,sha256=9yVDORAEpqEkL9MYSPU4m7C4q8Qj5UNsPh9sKmIt5gQ,14329
-brainstate/nn/_others.py,sha256=AYyrbbdKZj16kT0cVITnoZHck4xcccM1W3LX5XM5Z3Q,4513
-brainstate/nn/_poolings.py,sha256=wO1Q4s8blsLLv4CMlkrvZm0ravdL3dFGyOcg2QDendI,45754
-brainstate/nn/_poolings_test.py,sha256=Mj4gO86Xl4JS5hHNR_CgeUdZQIqAxUoeBldS-eoZoBg,7264
-brainstate/nn/_rate_rnns.py,sha256=Cebhy57UWzfwrCfq0v2qLDegmb__mXL5ht750y4aTro,14457
-brainstate/nn/_readout.py,sha256=jsQwhVnrJICKw4wFq-Du2AORPb_XXz_tZ4cURcckU-E,4240
-brainstate/nn/_synouts.py,sha256=gi3EyKlzt4UoyghwvNIr03r7YabZyl1idbq9aYG8zYM,4379
-brainstate/nn/_projection/__init__.py,sha256=L6svNHTb8BDh2rdX2eYmcx_NdscSdKykkQbzpdCSkTA,1207
-brainstate/nn/_projection/_align_post.py,sha256=S1huNBq3NkOfwrr7SXgTU6JvQk7KPVv86XwJ5iyvaBI,21106
-brainstate/nn/_projection/_align_pre.py,sha256=_wjdj8muuv2_fSW9m3KBUVjNkBg28BUmz3qZ9IA1rUM,24597
-brainstate/nn/_projection/_delta.py,sha256=KT8ySo3n_Q_7swzOH-ISDf0x9rjMkiv99H-vqeQZDR8,7122
-brainstate/nn/_projection/_utils.py,sha256=UcmELOqsINgqJr7eC5BSNNteyZ--1lyGjhUTJfxyMmA,813
-brainstate/nn/_projection/_vanilla.py,sha256=_bh_DLtF0o33SBtj6IGL8CTanFEtJwfjBrgxBEAmIlg,3397
-brainstate/optim/__init__.py,sha256=1L6x_qZprw3PJYddB1nX-uTFGUl6_Qt3PM0OdY6g968,917
-brainstate/optim/_lr_scheduler.py,sha256=emKnA52UVqOfUcX7LJqwP-FVDVlGGzTQi2djYmbCWUo,15627
-brainstate/optim/_lr_scheduler_test.py,sha256=OwF8Iz-PorEbO0gO--A7IIgQEytqEfYWbPucAgzqL90,1598
-brainstate/optim/_sgd_optimizer.py,sha256=JiK_AVGregL0wn8uHhRQvK9Qq7Qja7dEyLW6Aan7b70,45826
-brainstate/transform/__init__.py,sha256=hqef3a4sLQ_Oihuqs8E5IghSLr9o2bS7CWmwRL8jX6E,1887
-brainstate/transform/_autograd.py,sha256=Pj_YxpU52guaxQs1NcB6qDtXgkvaPcoJbuvIF8T-Wmk,23964
-brainstate/transform/_autograd_test.py,sha256=RWriMemIF9FVFUjQh4IHzLhT9LGyd1JXpjXfFZKHn10,38654
-brainstate/transform/_conditions.py,sha256=bdqdHCPCJIpRJxNr0enO2u81924YoIuA8kS8GUGY98g,12970
-brainstate/transform/_conditions_test.py,sha256=hg3gyOk4jn88F_ZYYqqwf6m87N3GlOUE9dC2V3BnMTA,7691
-brainstate/transform/_error_if.py,sha256=0JThfFqt9B3K3H6mS84qecBS22yTi3-FPzviaYacaMY,2597
-brainstate/transform/_error_if_test.py,sha256=kQZujlgr9bYnL-Vf7x4Zfc7jJk7rCLNVu-bsiry40dQ,1874
-brainstate/transform/_jit.py,sha256=sjQHFV8Tt75fpdl12jjPRDPT92_IZxBBJAG4gapdbNQ,11471
-brainstate/transform/_jit_test.py,sha256=5ltT7izh_OS9dcHnRymmVhq01QomjwZGdA8XzwJRLb4,2868
-brainstate/transform/_loop_collect_return.py,sha256=8X6-3T3YoL_Buph9LiGASdrqPnRhsgsH9GQg1wcRos0,20800
-brainstate/transform/_loop_no_collection.py,sha256=p2vHoNNesDH2cM7b5LgLzSv90M8iNQPkRZEl0jhf7yA,6476
-brainstate/transform/_make_jaxpr.py,sha256=ZkrOZu4_0xcILuPUA3RFEkorJ-xbDuDtXorJI_qVThE,30450
-brainstate/transform/_make_jaxpr_test.py,sha256=K3vRUBroDTCCx0lnmhgHtgrlWvWglJO2f1K2phTvU70,3819
-brainstate/transform/_mapping.py,sha256=G9XUsD1xKLCprwwE0wv0gSXS0NYZ-ZIsv-PKKRlOoTA,3821
-brainstate/transform/_progress_bar.py,sha256=VGoRZPRBmB8ELNwLc6c7S8QhUUTvn0FY46IbBm9cuYM,3502
-brainstate/transform/_unvmap.py,sha256=8Se_23QrwDdcJpFcUnnMgD6EP-4XylbhP9K5TDhW358,3311
-brainstate-0.0.2.post20240913.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
-brainstate-0.0.2.post20240913.dist-info/METADATA,sha256=DRuco413fBXd012b5O_sUgatBsggJA_i2aPdBSpn0PU,3311
-brainstate-0.0.2.post20240913.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-brainstate-0.0.2.post20240913.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
-brainstate-0.0.2.post20240913.dist-info/RECORD,,