brainstate 0.0.1.post20240623__py2.py3-none-any.whl → 0.0.1.post20240708__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.
Files changed (21) hide show
  1. brainstate/_module.py +6 -11
  2. brainstate/_module_test.py +1 -1
  3. brainstate/_random_for_unit.py +48 -0
  4. brainstate/_state.py +12 -6
  5. brainstate/init/_generic.py +97 -32
  6. brainstate/init/_random_inits.py +17 -7
  7. brainstate/init/_regular_inits.py +8 -7
  8. brainstate/mixin.py +3 -3
  9. brainstate/mixin_test.py +9 -9
  10. brainstate/nn/_projection/_align_post.py +11 -11
  11. brainstate/nn/_projection/_align_pre.py +3 -3
  12. brainstate/random.py +66 -36
  13. brainstate/transform/_jit_error.py +71 -49
  14. brainstate/transform/_jit_error_test.py +55 -0
  15. brainstate/transform/_make_jaxpr.py +10 -5
  16. brainstate/typing.py +2 -0
  17. {brainstate-0.0.1.post20240623.dist-info → brainstate-0.0.1.post20240708.dist-info}/METADATA +1 -1
  18. {brainstate-0.0.1.post20240623.dist-info → brainstate-0.0.1.post20240708.dist-info}/RECORD +21 -19
  19. {brainstate-0.0.1.post20240623.dist-info → brainstate-0.0.1.post20240708.dist-info}/LICENSE +0 -0
  20. {brainstate-0.0.1.post20240623.dist-info → brainstate-0.0.1.post20240708.dist-info}/WHEEL +0 -0
  21. {brainstate-0.0.1.post20240623.dist-info → brainstate-0.0.1.post20240708.dist-info}/top_level.txt +0 -0
{brainstate-0.0.1.post20240623.dist-info → brainstate-0.0.1.post20240708.dist-info}/RECORD RENAMED
@@ -1,16 +1,17 @@
1
1
  brainstate/__init__.py,sha256=DwgnJOghZ_qeFh0a_roiaMCDH-V_F6Ve7by3xjSVrwk,1408
2
- brainstate/_module.py,sha256=RN02rAqgsVVAHeZpXIpZEZSLbfo5YOstmTLlD-JcnN4,52625
3
- brainstate/_module_test.py,sha256=TJlxR4R5bf621y68hTgzTaf0PBN9YmVhwoGKNcpXbpE,7821
4
- brainstate/_state.py,sha256=RWnLjMeaidxWXNAA0X-8mxj4i61j3T8w5KhugACUYhI,11422
2
+ brainstate/_module.py,sha256=UjhfmY26VHQ-kF6U4l68AslGU6vhfD-dR7gF-4Io5ic,52520
3
+ brainstate/_module_test.py,sha256=oQaoaZBTo1o3wHrMEJTInQCc7RdcVs1gcfQGvdSb1SI,7843
4
+ brainstate/_random_for_unit.py,sha256=AiFjAsOO9urfBSL5nNkPpvvNxC3LntyhHmg9FXfRuXs,1906
5
+ brainstate/_state.py,sha256=ykQluBkdKIcQsd7-pU_UlpnByWFof_8TYQl1hGn7HS8,11629
5
6
  brainstate/_state_test.py,sha256=HDdipndRLhEHWEdTmyT1ayEBkbv6qJKykfCWKI6yJ_E,1253
6
7
  brainstate/_utils.py,sha256=RLorgGJkt2BhbX4C-ygd-PPG0wfcGCghjSP93sRvzqM,833
7
8
  brainstate/environ.py,sha256=LwRwnFaTbv8l7nHRIbSV46WzcN7pGLQFhT_xDUox2yA,10240
8
- brainstate/mixin.py,sha256=x4WIYMTCFZgtTp-uiZeNI5J4Qd2BYaV0Ccm_EMdzl9c,10748
9
- brainstate/mixin_test.py,sha256=qDYqhHbHw3aBFW8aHQdPhov29013Eo9TJDF7RW2dapE,2919
10
- brainstate/random.py,sha256=Mi5i0kAsR8C-VoI8LMuIbPPr6YFzq6NBxhJ5K0w2qW4,186392
9
+ brainstate/mixin.py,sha256=f0XovWlyTOQKC63WTtzSbztRSld3pKEHy9-I1gVZWLg,10748
10
+ brainstate/mixin_test.py,sha256=qFLw9Bq8TkOMg8M8CcI92BYvFekXkjyCC9lXSEVa8Ck,2919
11
+ brainstate/random.py,sha256=OMa4739GbrQpKspEx0TbeBqjA4yvwwJzdtpr-kJZN6s,187841
11
12
  brainstate/random_test.py,sha256=cCeuYvlZkCS2_RgG0vipZFNSHG8b-uJ7SXM9SZDCYQM,17866
12
13
  brainstate/surrogate.py,sha256=1kgbn82GSlpReIytIVl29yozk75gkdZv0gTBlixQ4C4,43798
13
- brainstate/typing.py,sha256=Ooweu7c17nYP686fyIeKNomChodSxx_OEpu8QRoB9cY,2180
14
+ brainstate/typing.py,sha256=tZyXpopNtp1UvqbntnmlUP6OBTM_ywyV9QBQ6Gu3IdU,2232
14
15
  brainstate/util.py,sha256=y-6eX1z3EMyg6pfZt4YdDalOnJ3HDAT1IPBCJDp-gQI,19876
15
16
  brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
16
17
  brainstate/functional/_activations.py,sha256=xlwvYG8qvpkfMEZTFxD_4amW63ZfEa8x3vzVH2hDgeY,17791
@@ -19,9 +20,9 @@ brainstate/functional/_others.py,sha256=ifB-l82y7ZB632yLUJOEcpkRY-yOoiJ0mtDOxNil
19
20
  brainstate/functional/_spikes.py,sha256=70qGvo4B--QtxfJMjLwGmk9pVsf2x2YNEEgjT-il_Jw,2574
20
21
  brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1097
21
22
  brainstate/init/_base.py,sha256=jRTmfoUsH_315vW9YMZzyIn2KDAAsv56SplBnvOyBW0,1148
22
- brainstate/init/_generic.py,sha256=OJFS7DHYmZV0JogdsgjnUseUfvTUrAUYiXZynCQqmG4,5163
23
- brainstate/init/_random_inits.py,sha256=STbX-mrHwNuICXkw7EldtJLdUUsWOAcGkEzx2ycV-Yc,15321
24
- brainstate/init/_regular_inits.py,sha256=n-vF-51FM1UcUh-8h5lUk5Jhjrn04KPcGXgGhUGFAAk,3065
23
+ brainstate/init/_generic.py,sha256=yLkmYRWpGHUK_nYx_fWPe74s_Cd-JDdQ6mUqpi4yqcc,7308
24
+ brainstate/init/_random_inits.py,sha256=ycxH9WyKvPhRbk9PkamCLlc9aX5YokREBfSCbrbFQQ4,16021
25
+ brainstate/init/_regular_inits.py,sha256=u77aSM0BkK9VULFJQZ1lIEYA_sJJzEZBTEttBSJ79RI,3090
25
26
  brainstate/nn/__init__.py,sha256=YJHoI8cXKVRS8f2vUl3Zegp5wm0svMz3qo9JmQJiMQk,2162
26
27
  brainstate/nn/_base.py,sha256=lzbZpku3Q2arH6ZaAwjs6bhbV0RcFChxo2UcpnX5t84,8481
27
28
  brainstate/nn/_connections.py,sha256=GSOW2IbpJRHdPyF4nFJ2RPgO8y6SVHT1Gn-pbri9pMk,22970
@@ -37,8 +38,8 @@ brainstate/nn/_rate_rnns.py,sha256=Cebhy57UWzfwrCfq0v2qLDegmb__mXL5ht750y4aTro,1
37
38
  brainstate/nn/_readout.py,sha256=jsQwhVnrJICKw4wFq-Du2AORPb_XXz_tZ4cURcckU-E,4240
38
39
  brainstate/nn/_synouts.py,sha256=gi3EyKlzt4UoyghwvNIr03r7YabZyl1idbq9aYG8zYM,4379
39
40
  brainstate/nn/_projection/__init__.py,sha256=L6svNHTb8BDh2rdX2eYmcx_NdscSdKykkQbzpdCSkTA,1207
40
- brainstate/nn/_projection/_align_post.py,sha256=dAdNsuNf7jo8Qsh3uHXLonv9iDi3J9AnWzmFaG3b3bo,20655
41
- brainstate/nn/_projection/_align_pre.py,sha256=R2U6_RQ_o8y6PWXpozeWE2cx_oQ7WMhhrBR9hZtEBTs,24597
41
+ brainstate/nn/_projection/_align_post.py,sha256=tV_-nizkiRy2B0bBsB4UFUANFUe1IjFZ8hDM9TGpSaU,20655
42
+ brainstate/nn/_projection/_align_pre.py,sha256=_wjdj8muuv2_fSW9m3KBUVjNkBg28BUmz3qZ9IA1rUM,24597
42
43
  brainstate/nn/_projection/_delta.py,sha256=KT8ySo3n_Q_7swzOH-ISDf0x9rjMkiv99H-vqeQZDR8,7122
43
44
  brainstate/nn/_projection/_utils.py,sha256=UcmELOqsINgqJr7eC5BSNNteyZ--1lyGjhUTJfxyMmA,813
44
45
  brainstate/nn/_projection/_vanilla.py,sha256=_bh_DLtF0o33SBtj6IGL8CTanFEtJwfjBrgxBEAmIlg,3397
@@ -52,13 +53,14 @@ brainstate/transform/_autograd_test.py,sha256=RWriMemIF9FVFUjQh4IHzLhT9LGyd1JXpj
52
53
  brainstate/transform/_control.py,sha256=NWceTIuLlj2uGTdNcqBAXgnaLuChOGgAtIXtFn5vdLU,26837
53
54
  brainstate/transform/_controls_test.py,sha256=mPUa_qmXXVxDziAJrPWRBwsGnc3cHR9co08eJB_fJwA,7648
54
55
  brainstate/transform/_jit.py,sha256=sjQHFV8Tt75fpdl12jjPRDPT92_IZxBBJAG4gapdbNQ,11471
55
- brainstate/transform/_jit_error.py,sha256=lO_e5AdhkjozHjM10q0b57OaXbeZ9gQkVmZMN6VQVCw,4450
56
+ brainstate/transform/_jit_error.py,sha256=8rGRx8dtvmPWmHVOsfz30EUMXSix-m2PKM3Ni_9-_7I,4829
57
+ brainstate/transform/_jit_error_test.py,sha256=MGFSCih6GVXTmLh0K-m8RX8N4x2_CAgtcGOIWfzol88,1862
56
58
  brainstate/transform/_jit_test.py,sha256=5ltT7izh_OS9dcHnRymmVhq01QomjwZGdA8XzwJRLb4,2868
57
- brainstate/transform/_make_jaxpr.py,sha256=q3OPy-1Gg0mVaB9pgSTWzzP8FSCAgquSjP-pDEw3Tpg,29970
59
+ brainstate/transform/_make_jaxpr.py,sha256=vMPruKfp5Ugv8RL-9wGfQdSumLZdLtThZvv3sU9MDjE,30426
58
60
  brainstate/transform/_make_jaxpr_test.py,sha256=K3vRUBroDTCCx0lnmhgHtgrlWvWglJO2f1K2phTvU70,3819
59
61
  brainstate/transform/_progress_bar.py,sha256=VGoRZPRBmB8ELNwLc6c7S8QhUUTvn0FY46IbBm9cuYM,3502
60
- brainstate-0.0.1.post20240623.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
61
- brainstate-0.0.1.post20240623.dist-info/METADATA,sha256=ezSoiXSzav8KX7NPa8jfR9aWqVPWwh6d46RZnac_Mdg,3814
62
- brainstate-0.0.1.post20240623.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
63
- brainstate-0.0.1.post20240623.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
64
- brainstate-0.0.1.post20240623.dist-info/RECORD,,
62
+ brainstate-0.0.1.post20240708.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
63
+ brainstate-0.0.1.post20240708.dist-info/METADATA,sha256=hNsfDWI5y7Swfr-xfZLj-JafRPGF_4kvM1NO-Sxvcqw,3814
64
+ brainstate-0.0.1.post20240708.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
65
+ brainstate-0.0.1.post20240708.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
66
+ brainstate-0.0.1.post20240708.dist-info/RECORD,,