brainstate 0.1.0.post20241122__py2.py3-none-any.whl → 0.1.0.post20241129__py2.py3-none-any.whl

brainstate/optim/_optax_optimizer.py

@@ -27,6 +27,7 @@ optax_installed = importlib.util.find_spec('optax') is not None
 
 __all__ = [
     'OptaxOptimizer',
+    'LBFGS',
 ]
 
 

{brainstate-0.1.0.post20241122.dist-info → brainstate-0.1.0.post20241129.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainstate
-Version: 0.1.0.post20241122
+Version: 0.1.0.post20241129
 Summary: A ``State``-based Transformation System for Program Compilation and Augmentation.
 Home-page: https://github.com/chaobrain/brainstate
 Author: BrainState Developers
@@ -31,7 +31,7 @@ License-File: LICENSE
 Requires-Dist: jax
 Requires-Dist: jaxlib
 Requires-Dist: numpy
-Requires-Dist: brainunit (>=0.0.2)
+Requires-Dist: brainunit (>=0.0.3)
 Provides-Extra: cpu
 Requires-Dist: jaxlib ; extra == 'cpu'
 Provides-Extra: cuda12

{brainstate-0.1.0.post20241122.dist-info → brainstate-0.1.0.post20241129.dist-info}/RECORD

@@ -1,7 +1,7 @@
 benchmark/COBA_2005.py,sha256=Q8PsZ0lxu14jsF3bCtlZW35iQB8S2_oFEUYQzK2hPiA,5561
 benchmark/CUBA_2005.py,sha256=_W94yOMh2ueqblU4ItEPeTLwHF0_lbEWlVNEBy0Tix0,6222
 brainstate/__init__.py,sha256=r7C3eLTg8LEusoH6PGgBFFt4ZgbketYLoLA0lQhUCsE,2098
-brainstate/_state.py,sha256=L-wDWR04ON1VAqBDv0cHdGM-DVK9lHdH1p2LCKMWoJA,28802
+brainstate/_state.py,sha256=0h3B32130Tvv41eeolYTbEGT9FZ9WUZ9yYoGyVnte_c,28808
 brainstate/_state_test.py,sha256=1boTp1w8DiCFLsPwNtlLrlIqGRpkasAmLid5bv2fgP4,2223
 brainstate/_utils.py,sha256=uJ6WWKq3yb05ZdktCQGLWOXsOJveL1H9pR7eev70Jes,1693
 brainstate/environ.py,sha256=G6r_rqfbofRbjFFalRu_DHaL7ruFTeLRXBQDXM6P-tQ,17477
@@ -12,8 +12,8 @@ brainstate/surrogate.py,sha256=YaY6RJ6kzpuPXWFjaWsxWt2MzJfdm5v_jeOR8V_jPoU,48369
 brainstate/transform.py,sha256=uryXL2vAD-oApUNvL4n6uzLPRwXXARCJ6lRwco_Gh_Y,772
 brainstate/typing.py,sha256=XaqpkfkMxBYkze5DoCQ48GSxuyO0C8rGOKeoN9qeZNo,10425
 brainstate/augment/__init__.py,sha256=BtXIBel7GbttmfBX6grxOxl0IiOJxLEa7qCGAXumamE,1286
-brainstate/augment/_autograd.py,sha256=tQ58yr4O5UXqbSsMIdab94hvymZzqNSUvTBMcFY7sEQ,25493
-brainstate/augment/_autograd_test.py,sha256=WIgUCENzW7pdtsR0PCHjOJnAz4eI1Ha9wlQsS6V8Yjc,41283
+brainstate/augment/_autograd.py,sha256=o9ivoEY7BmtdM1XmzdMmeRXpj6Tvn5xNB8LSGp2HKC8,25238
+brainstate/augment/_autograd_test.py,sha256=S2eEgrwTzdSi3u2nKE3u37WSThosLwx1WCP9ptJAGKo,44060
 brainstate/augment/_eval_shape.py,sha256=dGlRVHOAZ9LSRZsFi1erxgEWHrnhBO3Kq3WW11-Hvng,3819
 brainstate/augment/_eval_shape_test.py,sha256=1nnxbU7hPRbZPQWNWbQ518pw-H7FGDKKnQpZGBY9uRI,1390
 brainstate/augment/_mapping.py,sha256=cpxzVGCEYnP5jPqrowYoPXciw_-QR2F3wggrRj1OCPc,21850
@@ -37,18 +37,19 @@ brainstate/compile/_make_jaxpr_test.py,sha256=qJUtkyj50JQ6f4UJbOLhvRdkbNn3NSKibF
 brainstate/compile/_progress_bar.py,sha256=LML4DjrLSIeGYJWLjqy6BnHSz03fu1gnjf-7kljP384,3824
 brainstate/compile/_unvmap.py,sha256=ewbLLNXiI_dBsEBaVzSS0BEXNol22sd9gMzk606lSkM,4139
 brainstate/compile/_util.py,sha256=aCvkTV--g4NsqcodTdBAISt4EwgezCbKzNUV58n-Q_Y,6304
-brainstate/event/__init__.py,sha256=TFAxDZPHq5KvEcXXSH3jJkipMBZ4OSicoLX1CKhq3EY,1192
-brainstate/event/_csr.py,sha256=4PVSK6QuuDK6pfA5SXvUU4Cxwkd-alGJ2u7VI0irwaQ,10718
+brainstate/event/__init__.py,sha256=wOBkq7kDg90M8Y9FuoXRlSEuu1ZzbIhCJ1dHeLqN6_Q,1194
 brainstate/event/_csr_benchmark.py,sha256=xrj2DSWzw0pUHAE1jRBeSRhMW7ogXvDHEdeaZGioNE4,702
-brainstate/event/_csr_test.py,sha256=qx3TZKnvIC5grSubDFRp7vubu2RJGeCDZFPmof-bRiA,3874
-brainstate/event/_fixed_probability.py,sha256=uFJBlE3hh_QHl3TvNy061xMuLCykuAJGOP3kc-YfH2w,25187
-brainstate/event/_fixed_probability_benchmark.py,sha256=_F_8fH5MNMJZHeSqnq9DYMI9OgYr6JIxBKjbsgeWRv4,4720
-brainstate/event/_fixed_probability_test.py,sha256=jijrtJ5fnwcLmA7Tjd3vDlzwfbftmLoVTN4-MPuogVc,4201
-brainstate/event/_linear.py,sha256=n1NxBUQHWPHBp_xjwpLvtKW8RMvjR5cRjdV_fh_2gGo,11788
-brainstate/event/_linear_benckmark.py,sha256=hu0WqYMIa3jMoH7Fq9dgxcBjjXGFhghPx9vztyCo1KY,2411
-brainstate/event/_linear_test.py,sha256=kuzKCqAfteXkAoBRBIfTrHXacWhQteCkn6YOb9Ij9Rw,3733
+brainstate/event/_csr_mv.py,sha256=4PVSK6QuuDK6pfA5SXvUU4Cxwkd-alGJ2u7VI0irwaQ,10718
+brainstate/event/_csr_mv_benchmark.py,sha256=xrj2DSWzw0pUHAE1jRBeSRhMW7ogXvDHEdeaZGioNE4,702
+brainstate/event/_csr_mv_test.py,sha256=qx3TZKnvIC5grSubDFRp7vubu2RJGeCDZFPmof-bRiA,3874
+brainstate/event/_fixedprob_mv.py,sha256=uFJBlE3hh_QHl3TvNy061xMuLCykuAJGOP3kc-YfH2w,25187
+brainstate/event/_fixedprob_mv_benchmark.py,sha256=_F_8fH5MNMJZHeSqnq9DYMI9OgYr6JIxBKjbsgeWRv4,4720
+brainstate/event/_fixedprob_mv_test.py,sha256=jijrtJ5fnwcLmA7Tjd3vDlzwfbftmLoVTN4-MPuogVc,4201
+brainstate/event/_linear_mv.py,sha256=n1NxBUQHWPHBp_xjwpLvtKW8RMvjR5cRjdV_fh_2gGo,11788
+brainstate/event/_linear_mv_benckmark.py,sha256=hu0WqYMIa3jMoH7Fq9dgxcBjjXGFhghPx9vztyCo1KY,2411
+brainstate/event/_linear_mv_test.py,sha256=V9w41ZP2vu95CyCdCkm-j9Eftqs2kqmeBY809N1-syY,3736
 brainstate/event/_misc.py,sha256=8IpPooXjF2m0-tuo3pGHqThq2yLSNmYziy_zdurZ3NI,1040
-brainstate/event/_xla_custom_op.py,sha256=FquJ7DZ-gzsmHjjQ3PwV8GvtONMb8Y9ysSqJZv22FTg,10939
+brainstate/event/_xla_custom_op.py,sha256=QB4jz_fUEPF-efJCVKAxwx8U79AqdcKoEg2QrGwot8I,10864
 brainstate/event/_xla_custom_op_test.py,sha256=rnkGMleXzLfJj4y5QqwfBvCCLTAHe_uabwBDniY-URM,1745
 brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
 brainstate/functional/_activations.py,sha256=tsCGoV_35IyZ0mM6_4fV9v0-Vj3V9Qm55U8wSpZ_E4o,22180
@@ -97,7 +98,7 @@ brainstate/nn/_dynamics/_state_delay.py,sha256=nZYGmVKmQvAQu-W4YOUFH1gnr-ZS3rg_G
 brainstate/nn/_dynamics/_synouts.py,sha256=9TGAc-nVa50th7KKn4oKLbro-4W4rwxYvp-eu7ksAIE,4491
 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=jxqce-lcH5P0AotHZKhdp_Ho1n7qUuvSe6-CfledOEc,17298
+brainstate/nn/_elementwise/_dropout.py,sha256=lXy8ki5mnmhU9lacJGesy7mums0qH1pplXD6BGGdw-I,17338
 brainstate/nn/_elementwise/_dropout_test.py,sha256=Qn7xqZOyZMPCGF6tFjTiPId0yELOXjSsW5-hgihP3fE,4383
 brainstate/nn/_elementwise/_elementwise.py,sha256=om-KpwDTk5yFG5KBYXXHquRLV7s28_FJjk-omvyMyvQ,33342
 brainstate/nn/_elementwise/_elementwise_test.py,sha256=SZI9jB39sZ5SO1dpWGW-PhodthwN0GU9FY1nqf2fWcs,5341
@@ -105,9 +106,9 @@ brainstate/nn/_interaction/__init__.py,sha256=TTY_SeNrdx4VnUSw6vdyl02OHdS9Qs15cW
 brainstate/nn/_interaction/_conv.py,sha256=LgWO4TeKRru07UEUga3YX6xog6WHtOvKdKtgxGnHUvw,18512
 brainstate/nn/_interaction/_conv_test.py,sha256=fHXRFYnDghFiKre63RqMwIE_gbPKdK34UPhKOz-J3qU,8695
 brainstate/nn/_interaction/_embedding.py,sha256=iK0I1ExKWFa_QzV9UDGj32Ljsmdr1g_LlAtMcusebxU,2187
-brainstate/nn/_interaction/_linear.py,sha256=OO1RwXGLuAHeBplDFB9d9KP04-x3X3PZTBFlTZd96u4,21336
+brainstate/nn/_interaction/_linear.py,sha256=bjiWGJCe81ugQQOykwjWlLW5uhe0CHWwkPA20a4n5YQ,21340
 brainstate/nn/_interaction/_linear_test.py,sha256=KlvFZA0rpyaspf4LT4K7u-RR5jCEB_q1WReqAw9sFcU,1274
-brainstate/nn/_interaction/_normalizations.py,sha256=j0sI8prfK65daQBizPFBHbs57zfGXzlnN2DxKqoN8Wk,14842
+brainstate/nn/_interaction/_normalizations.py,sha256=pdVzuMMmKP49pFbLxZJ6A9m0_nBEMEvOUN4Y1lWnhf0,37004
 brainstate/nn/_interaction/_normalizations_test.py,sha256=2p1Jf8nA999VYGWbvOZfKYlKk6UmL0vaEB76xkXxkXw,2438
 brainstate/nn/_interaction/_poolings.py,sha256=LpwuyeNBVCaVFW7zWc7E-vvlYqx54h46Br5XT6zd_94,47020
 brainstate/nn/_interaction/_poolings_test.py,sha256=wmd5PngZ3E9tNyF3s0xk-DoDR5yFqpTi9A6nbNoIqn4,7429
@@ -115,7 +116,7 @@ brainstate/optim/__init__.py,sha256=7Ao0LCtDNAoxSRSXiLLKnd1_4mR2GSExizpN38il-Fo,
 brainstate/optim/_base.py,sha256=NbP3fzVslfnmJAOWAPD7o9TDWeRdw4CRdNfnfkMTfkU,1873
 brainstate/optim/_lr_scheduler.py,sha256=Tw-aH5knnMh9eAi2LdxkZ6cAk5DbEn0tW0C738xqjFA,15325
 brainstate/optim/_lr_scheduler_test.py,sha256=_amijy9WzuvVWRC4GiuzyaC_278QG97EYZ1WtsE2IyA,1778
-brainstate/optim/_optax_optimizer.py,sha256=cIx1dgjpQCjL_63qqYW9oJtctKyu64QPCiURvEl0IlI,5298
+brainstate/optim/_optax_optimizer.py,sha256=SuXV_xUBfhOw1_C2J5TIpy3dXDtI9VJFaSMLy8hLcXE,5312
 brainstate/optim/_optax_optimizer_test.py,sha256=DAomE8Eu3dn4gh1S3EZ_u4pW4rhcl16vWPbnDcN3Rs4,1762
 brainstate/optim/_sgd_optimizer.py,sha256=NVKYhGcw2D1ksNWUIXZcj-74LUaan8XL3EERk-EHMRI,46008
 brainstate/random/__init__.py,sha256=c5q-RC3grRIjx-HBb2IhKZpi_xzbFmUUxzRAzqfREic,1045
@@ -137,8 +138,8 @@ brainstate/util/_scaling.py,sha256=pc_eM_SZVwkY65I4tJh1ODiHNCoEhsfFXl2zBK0PLAg,7
 brainstate/util/_struct.py,sha256=0exv0oOiSt1hmx20Y4J2-pCGtCTx13WcAlEYSBkyung,17640
 brainstate/util/_tracers.py,sha256=-sX76GJRThdSpDJBejAIzDdBbVhmH6kb-1WoDJVI7V0,2556
 brainstate/util/_visualization.py,sha256=n4ZVz10z7VBqA0cKO6vyHwEMprWJgPeEqtITzDMai2Y,1519
-brainstate-0.1.0.post20241122.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
-brainstate-0.1.0.post20241122.dist-info/METADATA,sha256=br43yWsnDPv-BX93CCJtNljj9BNWylEb1XnL9OwMO6c,3401
-brainstate-0.1.0.post20241122.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-brainstate-0.1.0.post20241122.dist-info/top_level.txt,sha256=MVkn5SZk0qis32_AGxU2Tg1xADfj2IgCNS25CQD7_ng,21
-brainstate-0.1.0.post20241122.dist-info/RECORD,,
+brainstate-0.1.0.post20241129.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
+brainstate-0.1.0.post20241129.dist-info/METADATA,sha256=s2ks57prKGe9kPi_goPGwYVGr40TrgrEFjtC0W8ff6Q,3401
+brainstate-0.1.0.post20241129.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+brainstate-0.1.0.post20241129.dist-info/top_level.txt,sha256=MVkn5SZk0qis32_AGxU2Tg1xADfj2IgCNS25CQD7_ng,21
+brainstate-0.1.0.post20241129.dist-info/RECORD,,