hamiltonzero/__init__.py,sha256=YgB8WQVq_9is2ZxMcmmV-Hm14dBQdIGOt0Gskb-L2U8,710
hamiltonzero/checkpoint.py,sha256=57Gr1AEVMjNdpn5PbhuAhxEJuRd1NTtY4lmSX6WRuXE,1652
hamiltonzero/cli.py,sha256=paTVZqBeEaixNrJzWjapOCXuIepayyc1vC2qgz6elb0,2165
hamiltonzero/config.py,sha256=r_CK5OES-KSDSlbMAlCdO_HUBYj69eiTyg6jzqexQKM,9315
hamiltonzero/hamiltonian.py,sha256=EDRv13nFp-4R-KJDi18tldZIv85VSCA9TiRPyTP1awc,10065
hamiltonzero/inference.py,sha256=SuZZcjsfj6E5Hkh_atcRPXOrmUbVyZTVvTXhOJRsU3I,8623
hamiltonzero/observables.py,sha256=Sn0rMdFkwdgtJDHJurxyYE2DsmstmvszXEo-yyxzD6U,1945
hamiltonzero/renyi2.py,sha256=0Jk-yHErcn0nU3HBpJjW-4TV2Wki9sXLpe3iWSurYLI,24508
hamiltonzero/compiled/__init__.py,sha256=V1_yu8p7vZY5oVFL7LuqYuEEh_rKVF4xQR-k6k5ku94,83
hamiltonzero/compiled/api.py,sha256=TekQtBmxglbgpj5gVZa1iH9UX4jvfxVTpnodYp7itIM,1218
hamiltonzero/compiled/execute.py,sha256=5GyJbVxpRRgVA0PPKiSQX2PLVbO866ZMO-8RMWZYZQU,4890
hamiltonzero/compiled/model.py,sha256=dmWW9s2rH4MLjyH33kEEiw49-u7A7Zwd2Ud81Y8j6Ac,15747
hamiltonzero/compiled/tree.py,sha256=SC1EOQfIhmSd-N5z1B_TwToh_YaKqfj_9uyTzYEiwZ8,18425
hamiltonzero/compiled/trunk.py,sha256=YYIdbdcuOSuTj9vmZOctaOtUIc3YyjQXU5teVSQMl7U,4626
hamiltonzero/compiled/types.py,sha256=YyQ1rUreygb3s7AOhW9tFa9yFeDTUFyM8n4xjAHkopM,3732
hamiltonzero/data/__init__.py,sha256=vaP6rMBAnaZskDlD4g9pDNlNpUfr9qhP2OAg2Uiy-0k,440
hamiltonzero/data/systems.py,sha256=TDi6fNlYNJlPwmOu5UrA1Fl7kZxvtpx9kRK5PlN7GCc,8757
hamiltonzero/energy/__init__.py,sha256=Z4H7fuE5VzWEj1ori0CZG9ODbpDZAD1mIlm6uqf0tCE,471
hamiltonzero/energy/compiled.py,sha256=gEicr8nF-qujZ90xWAfYrEFyV3Y5HY3NDBEZkmWeIys,1296
hamiltonzero/energy/custom_lap.py,sha256=OQMnfcb9y2zsSbUi5odXZIxQ9ksXk-1aGJ4iGceclI8,46931
hamiltonzero/energy/frame.py,sha256=B0gQfzRxnAaQxQq7aS84ziWLtMMzF4c6Qwj6hczqDZA,4481
hamiltonzero/energy/kernel.py,sha256=QmiUAeMpH3c9mUEV08iiGAfm9K1G3ASRbyl4uZ-Gk1I,5143
hamiltonzero/evaluation/__init__.py,sha256=URUwK6GAbc-nG2r-64L5-vmFKkB3sbVEO7u5xS57lLo,989
hamiltonzero/evaluation/backend.py,sha256=Y5pp974FGxTHEJ5XIf3IsT72fY-mFm6tXfdgnCGu9Gg,3302
hamiltonzero/evaluation/greedy_router.py,sha256=HvBjeTqSaR8bUmH5aZAFdM6Y75b5ogbOTSR7j_u4EBM,3764
hamiltonzero/evaluation/large_n.py,sha256=PkTtarQIJeee6gGkVelkcbjq4LNcwtq7_uQcobkM1mM,12142
hamiltonzero/evaluation/pallas_mha.py,sha256=ortnoU_6G38xwoZfoiVuMvmtklUepfoKeB2hT1UbXr0,5274
hamiltonzero/evaluation/runner.py,sha256=TLbnqeZDjvKxlXbS_28iYwIfedRy7W1Cym13toZsbeo,16689
hamiltonzero/evaluation/runtime.py,sha256=5b6HPIxGtfZ5RaY9zpbKzFYXrNBsKCu799dZFF5Lkyk,41122
hamiltonzero/evaluation/sequence_parallel.py,sha256=3ShXriwZ9kR-AS3ULkw2Q--kJkSAtnsX-c9g7axy9d4,5797
hamiltonzero/evaluation/sequence_trunk.py,sha256=pIgwC4QL6SPFCQ2vaxRg0b7C-w_FMDlO-8DQ8zquDxI,55735
hamiltonzero/evaluation/statistics.py,sha256=NxzRNh9BSDrFYUnMeVR69q5go8P5sIwRiKwCSABZFLw,7288
hamiltonzero/evaluation/types.py,sha256=MqvxPw_OrcsegnmKOtHoOzmh0zdKgj7BEa2Gin5Axgk,2156
hamiltonzero/mcmc/__init__.py,sha256=zY13KOBVovC_QgbQgGRiWyDZ6No-c39LoD0O6EBST-Q,316
hamiltonzero/mcmc/_monotone_cubic.py,sha256=qXTpEl0oq3hWjRsxzhNYagOoqsgkq3I5zN8R3Jtc2nQ,2382
hamiltonzero/mcmc/langevin.py,sha256=pvQZfJhmUXnyTfycsromXq780qH_-IYNA3B8bYXfLJc,4711
hamiltonzero/mcmc/quaternion.py,sha256=DsZfyOQcCSmz9-AuZEVVNOg6KC88JDRTykZfQ1vd2nU,2331
hamiltonzero/mcmc/replica_exchange.py,sha256=z-K5prlBHVgEyymR4ELqnlr-Z8DPNOZen9sSOH9Zp70,13733
hamiltonzero/mcmc/runtime.py,sha256=tM2dsy84DLIzYJvAT7EvA9ppCtB2o5Lq6m4_pB5nwlY,5068
hamiltonzero/model/__init__.py,sha256=FDgVa4xbCtx8k37p2ASM-qc2Wv4ERwGeNBNcsDmxNCo,883
hamiltonzero/model/_custom_lap_primitives.py,sha256=KpqJb1uT2DGXw4BB8Bc4iAKA_8xt-qjaUXGqpP45KKg,2660
hamiltonzero/model/api.py,sha256=VLEFkLC_Pyz63sctgT0CSov-diVIchFUxtBitfDqdzw,6892
hamiltonzero/model/context.py,sha256=HK00MNcbsaWvsexfthkgCvWcogXTD2iowHZZKcn5Dtg,13796
hamiltonzero/model/featurizer.py,sha256=wj1TXoE1LzIeVlXajozwyT77YzwGTlg0MYtZOZw7mHU,43696
hamiltonzero/model/fp32.py,sha256=zU3OBrXCW7f8OkFq4cuPke0W7zlHzntRxF8SOLp-HDM,201
hamiltonzero/model/fused_silu.py,sha256=vOnHCHSdUxDHgIQfQ-OK_68L4GgQLzn00uQPA4BYKYk,169
hamiltonzero/model/global_ladder.py,sha256=SmX8JlQpyG7RLNfGTwRGIBGW3EhDgArDJ0dLTI8im4s,14825
hamiltonzero/model/model.py,sha256=lIZeC5qvtgqZBsyhOcuUfTvqfkpYiFaPCDLTAvv4Fvc,24003
hamiltonzero/model/odd_ops.py,sha256=kBKypAcCHAB3drp-jOJxB4weuTa34fy271w2JFrsGKY,11045
hamiltonzero/model/pallas_attention.py,sha256=zVYr-Gyxk4ZOf5PbtDazhDt23l0nhc15Ri7m-4IQGE0,2246
hamiltonzero/model/readout_leaf_context.py,sha256=zmnD4gwtoj7eJ_f0jisKTD-mUnVC3DrOOHevYgxiXQ8,43196
hamiltonzero/model/route_pointer.py,sha256=zZQF9U1euM68bA0or-1vWFXEo0wGelD22DOcuoHL8Tk,215390
hamiltonzero/model/route_quotient.py,sha256=RuzBTgzklpMpw7UbDB8XKkTpQ1wzLa9ia35ml7esfX4,11267
hamiltonzero/model/tree.py,sha256=mbW0j6mzshHPTJ94uKreGxaM-3MWIbTTaPmO04pIzeU,78877
hamiltonzero/model/trunk.py,sha256=fzFuasxH3Ozf3j_h87drtZHvQA3Oxj-O4pE6dY85DMc,17770
hamiltonzero/model/_pallas_attn/__init__.py,sha256=wvGwqdg3KWxsiEf217ry7g7fHB7YB1-t5efms4ZhnaI,164
hamiltonzero/model/_pallas_attn/custom_gradients.py,sha256=7L5ZOMy1Y8s0ORT6FsjO8mQxHmr7Edg5NBMAbEVZqtU,11477
hamiltonzero/model/_pallas_attn/mhsea.py,sha256=71yFt7Khyi4Ls528kr8M_H-OLRlrvs6ed1V8kjNIZfE,1249
hamiltonzero/model/_pallas_attn/mhsea_full_ad.py,sha256=CbG9aCtVR2gEwL5ooFUGbYrJDDy8UpstcNwSg3Snsuo,5806
hamiltonzero/model/_pallas_attn/mhsea_full_ad_jvp.py,sha256=RZFrjdQEtGawpIGTYSrT2ZVjCqg2buEcUSF5ZyBFc9E,3383
hamiltonzero/model/_pallas_attn/mhsea_tuned.py,sha256=eF-8-3BKv0TgwiAsv1jU3ApxtlrFAtHAlLw37W-u--A,8166
hamiltonzero/model/_pallas_attn/mhsea_tuned_jvp.py,sha256=NaUNVDDEA46AXIkDRV2l8GkgmbNqo5M1v9kSBrGdK20,7550
hamiltonzero/model/_pallas_attn/utils.py,sha256=oSis6HpsYpKdLLkgHFWYvm50t7tVHCcxMiIE6xafgnc,1936
hamiltonzero/modes/__init__.py,sha256=V1_yu8p7vZY5oVFL7LuqYuEEh_rKVF4xQR-k6k5ku94,83
hamiltonzero/modes/eval.py,sha256=TVo9UL94qRabbPMdZ__r6Vl0A94bT0uJSsrXU8h6S1s,1671
hamiltonzero/modes/finetune.py,sha256=Ax0CWb_PhS1BA6Gh82mlVCapUz1jbDWafcO3kBIXNWQ,12561
hamiltonzero/modes/train.py,sha256=mYCg0Y63b-uZJcFrcnkVZ8pVqQWMeg5SXtw8dhSl67E,30323
hamiltonzero/optim/__init__.py,sha256=t1R8Ck6PtvKBEavuqjt8vkJCNZuqGIV_shZU7GKfNUQ,614
hamiltonzero/optim/api.py,sha256=XQyCg5TQghj-ZeoEPb1MCZaDNOPKvHk1uDd0wn_nWHk,1114
hamiltonzero/optim/blocks.py,sha256=UHmQFM7eefKKgI2tYY3-sfVOL0JpHQcv1adgM_5oUw8,55699
hamiltonzero/optim/compat.py,sha256=FqIkE8Q47DAPr-7AR6AzKphaB2koUk38_69dNeca4Q4,34788
hamiltonzero/optim/production.py,sha256=5TsXID9Ebn1bh5_9z9sijNbAjXdbL1T-tHXiVT-SyVM,13839
hamiltonzero/optim/spin_blocks.py,sha256=k4DiS0lf5yWo3odA7ToWkdC6xfl8cJI7xwgw7HIpFnU,26003
hamiltonzero/optim/targets.py,sha256=2_gsaLbFk6NfKGhpheNaqs45ZzdNo8OvnC-Kn5OoJW0,3633
hamiltonzero/router/__init__.py,sha256=eGibGoRuENBtexyK3K6DYsbVN4cX2OSJQYrf7hJ0DO8,1002
hamiltonzero/router/api.py,sha256=0FJEIiXTxXgtw-USy9xJ85X6n2T5Uu7dd-kIWM466BQ,2313
hamiltonzero/router/baseline.py,sha256=9aWUYZ7UCZCHYbbWMrRj59SlKsTDZPr9lSK3yomRE_A,654
hamiltonzero/router/compiled.py,sha256=8BViaBgGKcyBtR-D8Ukkr7q8_OM_VkTNVdQELprvlLM,2839
hamiltonzero/router/decode.py,sha256=NybiYc3awtEXMT_Syca4nn7MH8eToMtz4OIGzmQ4y74,3559
hamiltonzero/router/permutation.py,sha256=rap6MCmFZHyIBK-jojnJb4vQzPxOHLXZNGRJ4rE3h9s,2163
hamiltonzero/router/state.py,sha256=u23ZvzBs0MpPuf9Da39ZuxpxtCgikXyS16oXHMM1Xzo,1321
hamiltonzero/router/types.py,sha256=eAgYXhIsbQQc0egtFsMiaFToNVyxpnMTrT6_5VLbh68,899
hamiltonzero-0.1.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
hamiltonzero-0.1.1.dist-info/licenses/NOTICE,sha256=UIPaJDeQ7HSZ5d4eDVzhzfD7eWJP3Iker_a86_lCEDA,165
hamiltonzero-0.1.1.dist-info/licenses/THIRD_PARTY_NOTICES.md,sha256=voCdk-8cfAS3twiSYNyFBwfVnZUcRn1NTVYI7REHls8,602
hamiltonzero-0.1.1.dist-info/licenses/third_party/folx/LICENSE,sha256=ffINzfkZfplFwUhY1Bxg8RtSuT5baeK2NBa4dNWY0yI,1074
hamiltonzero-0.1.1.dist-info/licenses/third_party/jax/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
hamiltonzero-0.1.1.dist-info/licenses/third_party/kfac_jax/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
kfac_jax/__init__.py,sha256=TnMiUENOBTpmRMMoHvs-nFVbVZcHxFlPBnA5_gMZ9HA,7768
kfac_jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
kfac_jax/_src/__init__.py,sha256=WIVmD14Sr1HfmJddHhkHkq6FPL1k3Dpk09EDS0ET13M,615
kfac_jax/_src/layers_and_loss_tags.py,sha256=K3g3kbS2Vqhch0lRR5cb1yUYfhDDR-Lk45O2McYwJS8,15357
kfac_jax/_src/loss_functions.py,sha256=rJe-xwqogdYh8Knkgdmgx4TvXjax0fISUqHyG2MblgI,50723
kfac_jax/_src/optimizer.py,sha256=LiZGQty5jEPECJn_vCbbIkxyWxjxD57TZBSlAzbKRLQ,87359
kfac_jax/_src/patches_second_moment.py,sha256=yTEmMJ6q4IBTP96gozlOdolvSlQaBRptA-ugkRt-Cjw,36845
kfac_jax/_src/tag_graph_matcher.py,sha256=hbOzE0z_SzZaUGMUHUJHmo2DdH6HOjM0ZHBFtY46kFM,72051
kfac_jax/_src/tracer.py,sha256=LgQzA1xAiP7h1MN6g7G6Ir-wvkyfXt0w2VetrI5orRM,50791
kfac_jax/_src/curvature_blocks/__init__.py,sha256=z-ki2bsMWcL5p1rgDiYuMEDVt-jNO6uK_TX5mNBBUYI,2208
kfac_jax/_src/curvature_blocks/curvature_block.py,sha256=PPnJD2JVmI00RxhubERlH4-uUW628ubWVfsdQwNQJzk,20948
kfac_jax/_src/curvature_blocks/diagonal.py,sha256=LPhm9Rc-wganJ_twnI_2Bcd9azjMSTC0JRZz8ZIHNYw,11417
kfac_jax/_src/curvature_blocks/full.py,sha256=bqu9QLngPzp1aoJCeSNYiPicPepAJq4lB9cSLyPzreQ,17067
kfac_jax/_src/curvature_blocks/kronecker_factored.py,sha256=GggvVHqyEmPms-ttX79cjNT2saMdvZLoCUZeydyfpXY,22469
kfac_jax/_src/curvature_blocks/tnt.py,sha256=zW5uDxpnm8XRNe3CyWa9QpX_2fEJm9y1EsaEUB39dEU,9071
kfac_jax/_src/curvature_blocks/utils.py,sha256=Gw6245bTqlNNMW8cKiov15QPkVu7UzcYrRbQ4oP7jGE,4836
kfac_jax/_src/curvature_estimator/__init__.py,sha256=1YKWFoMudZL8Eh05KPhkDgc1ewms0psYqGJLFmsm9Z8,4066
kfac_jax/_src/curvature_estimator/block_diagonal.py,sha256=J4mdeJ-ecemJJgGWudvShcFU2yAp3SXZM9fRTSTJ7kE,40466
kfac_jax/_src/curvature_estimator/curvature_estimator.py,sha256=yK9gHmqDjtONO6EukEPjK6mJ0SUk7U9SKVvMtcxWVQc,14415
kfac_jax/_src/curvature_estimator/explicit_exact.py,sha256=MTbM-IgGIHjAXkYGO8S_52WpxjTMU9S8WX24YiG_ITA,6366
kfac_jax/_src/curvature_estimator/implicit_exact.py,sha256=_HFPle9NWUoZuFKl732-2AegAdb4IMYR0k8hmc-4E0M,17161
kfac_jax/_src/curvature_estimator/optax_interface.py,sha256=pLnauYcH1puixGvmVSRsZxlsqkvPn3SOwiroHAKs5_A,19232
kfac_jax/_src/utils/__init__.py,sha256=D3b3d5V4WHYo3bcKKDbbDHrc0XFMz-q762TFMY-gwYY,5592
kfac_jax/_src/utils/accumulators.py,sha256=Gk1XjftIuvo_00fwyIikiPcMH7STNe1V_XkJlc_CtVI,9462
kfac_jax/_src/utils/math.py,sha256=jjz9s9vMJUjfu2tMu9SB3Spf3gFKLnoY_fgPNRyquxg,40910
kfac_jax/_src/utils/misc.py,sha256=NkDeCuIQE1MHTT4Xq6fyDdsJLlf0_MzSv9cJUcjWOew,13204
kfac_jax/_src/utils/parallel.py,sha256=LZZNPppIxwKeT5k6lRyVADAuI31hvEHqm16eDTaF5co,12963
kfac_jax/_src/utils/staging.py,sha256=cKIKlQEeviOG7LVu7FgCZTYIie3CwGkykdnBN8_H3ww,13597
kfac_jax/_src/utils/types.py,sha256=oIiAMQWNnBgC2MW2T0uO3c4xUiapP7GsMoqwUsP2u5Q,2832
hamiltonzero-0.1.1.dist-info/METADATA,sha256=MFRwx0U04Z9yq-QIda65_Ss39JIByWc9khAAbfoNRpU,13022
hamiltonzero-0.1.1.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
hamiltonzero-0.1.1.dist-info/entry_points.txt,sha256=Z9L2eleeqRwIkcfrKoPH1JnJEZTZYmFeARXXf1FigU0,55
hamiltonzero-0.1.1.dist-info/top_level.txt,sha256=AdC6jb8zKos3mnYNDMhKGS8T39KwurLhFCYwN1t8ulE,22
hamiltonzero-0.1.1.dist-info/RECORD,,
