tokamax/__init__.py,sha256=lt_MkdLh8QtdbJ_xdmTDX0XDA5lHG7PHopPfpQcrfjs,2810
tokamax/autotuning.py,sha256=Z3KLjS9wXkd49fTz0jyH8EGDPh7tTARSkcq3fk4jMUc,1059
tokamax/benchmarking.py,sha256=sJxtRSAuHgW4JbGMyHw07vvUwRR_z8NRVYGVHCZrehw,1097
tokamax/config.py,sha256=-YyhVr-cZVKqjTcjLSNKfdxS2cmxPvSlYtbCnLF09rw,956
tokamax/conftest.py,sha256=qjs1VrvMMZeefevbOkpj-TtXSqT2sZGJj-nQHH7Jrrg,1269
tokamax/tokamax_test.py,sha256=y4CYurAN3ypr-HajbeGkG42zF-BbS9AHtpaK4htbTcU,5609
tokamax/_src/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ad.py,sha256=ztYw3M7ZfoMcjdl4MowW7RHHtnoyK2ZU6lfAKnRVTTY,4334
tokamax/_src/ad_test.py,sha256=DmhH5K_PeUsuqblEvCoDtwVCYS2i0wqdsirRzkp5JYE,3947
tokamax/_src/batching.py,sha256=SbmyHjbF3l_9363AA0_cacoRxcwbGxl29zJXA4LMAkY,7733
tokamax/_src/batching_test.py,sha256=_5y9wN3Vafu8X6oxpto_2Xj0yJzAfBMP0JU5t4WFDvo,4693
tokamax/_src/benchmarking.py,sha256=KAWpqaDR-F0Yvk5FPUuqcWXbvfLIpHUM6syBfVHJShE,27519
tokamax/_src/benchmarking_test.py,sha256=mmCEjemEFXkS82D0IfzYu02eJpwMe3_Yr3mFJ6oCeOI,9757
tokamax/_src/config.py,sha256=dvHfkVdRxSq_j0eJqbFivjs49egNho9j2MUBfj43-eU,4538
tokamax/_src/config_test.py,sha256=ZShG9IQj0YbyWp4ontfZi8c3YWTyxotsd4eYlEfE0OQ,3946
tokamax/_src/gpu_utils.py,sha256=5kQyDWEMaSfy9lhAsRfp_Rvr3GOi9bn43qXfnux72w8,3398
tokamax/_src/gpu_utils_test.py,sha256=bh70qgHkuFWK64SfpT_HRSPVskpf1b2zPmGOVx4c5jk,2647
tokamax/_src/hlo_utils.py,sha256=Rv9m6_sYHzm9voWmyb1t4gk-DBLEntPwtB62LKJRH5w,11227
tokamax/_src/hlo_utils_common.py,sha256=25DVWKAOO6j4d-PnjsgTWSATHHdAOORE00Yd0JqViuE,4173
tokamax/_src/hlo_utils_common_test.py,sha256=uSNEu6beRJ3hNrPiyh-mpY_qwqTHJi3N0m72IHwdqqU,1752
tokamax/_src/hlo_utils_test.py,sha256=XuJihTw7OugJvirWe3PjvsktqHCnCSScE858rSfPr_k,18909
tokamax/_src/jaxtyping.py,sha256=mXl08fcQMMmiZVQ2I7IKgDONrL1GMBcl3f-WWLemabE,1863
tokamax/_src/jaxtyping_test.py,sha256=vDDdV7JyVds-SpyReKiqZr1-WbAT-3qRnXloYLVRYGE,1558
tokamax/_src/mosaic_gpu.py,sha256=MxTcOIvmIg3nk8NRtlj0TwNFG70JvB2181-XG45ZsCo,14763
tokamax/_src/mosaic_gpu_test.py,sha256=xL2CZnEavXzO-2D3RXXR-j3Rj82bgbDSCJUikOLE2pM,5593
tokamax/_src/mosaic_tpu.py,sha256=_8b9rZyL9TTPOn_zfAlcCYJELitwL_9V_1rm_2mWGIA,10481
tokamax/_src/numerics.py,sha256=eQhfZOJt2qsqncsVwOE2A7wtCLZ22CvwtE4K6Vrbwdw,9444
tokamax/_src/numerics_test.py,sha256=-SifpMeHXIWpBi3P5QcGa4qv_q5SuambZiI4Zp2-ZV8,7116
tokamax/_src/precision.py,sha256=7y0HcRF6wzgtJLYv57U4JX1XePE4ZORaSck7sopgcVA,6422
tokamax/_src/precision_test.py,sha256=fFifAxvC3-hv7tY4qLXUj6313laQkeIB9AN0c4uFxC8,3630
tokamax/_src/pydantic.py,sha256=bH-lMb8LRUBt-utI4ILv-j_SYdlvp8yStAlV4JykZ1w,13733
tokamax/_src/pydantic_test.py,sha256=q1ze1qCMaBp9_3NAtZzAGNE27t1RRX2b4jgJ29eLpWM,10313
tokamax/_src/quantization.py,sha256=Z2BpXz-zxHaDEIZmv2zeT0PcpTpsKY-8Gg52W0_COU0,2881
tokamax/_src/shape.py,sha256=38jE2TkGGqnYI_hETi-tMMAeuaVYIvz-yWTSxLIQv0o,2987
tokamax/_src/shape_test.py,sha256=8C6awpma4oJzPlcIpqwVvf5SSfKmATfghDA6hZkr5Os,1560
tokamax/_src/test_utils.py,sha256=fbvLriizN3l4LGadlORcIUvM91WaekPakIOPnZ7gb8c,3606
tokamax/_src/test_utils_test.py,sha256=O1457xfEX1w4v5X8tW8s7SLCWNWZ9l_lxz5tausF_48,2097
tokamax/_src/triton_utils.py,sha256=eUzLJrVSd8esqOL7Uqkc5cdAcVwE1K857Sucd4R01IY,6410
tokamax/_src/utils.py,sha256=M99bvel6cptkiocbsR2WHEAwe-QuqBbZ7izFhRcl9is,1898
tokamax/_src/version.py,sha256=ZY1lg7EtEM2xUmuXs8Ne5M-wpt0qqYuWPlWDS5Xlz7g,1213
tokamax/_src/autotuning/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/autotuning/api.py,sha256=Lk6LOLT29BZtGCg08-0yZ9z_-oubYlFXsGEjjwqSqas,17495
tokamax/_src/autotuning/api_test.py,sha256=QYZm3NQMc_73HaLT64X783qa5muZDTva4EzMuExOjVk,18810
tokamax/_src/autotuning/arg_spec.py,sha256=Risra5joSM7piwKprTO587b-_ERckvaxT1bQxwTmJ2Y,2475
tokamax/_src/autotuning/autotuner.py,sha256=FIwJs4arHH2BpXusiui9ea7uJhunLfOI7PysRMkcPfQ,8271
tokamax/_src/autotuning/cache.py,sha256=5SAjGxOD887eW4PIVi9ZEe2-zZNhYP_HKyPHSSOw5Rg,3778
tokamax/_src/autotuning/cache_test.py,sha256=1z2SuwoL0md0dZUtOnEHDHoGwRvCHYnf3zv4Vuy8mJI,3921
tokamax/_src/ops/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/op.py,sha256=NfL6QudU61v09m6u4nCIM1oqUl7q_D4XzU4jHJNjh4c,25909
tokamax/_src/ops/op_test.py,sha256=O8HTODmJzvAihg7H5CAcm2TbYwMllauN9jiZrr1ZtyI,11996
tokamax/_src/ops/attention/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/attention/api.py,sha256=neJXoFawWRu5xD92nWKfxCh8D0rAeA6r2MlkKIu4Maw,7003
tokamax/_src/ops/attention/api_sharding_test.py,sha256=oTMOSNzGFLOmtA0u7DAyWeANAWjABSBrVOxqjwgd8MI,3310
tokamax/_src/ops/attention/api_test.py,sha256=0s9Yod7PsDBEtCnuc-XzUn-f57gjoxvKT7jXAwttxN8,15684
tokamax/_src/ops/attention/arg_specs.py,sha256=1lR-ABTzWcBePhBGXb5HEySf1F1N9u4Py3m6pf0APDY,6593
tokamax/_src/ops/attention/base.py,sha256=Si4MSQr80Wipk4kyagIXXtH2HLTffE1UOsxrqbSians,37050
tokamax/_src/ops/attention/base_test.py,sha256=IWE370cFZsO261x0Q1dz1Yl_Dujh5jJaE5jYtY7HoOo,6754
tokamax/_src/ops/attention/bench.py,sha256=gdw893Vr5J18zASTcNa3Tc8LpfIWXVcE5YG7j9Er7KM,2501
tokamax/_src/ops/attention/jax_nn.py,sha256=7-5PaZNg3NAoSI6ZUBOVIhz4ja_mGepPtaT_orDKcHM,6010
tokamax/_src/ops/attention/jax_nn_test.py,sha256=4n03_GUpWW6wQa__TOVd6Lt-h2z7RItcLAmd8bihWCs,3886
tokamax/_src/ops/attention/pallas_mosaic_gpu.py,sha256=vimidsvEZw38tog8KKFKLzg6Yv8r2sRMAul_7yGMmT0,9107
tokamax/_src/ops/attention/pallas_mosaic_gpu_common.py,sha256=QgwjR237zows5xLucWKNU0_LLKKGjlShEg4v6lHoDrE,11471
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm100.py,sha256=7_k6jk-KoiKYL5W9SCxaq4w10cYV0EP-MKznJYoMnts,37931
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm80.py,sha256=hOxpphBcAI2OWL1yP_49JRI2cfQKv7OB7GZDRaCgi18,23588
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm90.py,sha256=olv5yRFvWXk0h-HDxjTF2Rd-9ryLc62F2kAet7KMwpc,23804
tokamax/_src/ops/attention/pallas_mosaic_gpu_test.py,sha256=KthQCJvxsfdRFmQVJEe0GFtRI_hs68AvXnTTSW4kgys,9534
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp.py,sha256=Mkcys_G58YizGxb7hNnDz2N-B6EUUV0ZZjCPA-W__so,7255
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_common.py,sha256=1vyg5qbP4IJDfDdGhZoCsglbEIJHiusp307LLqE4dPw,1629
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_kernel_sm100.py,sha256=JPfgmExqN94CPiLh0vJLUKsq4UMbK0yxXCfj78V2tgw,34898
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_kernel_sm90.py,sha256=a3dxU7f8CJwq_L0gFUMdi11q7wAI_rdqJh7f_Ix6hnU,25612
tokamax/_src/ops/attention/pallas_mosaic_tpu.py,sha256=XGq9oZSoZxGSAPNrQaMjHkmgW8DNy_oQ1l-pOA0YEkg,8025
tokamax/_src/ops/attention/pallas_mosaic_tpu_common.py,sha256=TjHgj6TDXG818pk-3mG_FAClbSDVgIVJpEdq2CkCNe4,4278
tokamax/_src/ops/attention/pallas_mosaic_tpu_test.py,sha256=HVcst01ECr3tbkyFbqag6Ru8zIf2eaemg7USmM0G_R8,6928
tokamax/_src/ops/attention/pallas_mosaic_tpu_vjp.py,sha256=GrplOqVhA5vUw8iV0Z7-oC-X2GzLG4xq2_47IWyglg0,8432
tokamax/_src/ops/attention/pallas_triton.py,sha256=rQKrCCAVWVxq86P0bXi3geNNJO-rzpuJpwM9NRtt4eY,24165
tokamax/_src/ops/attention/pallas_triton_test.py,sha256=h4LZ9rukqH5sCRjly_i9_VCHJ2Lstj_PznOORMqOopU,5701
tokamax/_src/ops/attention/pallas_triton_vjp.py,sha256=I2s2EsnNR_Z13a3wVZYidOHryPhgkS4xOn83wtyKbQQ,19076
tokamax/_src/ops/attention/test_base.py,sha256=n3HKw7KXWPa2Hp3wWp23kZZK9MjbdJEg0IiMy_3epR4,39318
tokamax/_src/ops/attention/triton.py,sha256=BF3Egi9QsOMPK2ZkdJ2ITRYgXUMGKbwSGoHHfPfIq90,11157
tokamax/_src/ops/attention/triton_test.py,sha256=oVX19KRmOOFKHHwDhs6lTCiB7MGgcLXcDcXBmLm3k-E,6552
tokamax/_src/ops/attention/triton_vjp.py,sha256=gThrjXn-RrIZZvenMgCk9rUY_q4rY7AzICmCaHg8390,7905
tokamax/_src/ops/attention/xla_chunked.py,sha256=4wgO3SZHNcVuh19thHzi2TOZnVEBg57XMvQeBaFAJqI,12963
tokamax/_src/ops/attention/xla_chunked_test.py,sha256=aQcFiI4yVhIUDGbOv-2dLWfaUuoR6c26E2sXmsz0-3s,8750
tokamax/_src/ops/batched_rpa/__init__.py,sha256=ZXRlLcMbY-OQGSiroo2YyieNsy7ImsY4E9g_R20dQds,696
tokamax/_src/ops/batched_rpa/api.py,sha256=Ntq5WDOpqKS73OH1uNuhsbJL2v_GIoESDhxnqHPdj3U,3944
tokamax/_src/ops/batched_rpa/arg_specs.py,sha256=sZej609WzXROH6pGQ2Z_Gn7CX0G4u5vlPlgiV3QkAw0,6653
tokamax/_src/ops/batched_rpa/base.py,sha256=DoAKAZZ-on2kVqm4aExsWlMUm86Sl8-HGfGvRWs-VdQ,7510
tokamax/_src/ops/batched_rpa/base_test.py,sha256=V10Qv-vLev9lJQP6KHMwSCWDvBog2BH0RGFXnJAKOFQ,6645
tokamax/_src/ops/batched_rpa/pallas_mosaic_tpu.py,sha256=uA9iWLCSbF4WefQt_tYN9ZtLMdguqvEWjGoAtnbPqxc,7527
tokamax/_src/ops/batched_rpa/pallas_mosaic_tpu_test.py,sha256=JjKiy6acn00s2bg-GQdNbyM9bnvv7NNDkCDHLQtKB2A,4249
tokamax/_src/ops/batched_rpa/reference.py,sha256=huA3pRn-Px2E8edPHP9SLYV_T0V0CynqVe2j2Y9tR7w,8029
tokamax/_src/ops/batched_rpa/kernel/__init__.py,sha256=YKJW4pOeOddU5njid32X3hV6RxrTb93syWZgZv4FHEc,950
tokamax/_src/ops/batched_rpa/kernel/bref_override.py,sha256=7mUfVRfHg2gjyIPG8c8gqoX8EXKEqOrzdO3QqQnrQYo,21464
tokamax/_src/ops/batched_rpa/kernel/configs.py,sha256=4vhCWfCi6wrz89LoagHztnRcPSh9qJmhIS75QWiKMiE,20198
tokamax/_src/ops/batched_rpa/kernel/configs_test.py,sha256=Cry3l0XJwgWsz99-4DPy65HRIDXfk3ivE4gUKbng6SE,4019
tokamax/_src/ops/batched_rpa/kernel/correctness_test_per_token_quantization.py,sha256=xLfpYt1AhGZdehs96UwCH2osuWWdAnwGWTOqVHBxEKU,43584
tokamax/_src/ops/batched_rpa/kernel/flash_attention.py,sha256=0WquhYAj8osAFf1It2G_L8uAceDqwC0XSLnh-RtKcDo,5667
tokamax/_src/ops/batched_rpa/kernel/kernel.py,sha256=HJIFeBTc0qCuhp5GF5N85w-muSzo2mwdB4v76KedCzk,34961
tokamax/_src/ops/batched_rpa/kernel/schedule.py,sha256=FNgRdbEl0fsMFs06AnmB5COUxZWFRN6TNGU7vN3h4qg,32662
tokamax/_src/ops/batched_rpa/kernel/schedule_cp.py,sha256=KBErQcnH8mX1UN8JwEhXNprB_M2iScgZDRtYW2wt1yg,10398
tokamax/_src/ops/batched_rpa/kernel/seq_along_lane_parity_test.py,sha256=6k-fGi-QI4PX0AI6AB2-eQSv607xY5DfKQZv1u43IMQ,7276
tokamax/_src/ops/batched_rpa/kernel/speculative_decode_test.py,sha256=JSQu4LZ3cDkA9njrp08zFH9Md-osrTpYB7gC0u5sZX8,6858
tokamax/_src/ops/batched_rpa/kernel/stacked_rpa_kv_writeback_test.py,sha256=0KfdtNqqXpqtTNAvbxCrFbcGIwitcpgps2dZVGFrfNE,8344
tokamax/_src/ops/batched_rpa/kernel/stitch_utils.py,sha256=QT8XzxeDUTIBTa5ZNje91LJg32ByOJgvRXnjfn0tq0c,7842
tokamax/_src/ops/batched_rpa/kernel/utils.py,sha256=M3fSX3Z1ze2PPvOtpk6Ig3exVovg-eMdislEUWKf4j4,5287
tokamax/_src/ops/batched_rpa/kernel/wrapper.py,sha256=TFWTo4YkzjLmrcSk3FRC9XykT_QiZDIPqImHUgOR0BY,32982
tokamax/_src/ops/causal_conv1d_gated_delta_rule/__init__.py,sha256=G9K9uL6YlXCFtWwE-NtwtxUwXV1RJzrteKzG10a-9dk,702
tokamax/_src/ops/causal_conv1d_gated_delta_rule/api.py,sha256=45aYn8TREJIB3d2-Gs2uQsdAlJKM3BAPJjZUl6P-cUo,1381
tokamax/_src/ops/causal_conv1d_gated_delta_rule/arg_specs.py,sha256=yf5Txm178I9HyxkBIcNyhiyzbMgX-XInpj5Y_gG4qCQ,19216
tokamax/_src/ops/causal_conv1d_gated_delta_rule/base.py,sha256=f-mWShWXZ4kjv_wZNozpavNsZQC_aDQH6PTo906d1xE,2701
tokamax/_src/ops/causal_conv1d_gated_delta_rule/base_test.py,sha256=rKYEE4XfrfdpdlN_LFh1124KqX9ZZ3B_utCqJN7L10Q,3557
tokamax/_src/ops/causal_conv1d_gated_delta_rule/compute_conv1d.py,sha256=-E2Kxe5fSR9OeesUBEDl-wZ3l0qTtavAGupY1fSEi9k,2662
tokamax/_src/ops/causal_conv1d_gated_delta_rule/compute_gdn.py,sha256=fESSr8PtuUFiFmGIzaTGMkafPR7XfuDE20owwKssDu8,19532
tokamax/_src/ops/causal_conv1d_gated_delta_rule/compute_kda.py,sha256=-ZkdPB3J2NgbtIPSFApa-_Epm6dtKC19ELuymWb1zCE,16987
tokamax/_src/ops/causal_conv1d_gated_delta_rule/config.py,sha256=f5h-eJO5eNjiVMdVmauRLkDUd8zh7DlWLY7RvopUQIg,8033
tokamax/_src/ops/causal_conv1d_gated_delta_rule/gdn_attention_test.py,sha256=KEv9dW6iqVYe1KYUQCjQcadd4u58jAqgn15ekfF5R7g,68775
tokamax/_src/ops/causal_conv1d_gated_delta_rule/kda_attention_test.py,sha256=COtMatr8_Ejlwf6ez7xJYIoWvXEjDdmexVbsw6WU7SM,24299
tokamax/_src/ops/causal_conv1d_gated_delta_rule/memory_ref.py,sha256=Of-hHvk-BOSuhPS-TW5yeF2SQeCrNZ8ImsNB0unXCU4,20678
tokamax/_src/ops/causal_conv1d_gated_delta_rule/metadata.py,sha256=Uwq2gE3B2P0Zp_DqwoOrPi4T4kA5hN1y0GXSJrv7W9Q,6690
tokamax/_src/ops/causal_conv1d_gated_delta_rule/pallas_mosaic_tpu.py,sha256=E0pju5Us4DLYHMWL0DwxJg_Eb2SSd43FS0wIQly1SEQ,6407
tokamax/_src/ops/causal_conv1d_gated_delta_rule/pallas_mosaic_tpu_test.py,sha256=KeS5XsRpvsANGZN9eBBa4CERaYtL-mgdXJEL80LRyUA,4761
tokamax/_src/ops/causal_conv1d_gated_delta_rule/reference.py,sha256=YGoKupq3jdflNCyIYA18uxDczfNIpsUkGAW9pV2asZ8,15116
tokamax/_src/ops/causal_conv1d_gated_delta_rule/test_base.py,sha256=h0p-0xQVzGcYmWBZtxir83o9I5jDOhnVFWMd403zyWY,19558
tokamax/_src/ops/causal_conv1d_gated_delta_rule/tiling.py,sha256=a4-g-1lKAumEnZerlCqXZqgwIL5jQwkf2vHyPaMx3II,13020
tokamax/_src/ops/causal_conv1d_gated_delta_rule/vmem_ldst.py,sha256=CRY80mulXxicPEFcNZAnUiYQd5QGKPHOsnUGm3aLO_w,12206
tokamax/_src/ops/causal_conv1d_gated_delta_rule/wrapper.py,sha256=YO_Zbe6ePqD-0zZIRre1As_JxdVUIHTB48U4pKkqnh8,24876
tokamax/_src/ops/csa_gather/__init__.py,sha256=Fyu1RjA1-ZGi6EbDO92VOs7ahZ6HNBDlSh6cgcJEteY,690
tokamax/_src/ops/csa_gather/api.py,sha256=iqXjJi22BnPvqLJj6Wdo5cTlQ6rby93Lalqz2eg_Yug,3831
tokamax/_src/ops/csa_gather/api_test.py,sha256=jwvesAu85shG0cv8qyRKMfS8Wteo1LOwGrJiq_c6Wuo,3360
tokamax/_src/ops/csa_gather/arg_specs.py,sha256=acF3fKMMnzbi9tYh_JHxWcqwnbQohPVCAZ_pvGBQRVU,2281
tokamax/_src/ops/csa_gather/base.py,sha256=HAyK9ZVl6GFLheDEfBmtXP9-dVtvHHeVuBBoI59YeC0,3157
tokamax/_src/ops/csa_gather/base_test.py,sha256=6Si6mLiCblpINkA1EzC6zcK81uyOfJFpfsT46sCbJ0s,3899
tokamax/_src/ops/csa_gather/csa_gather_test.py,sha256=XnCAT57PHLML6PNrd_MbpvnRiXGy7n8IdgCYXDQ-KtU,5217
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu.py,sha256=9kvaoE_hAogXAQSVYdpeKygOt1ndu7pboZuwaLQPSko,5636
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu_kernel.py,sha256=yifmBwX94bK4lUWnRUrT5EBs3fGwj2t2Yeft9SPsE88,12969
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu_kernel_test.py,sha256=qsaImrnUIbss9sEOAiIKRgV8Dv3dpH_ohboBjTxlKSs,7206
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu_test.py,sha256=qlmcq7OjfBxNCIpvpJ4vhHKAQKSVHsgu9pPW8qlxPx4,2575
tokamax/_src/ops/csa_gather/reference.py,sha256=quGbL2JVqImrWu9SO3A7zXUukj710HWtCvglceChTOY,3565
tokamax/_src/ops/csa_gather/test_base.py,sha256=2grsGGqZu5imrIRXuqJci5DUetUlvAGkHZUcyc4ur2Y,3588
tokamax/_src/ops/deepseek_rope/__init__.py,sha256=OW12E6mnjE0S4irbGbdaMcRo0_Ip9SCR_b5-JkHyE_E,696
tokamax/_src/ops/deepseek_rope/api.py,sha256=YAunda0BHiGBB5fr8DnnEj_y-rE4y-nir9GFa9LmKl4,4991
tokamax/_src/ops/deepseek_rope/api_test.py,sha256=CYvYR1xfneZT-fK5J-vQ9213GgLRYYsCzPwGXg1z9mw,4878
tokamax/_src/ops/deepseek_rope/arg_specs.py,sha256=sxFtOv2MvUUq3L3Eo-8MVfNFHlvmeLp2pi4wcG_pHzQ,2754
tokamax/_src/ops/deepseek_rope/base.py,sha256=-ILWvR8w-wAprsnoxXo4piiWNns4FwXzRWmuZ5ciekA,7487
tokamax/_src/ops/deepseek_rope/base_test.py,sha256=OnWkDjMRaDb5QB71yFOzYUHMyuIwdWMZd0xs-9Hu4Ns,3962
tokamax/_src/ops/deepseek_rope/deepseek_rope_test.py,sha256=OT6UVbIejJiQSJOMVOqjNaFDdWBkEwtORLIIKtnvx78,9269
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu.py,sha256=lKBe1uA3Q8m1XmOWEZbZUq47v653sCQPHwzQT4X8Ckc,8095
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu_kernel.py,sha256=N0EjCYBC_LevQ-4KX34qXUiV0KgrKrY_tjRUQ7oqkAY,20499
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu_kernel_test.py,sha256=IkPvPVdOFWEMKzC3JNHpI3Dei6MXVQE3hVAdA3IPdfs,10303
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu_test.py,sha256=dW2hNu7fVmW-hiPCtjRn18j0RJE4-qYluNm-BDpRd0g,5958
tokamax/_src/ops/deepseek_rope/reference.py,sha256=dgnvIr-8eTaoDiHC89fk2wicCzgNr6IpZGFi9J2_Y0E,5408
tokamax/_src/ops/deepseek_rope/test_base.py,sha256=wEakihTX72KwjGciMa-nneg9bUXwpzkgCud0MqteOss,6369
tokamax/_src/ops/dsv4_compressor/__init__.py,sha256=y_CpGGJ-P_GduSsHNwM6P3svzalHDduN9n3XqKhpCKY,698
tokamax/_src/ops/dsv4_compressor/api.py,sha256=7xMmKrqJnu6r9OUwwHwLR3MqWFyT5g_JcH3TDA1w6XI,5676
tokamax/_src/ops/dsv4_compressor/api_test.py,sha256=0wd9gg7Afa-gMtyasaFzbMmCgUk163zSMOiKQEXsi0s,2557
tokamax/_src/ops/dsv4_compressor/arg_specs.py,sha256=mRDsHqjeuCja2Dm0EGylnDpt_5hkJmq0ZexdsdeCUus,9087
tokamax/_src/ops/dsv4_compressor/base.py,sha256=m27nK6xHqqyUKpWjIxcyo6by0oyt-BZdfHOGHgO2N8E,10250
tokamax/_src/ops/dsv4_compressor/base_test.py,sha256=eAF1o67_NZ_dJeVjc2yycCipbewX5WyusWwwLpKDuiQ,2735
tokamax/_src/ops/dsv4_compressor/buffered_ref.py,sha256=_p-zqQD7g3cAGt-GDCJbjzbgJCzaW1g15YNapr55WyU,17066
tokamax/_src/ops/dsv4_compressor/compute.py,sha256=oRXkYNw-oEojINFIwtm8FtWP90ieYqQwLf2ovRf5MnE,15173
tokamax/_src/ops/dsv4_compressor/config.py,sha256=N5Ai3ZmaE-Fv3pdCQLX7hI7Wvtfju73JdO7zHFc9vpE,17411
tokamax/_src/ops/dsv4_compressor/csa_cache_layout.py,sha256=txVjf-kX7OUttuDDBJOsa5xX8iZFFpRvdvGjcNLlEW0,4832
tokamax/_src/ops/dsv4_compressor/dsv4_compressor_test.py,sha256=U7WhSQq5Z9HKF87h0e5FjgptSeMpnqsqNs2ruyadMRw,20207
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu.py,sha256=aakvrjAcPbUaY_Iw5LiXzP0XESboQuTolCeDx0_RwH0,6290
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu_kernel.py,sha256=foQAdHAvFJBhduO0VaVKqU2KBkBZuy7WfNS9iFy07U4,17585
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu_kernel_test.py,sha256=dWLA1RbRZgOYnz19VCQiASwl6hv9jeCQTNYHuhZnrwg,21337
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu_test.py,sha256=Fna0x2XxfQd7YGHMipAuin1qkKjMIIu_h9jz80W-jQY,3974
tokamax/_src/ops/dsv4_compressor/reference.py,sha256=cFP13n55wjZsk_Q1XixPxRMDF93URMvqEp8XPvT9T1w,22146
tokamax/_src/ops/dsv4_compressor/test_base.py,sha256=tr1Pd1o7FoR3pSTCwLA1FDrz8O9rZvWfhixVOaXZ-Ps,18411
tokamax/_src/ops/experimental/flex_attention/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/experimental/flex_attention/api.py,sha256=mDADkg4fpPVEDp0mbeWVY5cwM_kFHgEkRDZNelAPcmo,905
tokamax/_src/ops/experimental/flex_attention/arg_specs.py,sha256=DECEEmyM-WzMoD7z8nd48JBDKUK9zyhY2InIT76AVaU,820
tokamax/_src/ops/experimental/flex_attention/base.py,sha256=noxY4QSQ1o3hg4dTB-YH6FF9j2HgTD6Cmu6dHO9EUQI,12340
tokamax/_src/ops/experimental/flex_attention/base_test.py,sha256=gUx-MOXnycUOHnHSitSZWqtP8ZBaclTIf_cWpNpcPNk,1514
tokamax/_src/ops/experimental/flex_attention/test_base.py,sha256=RmAVNi29wphSWCX84Miv6sdnAtlJeNYTwpUsBn_catY,2865
tokamax/_src/ops/experimental/flex_attention/wrapper.py,sha256=_op2yZKYZVTwmA79mLvYmreAY1ennHKf1prAPkFUU8Y,3551
tokamax/_src/ops/experimental/flex_attention/wrapper_test_base.py,sha256=Xs1hf1AtQJ-529oxyC7wlzm6lvcbaO-2YGP850os6vI,1361
tokamax/_src/ops/experimental/fused_fp8_matmul/__init__.py,sha256=TGvQ6johv4CESH1L9Syn6UwXDTtp3DTXDLAfvvb6pU0,696
tokamax/_src/ops/experimental/fused_fp8_matmul/base.py,sha256=HUzmmDZyCgUkmqjcCWibawIv3QxaiVCpKQvFEUZ29Hk,11004
tokamax/_src/ops/experimental/fused_fp8_matmul/base_test.py,sha256=_YInpZgkLNcTXiPR5jMUmKfjlQ05sWR9siG2YjugG0c,5932
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu.py,sha256=3p8-96EwAmfpTswbQhaZC_oUWGZ9iogdePcjQH7AcPk,4784
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_dlhs_kernel.py,sha256=uSx3Oe9TgnuxhqRsJ92YmVACoFTzGlnkVUlG1e11fkc,8834
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_dlhs_kernel_test.py,sha256=Nuf5mJ-mewRfXBNGzB8MZns6EekGyisFIWWisN41wfY,5814
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_kernel.py,sha256=At0ITwAP5GmqOjLvntQzMW72zYJFDJX-vR8TOR9o_bs,18308
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_test.py,sha256=o82G749DkV8mhYfx93lfJsZfvasTEnq-MJJsULm4f_0,5148
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_vjp.py,sha256=porrpRpcmor3pMrOdzDAWnokEiBeYN-EgdWi9wMzoVk,8514
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_vjp_test.py,sha256=sadOD1QBdiptJ_2CahXHe3ELqq4iNLUx66nXWV1XF7k,7678
tokamax/_src/ops/experimental/fused_fp8_matmul/test_base.py,sha256=X_KiGAMXaQXpmZKs6zBaVVeZKKSbRGE5r3M6EbqQUi8,10269
tokamax/_src/ops/experimental/fused_moe/__init__.py,sha256=81z-IbB9dS9jFjt3kyQ4AFuMcGWoY5QQAumD5Vx86UA,2711
tokamax/_src/ops/experimental/fused_moe/base.py,sha256=O58sQNcqvAUuw0y-CdLMv1CLz0U5bfksxdXE-3AmyRY,2595
tokamax/_src/ops/experimental/fused_moe/base_test.py,sha256=l0DyNsIM7hXmsJOVbF3BuPtwpu7FC5yzy5apBAKhVwY,2247
tokamax/_src/ops/experimental/fused_moe/host.py,sha256=W3Q_v_qQOChPEpD4rMgKWS1gokqCsaLw7Spe7_k7hb4,68703
tokamax/_src/ops/experimental/fused_moe/kernel.py,sha256=UtPRhYO5jgzeSfSihExaksjbWhswXAYttOXi8XtOnwI,107008
tokamax/_src/ops/experimental/fused_moe/layer.py,sha256=ZD8Xj-qM-qWaFl4q_2Kd01V4AfTON3RxRkrczflmZ4k,32669
tokamax/_src/ops/experimental/fused_moe/pallas_mosaic_tpu.py,sha256=e225fv3C8cm1TlqL58h0Xr3DpUbFFACDH44XCeqoih4,3417
tokamax/_src/ops/experimental/fused_moe/pallas_mosaic_tpu_test.py,sha256=dWoViwa0UCqCZUSS0W5279DzJJvy153u-nXPb5RjZA0,2459
tokamax/_src/ops/experimental/fused_moe/reference.py,sha256=IQyI6qdrcCwOaO4yzzABmOFJK2p2NsNL8ZrkAGd-Q8A,3133
tokamax/_src/ops/experimental/fused_moe/router_ops.py,sha256=Rhq3-qEWzqPpuE3EYGn2tnoRbt3umM0J6r-z7XhkzFw,5041
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2.py,sha256=mwtqJUo0Mqp9FgGpVFb-B4ab2TI7D3Qp_lTn_IyN5fc,27221
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2_router.py,sha256=5cx_yBo5ATSZkSMgmEsyE2H2ue-rm6j0Tqkj4O0VsDU,5976
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2_tables.py,sha256=Og9qhu-sTHjUqTk8hghSMu0kvvVtRhWeO6I7HxQdqyA,10280
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2_tp_tokens.py,sha256=9a5Qoh-7xGvQ4DBCgwBfMW4fyI5A2vTEFoG9C7y9syc,12486
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2_weight_slots.py,sha256=t6ROU8kTmwpoAnaJ9oDmwARQAJlbAfGAIDeKDhkaT_U,6776
tokamax/_src/ops/experimental/fused_moe/test_tp_token_layout.py,sha256=MOuvErbbrvS9gOAAPKyYse-yI1Thxd4TAawVophy4wE,4409
tokamax/_src/ops/experimental/fused_moe/token_parallel.py,sha256=MJiKTTrsol5prpLTblPBJCms6zcyiLhqj9TTmb-Juc0,3761
tokamax/_src/ops/experimental/fused_moe_rs/__init__.py,sha256=5EvRgjJ9xY3epT4euvPnBYTUbRRZ6Xen35ck_dJIkWs,1093
tokamax/_src/ops/experimental/fused_moe_rs/fused_moe_rs.py,sha256=4LEA82nRgP2WpoGj2i-HuZZOkRd4jhZHxW_pZg9O55Q,19261
tokamax/_src/ops/experimental/fused_moe_rs/gmm_fused_rs_nodedup.py,sha256=5VQpSU6-29U9oacnZoDTh13XF7a6DmqUXygKYTTDbAU,77740
tokamax/_src/ops/experimental/fused_moe_rs/gmm_v2_gather_scatter.py,sha256=alEi1yCp-YGo68u4B8zSB7U3qqXC3nbJQE1fQbgJoHQ,37396
tokamax/_src/ops/experimental/mla/__init__.py,sha256=KpPrjAwIrgKMF4V7TtZtk46k9Qw3-SdgHZSvRCJLJho,693
tokamax/_src/ops/experimental/mla/api.py,sha256=ejIGMyD7opxhZNpdR5NfVfy09rGUJHvvBJ_qCzJs4vQ,1172
tokamax/_src/ops/experimental/mla/arg_specs.py,sha256=PWeVt58SdXILnyAbTJh4CGymbJ0A9uePhBkNelLZvdM,4453
tokamax/_src/ops/experimental/mla/base.py,sha256=Wvf5mvDB1N-qfoM4jBNJ7X-TJ6RNthC_LGyvnppw1q8,5072
tokamax/_src/ops/experimental/mla/base_test.py,sha256=hpq9XCmBi3xOXVWc5x0PdJbpFaHWVyRjwuaaoEziBkM,6536
tokamax/_src/ops/experimental/mla/pallas_mosaic_tpu.py,sha256=b6NdSKq3A1fso1J0JdXi4MH_ZV5UmvlR-o9cGe8hSVo,5060
tokamax/_src/ops/experimental/mla/pallas_mosaic_tpu_kernel.py,sha256=2FLT9iaXzp4p2QCi1ia2JqejsokVx0n8MFTK7TQIniE,91698
tokamax/_src/ops/experimental/mla/pallas_mosaic_tpu_kernel_test.py,sha256=Qjmh8GvvlJNPrepLxlPc7sr107xlsAAuoOaZ-M2EIgk,6084
tokamax/_src/ops/experimental/mla/pallas_mosaic_tpu_test.py,sha256=uVyPVANdzEQPjqPkTOdKenqEw3yFWYKylYA0UoE8GIE,3890
tokamax/_src/ops/experimental/mla/reference.py,sha256=fOvbEYFwHcs-FcFpI6_tSNP2Q9dqdEYRKdikqpiUx4I,9875
tokamax/_src/ops/experimental/mla/utils.py,sha256=kOHFcqEbVaRDH1CeymtJycrd5Mv-Bm4COmBS04utbuA,9677
tokamax/_src/ops/experimental/mla/v2/__init__.py,sha256=5DFUKg239Jo3JKSg9KnLeFojean388v3fu8qQuGBwS4,1441
tokamax/_src/ops/experimental/mla/v2/kernel.py,sha256=R7fUG4eB0kfVMB5i7f1IJLzFb0Yh-BITPhbNR7IbT_Y,104659
tokamax/_src/ops/experimental/mla/v2/kv_utils.py,sha256=mQ6nFdyebCLflIiYRC_KBIm06TGPYC__sCZqSFjv9i4,13784
tokamax/_src/ops/experimental/mla/v2/mla_kernel_v2_test.py,sha256=LWxsfmxuqosLRN17yKzYDM6tlEX3gRG3T6yJcbVv3dQ,49140
tokamax/_src/ops/experimental/mla/v2/mla_transpose_test.py,sha256=SkxhRNYXZy6wrk_xIc4IWh5y4_pPmz3W0whczr-1lrg,13378
tokamax/_src/ops/experimental/mla/v2/test_mla_tuned_params.py,sha256=VF4UK1pctWq5jd8ldwa_pFCSsS3WQueR8kmahAqSFAA,4299
tokamax/_src/ops/experimental/mla/v2/transpose.py,sha256=CT7RZJhKS67iNPwisaUfNB4f_EFWdvpS0eEtFCg3ldM,10406
tokamax/_src/ops/experimental/mla/v2/tuned_params.py,sha256=f82DOQ_aFJ3M1Y_1TJsSsHoluHrU7mlEyUEIW0GGXyE,31968
tokamax/_src/ops/experimental/routing_moe/common/arg.py,sha256=GzDj-bX6dbkrhUbfeeejoQ8_SISUZ_1uXvpKC82j4Dg,7932
tokamax/_src/ops/experimental/routing_moe/common/layout_util.py,sha256=HrnDtMOVKVydsacdDC-t1KjYZhxNoVUQH4RCk-PLEUQ,6006
tokamax/_src/ops/experimental/routing_moe/gmm/compute.py,sha256=SpOlyYDsJcQ5YHY7oUOGQIHNY1Mf8x6QYKc60_gIxIA,5934
tokamax/_src/ops/experimental/routing_moe/gmm/config.py,sha256=YtnbwlzdLTIgR_MSI6IWalQlwfi8rDD1JOCu_DEQDm0,8785
tokamax/_src/ops/experimental/routing_moe/gmm/kernel.py,sha256=cJ78uUvjWT8kStc9lsSbSYxwBk73C6ECRu10RjDqQB8,5279
tokamax/_src/ops/experimental/routing_moe/gmm/memory_ref.py,sha256=TaKi8d8VL_7DDP5YMNKCgwDRTGWoil6ktM02rSnAoKY,26643
tokamax/_src/ops/experimental/routing_moe/gmm/metadata.py,sha256=L2YcMDXzPzMDjaIgZeoDbiflHUb1aQiR9YWwvvCxfv0,5176
tokamax/_src/ops/experimental/routing_moe/gmm/pipeline.py,sha256=0dINqPTfKCMICEBxmUmoOCSNce9BW285U7gfUd9yCJA,7062
tokamax/_src/ops/experimental/routing_moe/gmm/plan.py,sha256=6GVpW-4f_jeUafuFgclK8xAkD9pewCQiuPgbQKUGex4,5473
tokamax/_src/ops/experimental/routing_moe/io_relayout/input_pack.py,sha256=7EfZtX1lscK_DAp3JODgMMh8soGXHZ5ukKMyBTcfUUQ,1265
tokamax/_src/ops/experimental/routing_moe/io_relayout/output_reduce.py,sha256=wbdRZTq0y4shN3g8NBoNZV_Fmite9Tx-MjaPQK0x_vs,1647
tokamax/_src/ops/experimental/routing_moe/layer/wrapper.py,sha256=Pk5m5YaFFlPLOBRUsf6BcZR_dczMGl42s-CUy5GNIVM,8155
tokamax/_src/ops/experimental/routing_moe/test/gmm_test.py,sha256=MWJXJsnSWe5twGrM5KzHuKynUvAfaLhn_jeIjiq6ToE,8138
tokamax/_src/ops/experimental/routing_moe/test/reference.py,sha256=t59jmyINLLGA62DYb5uqCIW7iCZI9kpTnu8tHGx1swo,13395
tokamax/_src/ops/experimental/routing_moe/test/wrapper_test.py,sha256=eqQoTAt1QiH_n1xMCemtDHTIxKiTIPQ_f2v3k7gd_UI,18697
tokamax/_src/ops/experimental/tpu/splash_attention/__init__.py,sha256=fmfcX8cskB_eEmnhvsaai-tSLNc3gaGwxL_A9UNTEn8,688
tokamax/_src/ops/experimental/tpu/splash_attention/api.py,sha256=eM-fwGGuoECvUemdLDD2QkPh-h3n1pFgpXXEvSOtQtA,4761
tokamax/_src/ops/experimental/tpu/splash_attention/api_test.py,sha256=yKEs2saR68RlIpQ1Zar9x9oWLsIM9Wni-x6R16Yfuxs,12736
tokamax/_src/ops/experimental/tpu/splash_attention/arg_specs.py,sha256=AqLs1ChqG1Jx_H06tE6KFq94jf1K-z37DcG50jTLmV0,4852
tokamax/_src/ops/experimental/tpu/splash_attention/base.py,sha256=okWeC16b09Cx0S1zPJNJS7cODiejdeT4lpZLKP0BV20,10878
tokamax/_src/ops/experimental/tpu/splash_attention/base_test.py,sha256=8IrTjfbvWhoCmK_YOjd-buWXvB1sr9kgExIvuiIEA1I,9281
tokamax/_src/ops/experimental/tpu/splash_attention/pallas_mosaic_tpu.py,sha256=HhsNcsX7_KnUO7rZDfLgtYm5Y-bILh3z7EXk0ja2l8E,11280
tokamax/_src/ops/experimental/tpu/splash_attention/pallas_mosaic_tpu_test.py,sha256=PkYO4S88r7sh_HYYbc90nqNtF_PFpbrmGikwVsyVvqI,12122
tokamax/_src/ops/experimental/tpu/splash_attention/pallas_mosaic_tpu_vjp.py,sha256=oQSwh0W9OrXXtVU1vKhQtAl0E2G6tg0RItZ5WxV4GbI,10494
tokamax/_src/ops/experimental/tpu/splash_attention/reference.py,sha256=xOhWh57-xMVDSKV4qHZZODaScsIBlNatdGfGGPjEITs,10663
tokamax/_src/ops/experimental/tpu/splash_attention/ring_attention_kernel.py,sha256=JtCYQC2TQXtL86g-nVLwiiJEK8YvuWF2RNUl98vtZrc,22554
tokamax/_src/ops/experimental/tpu/splash_attention/ring_attention_kernel_test.py,sha256=E-IgH6IlWkjTm6ec2aUhCCNcM8ddWI4GfgQgl1oxJSA,5715
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_kernel.py,sha256=kpHKMWwYlgInhMCdjzSOvat3f4H5536WACYCLu-H81o,104141
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_kernel_sharded_test.py,sha256=ZhviWMIpCPMzIYaM55MepNfgAdKQrg-fGEoysFunUzY,8157
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_kernel_test.py,sha256=XSeKwhH51M7tz6-UKSa80IUFNErbQ8SzDaH0W9gLk9U,53939
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_mask.py,sha256=8D5BGSx2uZZOhHSblz_BRIYASLN2hnOG0_ltRIpxegY,14950
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_mask_info.py,sha256=Ig-gRWOiZ9IVjSHiJ3OujbhiSq7RwBMSOwkF3bSm0nQ,20201
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_mask_test.py,sha256=OMbCeDrxjkPU706sQ5Jv3wQjaSKmrBenz-UzLFaXZ-I,54501
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_segments_test.py,sha256=CK_5WV3Cugoox-px4xA9hMaPeBbV2-dKBsjzPeVxIrQ,23179
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_test_utils.py,sha256=VunvVDHIKvcza1hJw0GkYCZOAqfTg3URqDA9YMn5RVE,2797
tokamax/_src/ops/gated_linear_unit/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/gated_linear_unit/api.py,sha256=A3_ui_lv7xlR7TYiJkoT5W-1nCrHAMo8NmZunl0buhA,4798
tokamax/_src/ops/gated_linear_unit/api_test.py,sha256=UT4GHSvVXR5tVjvGto9fLtgZbVNxokcvc5CtKf3ymlo,4237
tokamax/_src/ops/gated_linear_unit/arg_specs.py,sha256=H2stvhF72y5tAfW0wSdLrM2KxuZsi7NcpSVOQXf3v-U,3503
tokamax/_src/ops/gated_linear_unit/base.py,sha256=l8DzaLsUTnen2Mv7s4TQSc9TEAU6WNMQWGXXF3WoMMk,6535
tokamax/_src/ops/gated_linear_unit/base_test.py,sha256=zNoOZbucSKmx2up3WFSuuiNG3Nw8wBzs3ekHfm21vUw,2701
tokamax/_src/ops/gated_linear_unit/cutedsl.py,sha256=JtETXuos2BzlTITAH4QIbtj3kDms7a0S4l-JNHulG3I,6936
tokamax/_src/ops/gated_linear_unit/cutedsl_test.py,sha256=EDP8xbOoMR4sDCCrxUAfZq0ohQlGxXCnXz-24rIgo-I,2930
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu.py,sha256=r_wBAC4I_eKtOvN1NcYFCzLXvmlTXVg74AReveZuJ1M,4159
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu_common.py,sha256=RBFj8_qOjpznfPZHv6iq0FxKUl1oJgpFM-tblalCxgo,2999
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu_kernel_sm100.py,sha256=p9ld8cuW8N7lfx8KhAnTfgu3sjI78MNNG8kJPHlXgoM,10261
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu_kernel_sm80.py,sha256=5Ink4KJrJEj2QiUlhzyjafoGsretXWm1ZGjejpZoXRU,7982
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu_kernel_sm90.py,sha256=MiqFQsStJDgBpB29akJNRonU3agogxzJ73Ggio2kD1E,11890
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu_test.py,sha256=WflvMk-2G2-cIAsbwXkS7Mhw9Ns_d7yX6HmOYi4P910,1897
tokamax/_src/ops/gated_linear_unit/test_base.py,sha256=B5F5w8JII7ClEVK4HWXFL9vX6qfXiFfw_gskq3zefU4,3869
tokamax/_src/ops/gated_linear_unit/triton.py,sha256=GgIjy-gQrin2H6o_3Kf8r-CmIu1L3i7680IwI26MMTk,9995
tokamax/_src/ops/gated_linear_unit/triton_test.py,sha256=hXoC77kbg7_RfxoM0Na1zy2894XvpyMeh5JADZrdtZc,2043
tokamax/_src/ops/gmm_v2/__init__.py,sha256=OtXRXu-czwR1bk_ECYH3uJ5bcOomEIkjsyLFNJ4QZQo,655
tokamax/_src/ops/gmm_v2/gmm_v2.py,sha256=qNG5cKmK-D5mxYoMzag5OGI_-XhhWgVhqoCpFmrEAdQ,63910
tokamax/_src/ops/gmm_v2/gmm_v2_perf_test.py,sha256=UsblSWtG2gMESdKZAGpAjdihxo_9bZw6oOmbNNwVWik,10795
tokamax/_src/ops/gmm_v2/gmm_v2_test.py,sha256=0M6sIs4oywWZ-NFsDnwB60ku3OKSbfsKTDVQScZZX9I,64933
tokamax/_src/ops/gmm_v2/tgmm_v2.py,sha256=F_cY4ArZaOuewhtqfjerxdqmUUIQUPTu5HtrMeSgNeU,31269
tokamax/_src/ops/gmm_v2/util.py,sha256=A1tpYOIhCjHvupDTCDlnffAmjrZ5UNQ8rNZ1MNaX2J8,1899
tokamax/_src/ops/grouped_lowrank_proj/__init__.py,sha256=P0isQ1Ev-qzsu1ahtJ6vzc_8biYHZolgNCffD8IGPHA,735
tokamax/_src/ops/grouped_lowrank_proj/api.py,sha256=uwO-pNVwE6NzRZUKXYFZTIzdbu5I-jxEEUe9fK93dBM,4460
tokamax/_src/ops/grouped_lowrank_proj/api_test.py,sha256=g5Pq_hKnHPGbZ6ly2AvFBox4AKaQIYAplJomrWIv17Y,2989
tokamax/_src/ops/grouped_lowrank_proj/arg_specs.py,sha256=DbcDKq3rf8akULKVspwU1FbXglt-oF66GKAp1bPtMQQ,2888
tokamax/_src/ops/grouped_lowrank_proj/base.py,sha256=T9YmrIgO-fi5LFkcBUWCCyssAJ9Iyt6s_wNr35ggyIE,6815
tokamax/_src/ops/grouped_lowrank_proj/base_test.py,sha256=13q6NFXU0DCYHaHK1_81jWMMc5b8U-h0zXl9cJ3tADo,4103
tokamax/_src/ops/grouped_lowrank_proj/grouped_lowrank_proj_test.py,sha256=kWXF644H8u64LEvvoThLbf8Sh-fsWUKYT7N-6Xga6EE,6668
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu.py,sha256=Sd6-36NLlHrok-txOJ7L0Kj0o5sYnHiIy6wsYVpuAyM,8944
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu_kernel.py,sha256=t_A_ZGFG2bPPvQJaHKlq6YxUKUIeH1sfsJlc60az8JI,11411
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu_kernel_test.py,sha256=_EUAxjYFcLd_GRp5JFBChPUev_z8hA897QsC-bv9uk4,9353
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu_test.py,sha256=hrpc32IntcSzquYnfl-dO00JcrviO3oRzTuJJiZbpj4,4868
tokamax/_src/ops/grouped_lowrank_proj/reference.py,sha256=gSg5Fbu5kP565wUA6_r_5AWvuHyQtKPzzgN1Eg8xZd0,4105
tokamax/_src/ops/grouped_lowrank_proj/test_base.py,sha256=yCxX6eKDvqL8fv9gPzdOZ-CMOEceTG3mXlyR7hjmiq8,6443
tokamax/_src/ops/kda/__init__.py,sha256=5GYDGBGbUUq9RtvdIzQamyT8GsBUW7jy1uhYopJKn5w,865
tokamax/_src/ops/kda/api.py,sha256=LyyiEVpusyhIR0Jq5b03NSTOneTpQwBOUGSGLdoEiM0,7015
tokamax/_src/ops/kda/arg_specs.py,sha256=IkAioLlUEje9BEyM8oFYnJbenHZCFH3cCQSWkvAK19A,3427
tokamax/_src/ops/kda/base.py,sha256=3qBIgWQFOmYFc1cFJseegoRfyMZYLikWdw29hMWGL2I,7297
tokamax/_src/ops/kda/base_test.py,sha256=foz1G6MdOrUFqCldlpSESI92dKEynG0RYgufhG3xRqg,10457
tokamax/_src/ops/kda/common.py,sha256=m_zLU5gDWnsXFtw_DhIPzrGL4CemGQu9SFH7Rdf2X5Q,28157
tokamax/_src/ops/kda/cp_utils.py,sha256=-HfDEcdV4oEt7b1dzKhuKLeyVqicsVjyoop_tdDxTuY,17435
tokamax/_src/ops/kda/pallas_mosaic_tpu.py,sha256=g7Q2P6orzv2uRVhJGZj6mUIDMB6ZfZ-ExCXYTBKh7eo,18488
tokamax/_src/ops/kda/pallas_mosaic_tpu_bwd_kernel.py,sha256=tF7w0MuGGD7ziveeqliRtwrXaTEv3-0YzgrW1NwFFKE,60702
tokamax/_src/ops/kda/pallas_mosaic_tpu_fwd_kernel.py,sha256=fv7uULRevdEZh52JZhBOKIDsWpV3zVtdFX2ohPCPb1I,57380
tokamax/_src/ops/kda/pallas_mosaic_tpu_kernel.py,sha256=n-vjY-HkQTrWz5q_5xpO4cwsvWYuKvP6XVSQ_0ItAAQ,1000
tokamax/_src/ops/kda/pallas_mosaic_tpu_kernel_test.py,sha256=U27paxv6mo3rvoTmCoq8ZYwk-6Zym55gztP34yMtdpw,63782
tokamax/_src/ops/kda/pallas_mosaic_tpu_test.py,sha256=qxGsiSr6C92RPIF9Mt3w_PGtFWzxnEKXU3SoCIjCP_8,9517
tokamax/_src/ops/kda/pallas_mosaic_tpu_types.py,sha256=OjfVHD_YBUxHMK6kqKi9cxJpJ8xTYSvQQR1_Ww82eaA,2256
tokamax/_src/ops/kda/reference.py,sha256=x6CUpkp7gUf9LHj9h-fxxAY1xsRszNmQDqJBSCG6QQM,6812
tokamax/_src/ops/kda/utils.py,sha256=k8cmmbCr7_pVcsRQt3VqNGF4ICMZ59zMxVzvToX-WYw,16530
tokamax/_src/ops/kda/xla_chunked.py,sha256=c3-BipXodHky1RVFpUccwJACqM43GxUPurFDDvgOFvE,8452
tokamax/_src/ops/kda/xla_chunked_bwd_kernel.py,sha256=QkZ-xzSctVGVTRJFHINLr2gOKR07kzVsnKx8Xtk0WMI,2412
tokamax/_src/ops/kda/xla_chunked_fwd_kernel.py,sha256=zggfRQK4J34q3f07i6kZ7yf8ffpRT_KA0_e54RHNqCI,16338
tokamax/_src/ops/kda/xla_chunked_test.py,sha256=-Rx6Qw55VlurceCPedrOzzxhlXQzuWq1alrBQcjVxI4,8042
tokamax/_src/ops/lightning_indexer/__init__.py,sha256=Zx5fOsw9LvX-EK0a3SIACWiHqds0wZ4N-i0B8UxDxyk,689
tokamax/_src/ops/lightning_indexer/api.py,sha256=kFJC3GE0v5ceh9uw6lK0puQaJYvZq1DOkCDIwvaZzu8,4872
tokamax/_src/ops/lightning_indexer/api_test.py,sha256=Eu-TAXsTO2uNfQpreUdhrH8GKR08jGGcdXPIL0OEm7w,3346
tokamax/_src/ops/lightning_indexer/base.py,sha256=HvZsWwTKM7pQ1-68y0LxzlrjyleaObB8Rq7IM6lby3c,4990
tokamax/_src/ops/lightning_indexer/base_test.py,sha256=KsUD31sRrz7e0aZP8LsB8G5yEhn2G7vYKeGNGckdLMk,1894
tokamax/_src/ops/lightning_indexer/pallas_mosaic_tpu.py,sha256=adJVqMQLOho_XcCxKaHzVvCIvcee2OSSRgArsdZNiNY,4926
tokamax/_src/ops/lightning_indexer/pallas_mosaic_tpu_test.py,sha256=HbfFt9YQ3oCdr__l1qT59EBdrU-nh1cPqEJTgLeKUiY,2505
tokamax/_src/ops/lightning_indexer/reference.py,sha256=LRTAQWLrU4pvj-AFVEtLcIbwAJAeGVIbRbjbfQpDCGs,6891
tokamax/_src/ops/lightning_indexer/test_base.py,sha256=eIJDnP5Dbv7ZbD2gcTNhF-S1_NUt18idtA3scTOE9C0,7416
tokamax/_src/ops/lightning_indexer/kernel/__init__.py,sha256=ipixnHwBtMoXGa9jiYV5WmQfKpZrfA4tiI3GN9DCzdQ,705
tokamax/_src/ops/lightning_indexer/kernel/bref_override.py,sha256=_7HERrL_tx5L5jQC-963IW4UAZ2pjE5VHWcVvVdjUaQ,19248
tokamax/_src/ops/lightning_indexer/kernel/config.py,sha256=HG7FnjDfwbgxL_jtIxtUUeONFAIM01lh5kEAQ9_ENeY,5344
tokamax/_src/ops/lightning_indexer/kernel/dcp_sc_compact.py,sha256=Gk5EWt4C5OnGIIZOPK1xgtGrnjiHhuOaUMTxjvAY0C4,5575
tokamax/_src/ops/lightning_indexer/kernel/metadata.py,sha256=KfaVgVaWHPctbI74yLX2_tg-EXS2w1GSC25v2a8ksLo,13720
tokamax/_src/ops/lightning_indexer/kernel/metadata_test.py,sha256=S5oz2Gf5uvEqgux57i_vs7EhR7YCRfJz-WNwhGoIYHc,18160
tokamax/_src/ops/lightning_indexer/kernel/streamindex_topk.py,sha256=vxjftaWPAv1IDR5Op8LAjXCd7qhOxdCfEdqZZC1wrkM,50918
tokamax/_src/ops/lightning_indexer/kernel/streamindex_topk_dcp_test.py,sha256=NnJ8lB4hRkmnMFzFICcV_zdEAwJeC37NWTUQh-KSwJo,12683
tokamax/_src/ops/lightning_indexer/kernel/streamindex_topk_test.py,sha256=Rrzq-71hC0t1KAV_De_TAItnGaQwO552j1FT4XbUjy0,53090
tokamax/_src/ops/linear_softmax_cross_entropy_loss/__init__.py,sha256=KjQvwjX9P90Ur6p00QPLGux-pBqPQrSYlirJoIWvhoA,732
tokamax/_src/ops/linear_softmax_cross_entropy_loss/api.py,sha256=4nBK_cIPyntCpZlAmXfOKsTKEiZoe1J8FTIcbAYCvnw,4712
tokamax/_src/ops/linear_softmax_cross_entropy_loss/api_test.py,sha256=-9Nvvegk9FpiwqEAYd1CPkTVUE-kvbk7wynxEte5Hbg,4107
tokamax/_src/ops/linear_softmax_cross_entropy_loss/arg_specs.py,sha256=wmLi_JJHHMMLJ-m5InX8HyBgyBn5DEWuBuINMKTlPJY,7410
tokamax/_src/ops/linear_softmax_cross_entropy_loss/base.py,sha256=mPH7Roq1_N5d5bbJoc-Fzawinj6rIJ-u-3XepQSSFqo,2842
tokamax/_src/ops/linear_softmax_cross_entropy_loss/base_test.py,sha256=dk9kvrTaS4hFjSlxjWxXmw7rAySnVycKACXYQaUz_RU,2464
tokamax/_src/ops/linear_softmax_cross_entropy_loss/chunked_xla.py,sha256=KCa9rZt1oI_x0Zzkru_5u9yLyQAhUDTEkdRQmf4TQV4,12246
tokamax/_src/ops/linear_softmax_cross_entropy_loss/chunked_xla_test.py,sha256=2NV4vrTdNvtx9Nc5wnpDdsHncg8wyXPDTlmBWZg8lQg,5996
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu.py,sha256=Hwh7J1tuKqaSgWf3jDGMpsz8axNlTLglAW-DebbycHM,9705
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel.py,sha256=w3q6bPK1AM-simWe33GVtPHoeh9UDSeKDlFzZIxaAAc,57084
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel_test.py,sha256=795rRI5_Mx76_uhsndhrfqWYLf6KuVK-WyWUZCVwISI,59113
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_test.py,sha256=fc0zhKXHWtas9gtTRkiIUjdIooUFpUaYpYpPahOoZRk,3517
tokamax/_src/ops/linear_softmax_cross_entropy_loss/reference.py,sha256=OAR_er8dlD7WLHfd-FIamL_2e_oDPZwIfSNCu5JhWNQ,2981
tokamax/_src/ops/linear_softmax_cross_entropy_loss/reference_test.py,sha256=mVNHDTuQluJa88CY617OyAbZvZwMoMUdfEFW8KcqfMA,4932
tokamax/_src/ops/linear_softmax_cross_entropy_loss/test_utils.py,sha256=0gjGXVuLWy7dEePckg4zNTbjwmYrnQCNz2KQao3JUpA,1963
tokamax/_src/ops/masked_dense_mla/__init__.py,sha256=E778mfVnSKF_KQi9lXNvRbAVJ-LjoAqirGPyjaCw19Y,688
tokamax/_src/ops/masked_dense_mla/api.py,sha256=wpJEMJc0_hOzZnlrxOhIuZ_wk_x2VZMFy8SPGAbpkbM,4676
tokamax/_src/ops/masked_dense_mla/api_test.py,sha256=uzyJHgLSpGnUDeWxs-E6yWYD_6gRphpSNPB7IWTv7_Y,2153
tokamax/_src/ops/masked_dense_mla/base.py,sha256=kO3tYqYRFuW0Hl6b-OmJTzo0VuruJwKPp43F9gC9FhY,4917
tokamax/_src/ops/masked_dense_mla/base_test.py,sha256=D-8M95MWvtQxP-KtXjHDqoQOdgvbBt9RlvL3bSyuBYY,1986
tokamax/_src/ops/masked_dense_mla/csa_mask.py,sha256=4_pVzxzEJuz1EeX6APoCjgpcflRV-or2u8EaKrNsULg,5255
tokamax/_src/ops/masked_dense_mla/kv_cache_utils.py,sha256=-VdcGqodouMCBjw4ouF-H9T7SWFH6XSoSVAJmDedjjY,16907
tokamax/_src/ops/masked_dense_mla/kv_cache_utils_test.py,sha256=1_qFH3CFTbZVsX4GMsjVAosmI8UZ3ry07EAmXjzulzU,18833
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu.py,sha256=Y0jEfUFJqBYNBSQcjDy5n-KiQQ70wDtv0gUIPZ9p878,4765
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu_kernel.py,sha256=U_Qq5OuX7M69ez-BtIGBhjCLiaZLMdDiLJCdIM_a8h4,33293
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu_kernel_test.py,sha256=nDtaazaIk_KsXIckkh3RKrNC_aKysT7b-GbH8FiPeDs,15886
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu_test.py,sha256=K8J9y5kXtDFaBtGIS833eCdmrmoT0CHR3xK4MQaWWHk,2067
tokamax/_src/ops/masked_dense_mla/reference.py,sha256=SAmUIfzvnIFXNxltRsjOzobL1CGuWGA0273bBiRnBGQ,6032
tokamax/_src/ops/masked_dense_mla/scatter.py,sha256=t43oRFxij3Vom5_ODm2VZjW01856Ef7SMThX8TKGBlo,2806
tokamax/_src/ops/masked_dense_mla/scatter_test.py,sha256=EQB_fotVENFvvupVto6bNqJKsLdoWny9GoAe6M-ZXuM,14728
tokamax/_src/ops/masked_dense_mla/scatter_utils.py,sha256=YHx3v-xqpsQi5mp0oTyy5K75fIne8nB9Uz2arm4rHng,7411
tokamax/_src/ops/masked_dense_mla/test_base.py,sha256=JEQeNyi6_Lik6rFDf7jJRf4m-k1QyBVzsIH58JhNgok,6687
tokamax/_src/ops/mhc/__init__.py,sha256=xpTpj-P_eNfir6pCWMzDwzPUzd7Iw7l7RJ2586sGcR4,736
tokamax/_src/ops/mhc/api.py,sha256=6WkYYTLrI0n-TtjWBSM8JfQft3QGdDORx5QXX8qSOm4,9713
tokamax/_src/ops/mhc/api_test.py,sha256=QvDKuKPT60e2zRyhmvB7TaeP9z6I-dJrnpqwLh9FBBk,4534
tokamax/_src/ops/mhc/arg_specs.py,sha256=gnu7MdxcqVRVVDLpD__TNIt5VXQ4uagr8nwnJZqChko,4019
tokamax/_src/ops/mhc/base.py,sha256=eTiqzwn6latVnUXLAwvF_oxd4qEQGubnLGfHSZXsT_g,9355
tokamax/_src/ops/mhc/base_test.py,sha256=d6ARSLngBSmJ2l04JO1rPPEkeIA_msYwXPlLdyapwmc,7141
tokamax/_src/ops/mhc/fused_post_pre_kernel.py,sha256=J4_mU1xnga67KKgjjay-qqKaHe6Tq_bt4WRh2e1Q5tQ,11030
tokamax/_src/ops/mhc/mhc_test.py,sha256=QJHcsEQYLw_XvIzXd_6AV5vlrI9h3HjTQV4QcsMMgCQ,19629
tokamax/_src/ops/mhc/pallas_mosaic_tpu.py,sha256=dleyzJHdmuRsIJCrix3Fd1W5nu224XAxwxfYljs6pzc,8170
tokamax/_src/ops/mhc/pallas_mosaic_tpu_kernel_test.py,sha256=cCTHVqYXFTubJtb9PVZkYG3dq3mLGX1D-8CGrvnJH-s,5614
tokamax/_src/ops/mhc/pallas_mosaic_tpu_test.py,sha256=4RDunR0qTmF8XlMSBC9w5eRPHvbGpWCSNb6qR2M4qLs,4051
tokamax/_src/ops/mhc/post_kernel.py,sha256=LCRHGPzu3OtPEzkrAGUjxovS55flqVCL3a6_mC2vtCE,5881
tokamax/_src/ops/mhc/pre_kernel.py,sha256=iORjYQTMJ4xKTt3GoQDIFuh-P03ofysV9nv-8J4zNzM,11905
tokamax/_src/ops/mhc/reference.py,sha256=Zkd72suUXdnGo-sORUU4X66oHF60f7DrIi7Kgvh6nLs,8442
tokamax/_src/ops/mhc/test_base.py,sha256=Wg8V6R6tdj6i-KAkhVxeXwQjVHEBvOTtrwixSS-2qPQ,9077
tokamax/_src/ops/mhc/utils.py,sha256=4MDYnB3j23bHp2G6ouqAa9giJbQXvequx19Nh5BdgaY,5623
tokamax/_src/ops/normalization/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/normalization/api.py,sha256=z04jE0sXt1zq44GJrQPSPq-Wia57k2kWkiyp7OVO76U,4516
tokamax/_src/ops/normalization/api_test.py,sha256=4150si382wUWSdBJTOXrMoLxMXgvPCQAuvOi_rEg3rQ,5167
tokamax/_src/ops/normalization/arg_specs.py,sha256=Qai5dlv7OgAvmQb7RmMOl2dc8qtkpL7Uno0hWPAeP2c,2750
tokamax/_src/ops/normalization/base.py,sha256=1w50Ew4BDNLvb7onehV9Xv7Xq8HL2CEL2yPhni0Cpd8,6241
tokamax/_src/ops/normalization/base_test.py,sha256=XKZTDAMwsol5oEzzdB9Wu-pu2_12eSSadGLyMeKKweg,1027
tokamax/_src/ops/normalization/bench.py,sha256=DxvvboTF2qqmEVs4tSeCutrXfsKX0uZcex51ogzK7ZU,2188
tokamax/_src/ops/normalization/test_base.py,sha256=ODAR0SG-83FHYDIlGxR54ZuJkEQ-1T4hqOH7NDocdqQ,6684
tokamax/_src/ops/normalization/triton.py,sha256=j84k49vOqOfEvW8prKAdiqDbshQmZWV9_ZhQQ_nyD4s,17980
tokamax/_src/ops/normalization/triton_config.py,sha256=Bunc35aBzo3jT1DyeKVRApKfSFAvuDC2bFWylkRhdYk,4157
tokamax/_src/ops/normalization/triton_test.py,sha256=f3f2wLNGG7AVO8Z13nIz69WssVdTQ2GQYxHPXDKe4GY,4963
tokamax/_src/ops/normalization/triton_vjp_config.py,sha256=R-7FMLYOlvs8aZUw3uVmjtus9Z_v-1EMqKq-CVgxdSE,3043
tokamax/_src/ops/proj_and_save_state/__init__.py,sha256=EyGX1VT2fUTgHUF2DrOYOTHZh_jU5Z4W-32TKxlAJx8,735
tokamax/_src/ops/proj_and_save_state/api.py,sha256=CvAN64b8TyQpa686ZXPTP4R3xfmFGYzUjb-fvi4fKng,4155
tokamax/_src/ops/proj_and_save_state/api_test.py,sha256=s5JH3O87pRqJCToiQNKQ36xln9d6ortebgqbS-mOKj0,2745
tokamax/_src/ops/proj_and_save_state/arg_specs.py,sha256=rSTzFTtUEB7b9hDFlD4Utwi8p4oI2_-B5UjBI-2rE94,5291
tokamax/_src/ops/proj_and_save_state/base.py,sha256=Lo2j4APoPx-oJAMdi3vI9xsdVXwwdbYmQMA7WpGs9Ks,5384
tokamax/_src/ops/proj_and_save_state/base_test.py,sha256=2HBYue-rVBIezedv6wEYmfKdnly4iODdbZ22w6ohS1g,3896
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu.py,sha256=3-tSH3Fkk2uIOAG42FLY-maUdhDqAF_NJUd3E3u_xSM,7686
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu_kernel.py,sha256=sl9wVi14Jbz_o9Dww8llxCDUMn3pX1oqQtlgqbgVKps,15650
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu_kernel_test.py,sha256=VZqsnL_frJVoFUQ1_kjpOsJgxPobvz4MrlEkbTx9MH8,13006
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu_test.py,sha256=WoI53DbYkFJGoxLI99zcnaYDLci1007tfZVeazn7KiY,4361
tokamax/_src/ops/proj_and_save_state/reference.py,sha256=aEcpHlkxoHwR2USdIRp6PPoYWRCScmntSTF8vWYKPmc,6470
tokamax/_src/ops/proj_and_save_state/test_base.py,sha256=clQPpXCXjxkgDGrk3io-C92n8bSVdBzjw4aqAldGaAc,9612
tokamax/_src/ops/ragged_dot/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/ragged_dot/api.py,sha256=kXJHavZG8hNRfBVK5IHdDJnDo3DLzAqdTGhSiS7Fp-w,11610
tokamax/_src/ops/ragged_dot/api_test.py,sha256=BWv7mC9l-SyvKsykCRynZ0mUowqKWZ54BZTVhZHD9BE,19650
tokamax/_src/ops/ragged_dot/arg_specs.py,sha256=IRGzFI1fMVgWq5Z7ER3gqnfB3eUDiVRPPTPkFw98eg4,16532
tokamax/_src/ops/ragged_dot/base.py,sha256=QzPE6EWnkzKW48aHWUWb6nEmLHupaQOEmkNPGuMqgNc,16441
tokamax/_src/ops/ragged_dot/base_test.py,sha256=AKSHFalUo4y4UFhKqytLu6hfZOObdYpBOuLI2jOrqjE,2151
tokamax/_src/ops/ragged_dot/bench.py,sha256=FKGCBq6b8hIsn0vbvH79xo3-FnIHfuMAJ4C-ROFTYDY,3779
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu.py,sha256=mbNirrMat02Z8Sv7D-RrxPz3AfFgpo3792V_2ZCCcdE,20469
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_common.py,sha256=kddO_fVfBplTaQtnSN-r3Sx7urNFWu7MP176y5a27cg,7695
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_common_test.py,sha256=CultD9jt7z1sf1JsTTUEjCDqisGVrj8kdVrnUSxmRkI,3648
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100.py,sha256=7curVx6Dn1zqRe7ryTLOu5QI7dlYnac3wuZ5mYBA_78,8991
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_fp8_quant.py,sha256=_g987k7fTymUSny5BjF2LsEjGcgQIdOOxS1XiWdrBEU,35206
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_i8_quant.py,sha256=nQ3jCkAUJrGTVrVQJc39FYfTGQngiCcbU_0I-9PoPQk,21064
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant.py,sha256=5P36UWiyApeb27YoHkqeTv5z42MHirOpmoiWMUHeFow,17240
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant_post_scale.py,sha256=yW7HrH2ZWC7hJV8NUjvQRLvIcUP9iJUB4DCO2YokOZ0,16303
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant_test.py,sha256=dQlxWKkwi8rC_EXH6CfncxbeDxkdF29iH5bwtEOrVw4,22935
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm80.py,sha256=69vl1FcfvL3NmnroOBmh7b5x2AEJrGxxsA31jPE6GlM,6710
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90.py,sha256=U33cxx6r8FFUIxdi7znuo7N9oXRydiDOoSis7YeH_Nw,9241
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90_quant.py,sha256=ef1C7IJ_7qBrUagev0qk4X-ftyhXfE7ioyRoR56rb-s,13352
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_test.py,sha256=wvNd2SrvJ3idef6bT0kbU43vVO8HjcZZ3wc4wc5lRl8,4795
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu.py,sha256=0be0vlvfsPr2UUvhwTK02G6EoHk-3xpFPv4K9sBrVdw,17645
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_kernel.py,sha256=uHEuufzI-5Ljw74zfiGCPIKDkF_jzumenm0ih7X9n8M,34961
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_test.py,sha256=CdAnG5GKpwvxJDiHXT6JT0SN7S3SXw875z-NoQs-fsU,11872
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_v2.py,sha256=mfZwroX-E_8-bpYOwQKcV1IQS1CYaLt-2vae-JvfL7U,17397
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_v2_test.py,sha256=fBW7B5Sjl2EkwSKBt04lQUGagHLu8BbWqBhI1nPpRAI,22182
tokamax/_src/ops/ragged_dot/test_base.py,sha256=tkN_V64DKMZT6odcNAQikW5MXY5peFCNeVuGYiJoYVc,11788
tokamax/_src/ops/ragged_dot/triton.py,sha256=FNY31Qya-3MLUTUpzhbyKPwjEZZIXr7I4OgD423drmQ,21886
tokamax/_src/ops/ragged_dot/triton_test.py,sha256=P2QLj3QVWw3dBELYC1dwSA_13zmATCUjDy5Wmw_8TnA,3320
tokamax/_src/ops/ragged_gather/__init__.py,sha256=Xh8d6rEAf-Bp-eZ-luzfGMPT-92LDIRQeN7Fncy0XCE,693
tokamax/_src/ops/ragged_gather/api.py,sha256=qRy5aLkFtGAAKK-U77y7_HyRqlfnH4d3Xd9WicH0fuo,2908
tokamax/_src/ops/ragged_gather/api_test.py,sha256=-1lMdGhiOdKbEsO1JDJMx8Mkhxn6yQ37_Ke1mIGDHVM,2732
tokamax/_src/ops/ragged_gather/arg_specs.py,sha256=f5qEZmpvbF17tsVKBvqeSzOxwXKi3-3b4tA1YpegiTI,2779
tokamax/_src/ops/ragged_gather/base.py,sha256=dQheMnepsIL1Ma7cQKvqRkyD6o0j7LtE5oQidKSieoI,2145
tokamax/_src/ops/ragged_gather/base_test.py,sha256=5VJtELwCpsA6ybuB77nyzv8wbgLbQwuONLDWJ6kjIeI,1460
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu.py,sha256=u9V-nMjz8W-aho7O20IR8PFHTAbrttKJjzERretuWyA,2646
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu_kernel.py,sha256=hdU2drzRkzJ0tBt-h1u6T1KsDyQ29PRbYTGH-4yC9NQ,10379
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu_test.py,sha256=9lPCuXqJBI9_2Tb8PUlFmgg1T5mThIMOxNVbKqx_DFM,1670
tokamax/_src/ops/ragged_gather/test_base.py,sha256=hqqly9uQCvMPEnMpCCx-7dIb_OjZSJnmUbfDu9FisW0,3312
tokamax/_src/ops/ragged_gather_reduce/__init__.py,sha256=1yQAi7-n8YX_dQ8BbXTYO42JFSQSFzVPm5gtW712oe8,700
tokamax/_src/ops/ragged_gather_reduce/api.py,sha256=tgZZW5zkYFvZs_jJE8PlU7Nk32Lt1fzAuAa51j2P510,3670
tokamax/_src/ops/ragged_gather_reduce/api_test.py,sha256=S-HPtU2PHPLHzyyE_IFbcrNUL8xhTPQH-5use0EK-ZI,4585
tokamax/_src/ops/ragged_gather_reduce/arg_specs.py,sha256=UnYPWnBbFtgaHXRJ5kc8Ii71vhhGynlQoLlJItEDOks,2461
tokamax/_src/ops/ragged_gather_reduce/base.py,sha256=hVPleKgA4b5VFbkk7V_wd83SKYCcHch1SQ4ighHmo_8,2719
tokamax/_src/ops/ragged_gather_reduce/base_test.py,sha256=tCG7_0ZQEwRcsYjHOz_fYExWi2zMIm_d8sv28bCjp_c,3674
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu.py,sha256=SVGzNNCXjcHQHD07erJu772B1-eWkYN4oWtvpMSjYcA,6004
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu_kernel.py,sha256=G3e87hAZloWyZoGQWSzKKYJZoTMPIeeN71XC9L4lZ_A,26969
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu_test.py,sha256=nweYnH1yCzqtmjQxNviogUxB_jGty2_ieqzJ13hjRos,9251
tokamax/_src/ops/ragged_gather_reduce/reference.py,sha256=C9MA8og2PQsFO3LlCq37Oxp5b957RQ-WNRlF8GdT9g8,2265
tokamax/_src/ops/ragged_gather_reduce/test_base.py,sha256=e5-DQTb6ySvwCUcxgUevkRXnEPYhEpGbHsWpJsLkjEw,5708
tokamax/_src/ops/ragged_scatter/__init__.py,sha256=viUEEWXyimyRWWopTjeg2YK_wDkj16NyxJGm4MPdsOQ,694
tokamax/_src/ops/ragged_scatter/api.py,sha256=X7kWOtllTQalMYSMh2bYFHrounAB_Txbpee_nWOcTAk,3194
tokamax/_src/ops/ragged_scatter/api_test.py,sha256=0EpTwQhsAWnjzvt9_yI_uEore_cofz2kfSMRg74ztd8,3884
tokamax/_src/ops/ragged_scatter/arg_specs.py,sha256=68lCB2dIeIc0xzIi0HxiRPBDFx5mZcOYW3UMPcsJQgE,2197
tokamax/_src/ops/ragged_scatter/base.py,sha256=Gc8xEDka5qVoGY85bsD-qg_1CpuyDp2qTq9434iZjSk,2407
tokamax/_src/ops/ragged_scatter/base_test.py,sha256=hJQxh7GzvNj8mSUh_wFpoocey_C3Bab_hvOM_MQS5QA,1559
tokamax/_src/ops/ragged_scatter/pallas_mosaic_tpu.py,sha256=zwqrtfOrVSXZZSvfgrGwB-9DOOlNGMsmAuLhaGp_Los,2344
tokamax/_src/ops/ragged_scatter/pallas_mosaic_tpu_kernel.py,sha256=LQfbxJSl5JWQlekY23U_42mJtPyQ8dHRS1xrTM7-KAQ,13702
tokamax/_src/ops/ragged_scatter/pallas_mosaic_tpu_test.py,sha256=IGlB-J5wqx1x7tkRK5cJXH4Gucy4dQMM29BVrKqE-l8,2446
tokamax/_src/ops/router_topk/__init__.py,sha256=N4NCVxareHoVJ_e4t5ohe_0niltDrq2564ujDjMbASY,706
tokamax/_src/ops/router_topk/api.py,sha256=Lvp-LoAVymupH8w8JAdAimqWyGwg4olvB-qnffEDRgk,3720
tokamax/_src/ops/router_topk/api_test.py,sha256=bmET9--aBMvUtWXXcbv4BwpL8s_R8jBvHELNlGvfyMw,3357
tokamax/_src/ops/router_topk/arg_specs.py,sha256=PCAVrFU-L2C63zC3beO77mCt3QsJM8MqDGO3gnAdMWA,2196
tokamax/_src/ops/router_topk/base.py,sha256=VOs2MHlFtY5-9Nm8nPE2Nwd9FQGPfEFpJ9yz9G7ImcI,4235
tokamax/_src/ops/router_topk/base_test.py,sha256=LGqr70jhXAqm4kSi9gjNmAMtgvY74uIsoIt2aIROLJI,2521
tokamax/_src/ops/router_topk/pallas_mosaic_tpu.py,sha256=B13NR6jqnibdyvylHEOQhdvXOyKsLOeLxPtCAV55Qiw,6241
tokamax/_src/ops/router_topk/pallas_mosaic_tpu_kernel.py,sha256=v3MRJ86caxh4jZdxtbK_GWmJAXlBcIn2kaJZ0xDnKkA,4399
tokamax/_src/ops/router_topk/pallas_mosaic_tpu_test.py,sha256=btrx3-phJdCyWJf6mFARHPdye3AFBKfowmG-6QutkyI,7912
tokamax/_src/ops/router_topk/reference.py,sha256=4_SHnCV3v2E8oXVTHYyC2weljdHoYx9Z1rC8Lykt_aU,2811
tokamax/_src/ops/router_topk/test_base.py,sha256=X4UfmP36FZlHLb8TEgx-KdyYAbd66Rz3g_BTm5CkWwg,8330
tokamax/_src/ops/sparse_mla/__init__.py,sha256=c03KJKlKUVPo8tcxS66T-mpz5V9rhNdEz1b9Ydukp0Y,704
tokamax/_src/ops/sparse_mla/api.py,sha256=L-e9vZDuOLuCEVCNyMb4Sdu6rDTV5aRDU9SXBSbvFsI,4261
tokamax/_src/ops/sparse_mla/api_test.py,sha256=hFQXjJhw--p4hpuHjSueIm9eVuKkWSDMbB4mAfcDREQ,2061
tokamax/_src/ops/sparse_mla/base.py,sha256=nyyTwYLUyGv8uCQEYFkrLj4YHlCknCwT1I3bZiBzK6s,4382
tokamax/_src/ops/sparse_mla/base_test.py,sha256=47yhkXB1WahHHV-o7xMAEM24GdPHNRk8BFVrXml1xOo,1975
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu.py,sha256=hUa2uBusRQHAdZa44rvYocox-Lchia30E5_unQ7RnKc,4434
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu_kernel.py,sha256=iQ7XJIxac3HawMV1Hy4CvnFdMbjV-g65TAUApSXb8kc,31495
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu_kernel_test.py,sha256=IMKfQ7de8art7VTtwfGVYSG76YlcxlCLcK2RQQIcVYI,14570
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu_test.py,sha256=oCyW1BCTP8IEVYZ6_si2YD_DWbyQLdCg2GDIvHRuoqM,1901
tokamax/_src/ops/sparse_mla/reference.py,sha256=qjdWzAP0WZU1tZg7isovq3bUqN41F_-KAFXJmOH-R_c,5361
tokamax/_src/ops/sparse_mla/test_base.py,sha256=LMGbVtmXcI-3NqA6e1gSLtT95Lk6Q0kIRuqe7C69CZg,6197
tokamax/_src/ops/topk/__init__.py,sha256=T249jb1MwC6WFuwr1-Evu-bDB4OG_3NmhYqqkvhpbwA,676
tokamax/_src/ops/topk/api.py,sha256=bgwj1-KLaSwpzeo1_dvLuUia46IwdyK9-A0koA72ZNo,3377
tokamax/_src/ops/topk/api_test.py,sha256=YvV0oV40O1zZBupHT43fbdL5DRmNP9QpXueS1xzKjTo,2301
tokamax/_src/ops/topk/base.py,sha256=MXrKwOOrCvgrCmhLiFbU6Ja7CNLTGCJlT9F_eVJJRfY,2649
tokamax/_src/ops/topk/base_test.py,sha256=w89OfBe4eGrv6CGM4aubwDqVzKxYIBOsmg0HWr3Z4aQ,1873
tokamax/_src/ops/topk/pallas_mosaic_tpu.py,sha256=hyk_7TaNJDVcG2pnDF7xvVSCfSXB1pUEHRechHGia8M,2682
tokamax/_src/ops/topk/pallas_mosaic_tpu_kernel.py,sha256=OMqsbbx-P8_7Sa9JEFTsGB-cjTlKB_7eCwMFFwaMdcU,19385
tokamax/_src/ops/topk/pallas_mosaic_tpu_kernel_test.py,sha256=3kiW6fGUh5vSt6ZTg-rIWdIE9QIvEon6_RyMXgG5KU8,14283
tokamax/_src/ops/topk/pallas_mosaic_tpu_test.py,sha256=upuraQjDWdBG26Ll6YTmHJ2ZnXVxq1A_HVgRqfWKVzY,1395
tokamax/_src/ops/topk/reference.py,sha256=2FMQc62K5Lxk_6jqwA4GBEJwIN3Ni6z-i2H1a4dRJz4,2797
tokamax/_src/ops/topk/test_base.py,sha256=YHjFPZDTqV1Ysz7LouPNL1sORcSwQop7WRQpMvWGFVI,5920
tokamax/_src/ops/triangle_multiplication/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/triangle_multiplication/api.py,sha256=s0ud_YQW9ZGLoalFd4WNNQ4mGl8XhvvXwiH72j069Jw,5516
tokamax/_src/ops/triangle_multiplication/api_test.py,sha256=8FSAE3qe8jFb6KmzwlwcdAb3_eg8jVTT8WVzBfhqyyM,4449
tokamax/_src/ops/triangle_multiplication/arg_specs.py,sha256=BN9SSgfaCd-2Bz_CQMGbUQADuRx2iMn9heeWPWe-WM4,1931
tokamax/_src/ops/triangle_multiplication/base.py,sha256=dUxCqOrerpVQmNUyMmmj8CItyuFLgkFbkYpKagZysog,6919
tokamax/_src/ops/triangle_multiplication/base_test.py,sha256=LLjXY0wTXuUMsradIVGN1VLCMI3zhj145saHr4XeZbU,3452
tokamax/_src/pallas/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/pallas/block.py,sha256=QA-qu_Li_U4S1iyP2e42pe2OFDTvGa2SXIb0KgU_ois,11845
tokamax/_src/pallas/block_test.py,sha256=s1lNHBl_guAh2BwMjjWiqeKR0Nytl8jn9SF0dkFdN5c,5519
tokamax/_src/pallas/grid.py,sha256=RMBVH-rZ8A0V4CDeupFdqU1Berp9_rDlq4LEQN80Zpk,4128
tokamax/_src/tools/xplane_to_bound_args.py,sha256=BNCzbD9yXO1C8ftb9od14IjnWj2YvH9b6lwtJmP3M90,4111
tokamax/benchmarks/attention.py,sha256=7TdDDa2h5wqzneoqeT5IWe0aMzejN_C7sm_AlG8TLu4,6764
tokamax/benchmarks/common.py,sha256=eyBrNzVskGHa6S8PUscj8rnhQ8yuiXCNIdu2m7lWS3o,1741
tokamax/benchmarks/kda.py,sha256=t9llpylh5mNEtlqIQ2oN0LEi0-PMD2ADqz9bYqMjQ0s,3597
tokamax/benchmarks/linear_softmax_cross_entropy_loss.py,sha256=muhqAGyGAIUyaUxfbrNN-W6pRynqvdy7IFnZpAidVPg,3283
tokamax/benchmarks/mla.py,sha256=XiNOq5D9O_V4NGSw3zdxBapy5V13iYCZBm8MNK92IpA,4839
tokamax/benchmarks/ragged_dot.py,sha256=CwURLZ6YEcYe3M8rV9tmO8KYMorkrtK6OqrhycZFptg,10234
tokamax/benchmarks/triangle_attention.py,sha256=-_UDevuRv3yBZesNugZJqHTYSRAlTfDjPxouprkJV6c,7008
tokamax/benchmarks/triangle_multiplication.py,sha256=XgeivcsmcqOCxzOJysJu-CSmUXQ3OnRkxMqqBxKFzUw,7115
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/dot_product_attention.json,sha256=2CMUbZUkLpcogjQ9aGzBLToNDRGq_AmFajhJ6oLek_w,154660
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/gated_linear_unit.json,sha256=v69-hI5rj8i3nz_tQawcnqexh8TPVXEsEgmMKugORZg,113423
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/jax_nn_dot_product_attention.json,sha256=vJ06XKpbF1XNzU7nEKWj7d62gEO7MZOs1CgAcpV4fEs,132913
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/normalization.json,sha256=EtHU1wN_WZxkytJVhho2bjWnn1tOh3OedGhtnhCLCeo,4816
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/normalization_vjp.json,sha256=EFQXwFNKzYzNI331_3l5vav9IB6N0EQKnj20DHKdB48,2434
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/pallas_mosaic_gpu_gated_linear_unit.json,sha256=MfOt1p1N219Eq4wz7HjRQv8nsa7jVVHxRioLi5tbwi4,98545
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/pallas_triton_flash_attention.json,sha256=8c7p4-TjXmwLzFpSq22vsyCpKnUOgd7OFR9wZ_XPcnU,185845
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/pallas_triton_gated_linear_unit.json,sha256=1vsMc3EChNe2qp3hjXhWq2sfAduiJMxBDS-Peijojqg,124542
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/pallas_triton_ragged_dot.json,sha256=U2tyz6CmFSZpo62ZASK3F5LCLwnrWtELcz5eLLyY54Y,22716
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/ragged_dot.json,sha256=4QRFghu37v3PM6TnzysDWl5qNmSbID5Ewu_CYsDKQhs,55927
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/triton_normalization.json,sha256=O-C7zYu8x3h7HJ0McOJlZ33AIW5i7r6wAqzXwPqhH_U,5226
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/triton_normalization_vjp.json,sha256=rHQftC_-vmNpBbY7g15iV-gbiHj5nDul80zIOQ_mulg,3264
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/xla_chunked_dot_product_attention.json,sha256=JbwQXwMQAiGqKF83tYHBWiY6eO_KHT7ou_BTUu-9sJA,139064
tokamax/data/autotuning/nvidia_b200/pallas_mosaic_gpu_ragged_dot.json,sha256=U-KOfB8WQ4FVjamhSXWRXOfLvNQGo8jA4hBgaPoQCwU,70111
tokamax/data/autotuning/nvidia_b200/pallas_triton_ragged_dot.json,sha256=zH4S-wh5JLAv0oLWUV54hKfEfeKnIiJC4br-F-tu1Gw,95397
tokamax/data/autotuning/nvidia_b200/ragged_dot.json,sha256=oH0kUrRXgj54iWOQ--ynFAwQTjbYdXa9GT4GUpl8Tsg,59655
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/batched_rpa.json,sha256=qbr538-1Z0thso_qgjkXrTInmLtTzHUGdXpmaCWP0UI,10778
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/causal_conv1d_gated_delta_rule.json,sha256=7wFbgOyaMC0cbohntqglEmtNw4t9K8J6IeWU7BoPdNw,31376
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=PMAH99JL_a_S9hWS800Olv_lPP6GckZRGgzD-447dgQ,154456
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=pGMxV_arMOLSPvXImYmJnDmU-gmJMzp2SyrSbj-LRbE,85467
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/dot_product_attention.json,sha256=OKveeFqxkVU_lMJRdTKOhYsbxty8Co3mj8butHJdqyM,169418
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/dot_product_attention_vjp.json,sha256=3evWYyBFhzZ77qX2Kt32iX1l74wUoIRkjz8yx3KyN4U,117888
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/gated_linear_unit.json,sha256=PujT7eMcFWEtddqcYPwiXPIaAdtBUmHFOsvt186hYbs,124530
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/jax_nn_dot_product_attention.json,sha256=nMMPPdenKF-VWEiwsKwTGCDAfYx_S7Z-vVMNaUmzCy8,154060
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/linear_softmax_cross_entropy_loss.json,sha256=MN72j9cfEDvPazp2O5hyYryGQlvMrG5tO0O7H3uKrxY,125366
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/linear_softmax_cross_entropy_loss_vjp.json,sha256=_JA1IlkeLPLwnAlzL8NaCtaryFR5ay6X5Mc406T1R18,72098
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/multi_head_latent_attention.json,sha256=55iVUl-4s9Vc5mVsBVXlJ9kVAtjU-RPcnhk7ouSve6Q,2399
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/normalization.json,sha256=wLUycWrDYXQvccIVdr18ErfCtUVqCt5H-OIMqeKxcaw,5272
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/normalization_vjp.json,sha256=Oz7gUFw3wUjRiB7bbnUV-GV2fkwzigxZA_Td5jdmzi0,2416
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_flash_attention.json,sha256=WICx-ueUjhyBTDsvYW9YOaX3w9rJVle0mRqZ8CJ0WJU,201162
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_flash_attention_vjp.json,sha256=duv2eEzzv_9yTcCyXszV-7YqAveI0Gkt8Eu0LTJh2co,139644
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_gated_linear_unit.json,sha256=WZwoBBcKwpqxX_dI682osW3BGloMvFUA_GltWIZkzkQ,128308
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_ragged_dot.json,sha256=Oc1AhVFtIt9sNiswIP46o74P8ELPioGAxkBAFW2LMLw,119691
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_flash_attention.json,sha256=RNye031OmW8iQ3eD4Q-Tn7i6gZT1ggPCong5clrm6VA,202334
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_flash_attention_vjp.json,sha256=8Kbojt3_ukGHXR3QwgtBapYc83z3a2aEOSZLLCTU27I,137772
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_gated_linear_unit.json,sha256=uRWPDPVmj8nzO2CWp0DkyQCXRDSCr_Cz1B1fzEF3RnE,129974
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_ragged_dot.json,sha256=5fAKLaCabHwbrMIAeu9IBBR-2AK4bYg94TuXCYPyNNk,49357
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_dot.json,sha256=yXk4OZSWpB1BvUKgMhxvWit9schsZ5kKndZ802ALIRE,68779
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_gather.json,sha256=LIe9Rzf8hd9-IQuyiTmsbx8A2WTSm-7TiaV41nSw3bQ,4224
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_gather_reduce.json,sha256=oVxuALn6S6SYhieAc01i2JbNiFrKtnA38-heYhIvlRk,1892
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_scatter.json,sha256=OpWXsrgx6VlDQ7vTSrRRCNay3FEXmFLZX_bLUwOIyeo,1723
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/splash_attention.json,sha256=a-DeBOqUicy1V5gFDCQpsj8PFwTlMzHNzIyJbEN2gVk,3104
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/splash_attention_vjp.json,sha256=zbryFkUzkmooq3jvOcoYO1LyoFgRrvXJn1xRol7Wpsw,1899
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triangle_multiplication.json,sha256=C_At5fvgpp2N6dgRmSgYO9WhJ4MybF17E4OtcBqffzM,13025
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_normalization.json,sha256=NJ3reixmG-8iXRS1w8-avTspe1Hp312CK_--7T8SZOY,5363
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_normalization_vjp.json,sha256=HZruCDQpJRLo2poVQEk-IC-t031oeTKRT3SIBHsdJzc,3249
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_ragged_dot.json,sha256=m8_ZkIxeItSKyzoOCgkJQ54ixCoj61R7Iq_L1LuYJ7s,104186
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/xla_chunked_dot_product_attention.json,sha256=ee-gTDTHwL6b5IovcPF32qIElR0X529vmuvkk2Q9t50,153206
tokamax/data/autotuning/tpu7x/batched_rpa.json,sha256=6hiHrlaaeks_LcCEizcKeDVsr-U2R-A0pu1NCI8F_SI,54504
tokamax/data/autotuning/tpu7x/causal_conv1d_gated_delta_rule.json,sha256=hwCohI4FUrRYSGJuSLYbpRoiHp1uYD74aWWvsdsZto8,20334
tokamax/data/autotuning/tpu7x/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=U9aryLBk5dAnaciZwvN_LoDJU0n8vm-KofqEn_Wkao0,148634
tokamax/data/autotuning/tpu7x/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=hgwdCO8-SZz2VR8NzNAUsezcRllHOJ3jD-bFsSuq-fM,85931
tokamax/data/autotuning/tpu7x/csa_gather.json,sha256=UQrICpkw-KCfjaqg2NAdbZodwBnm0TLDmF5bs8DUGC8,1376
tokamax/data/autotuning/tpu7x/deepseek_rope.json,sha256=zIQXA-DBlvCxOBN7kp6DLzvf_DT0dV9GPqsksTK68q4,1945
tokamax/data/autotuning/tpu7x/dot_product_attention.json,sha256=sfNirlaZQOT_7x_oaJzssGVUkU6psYor4ENDAVUo9I0,166539
tokamax/data/autotuning/tpu7x/dot_product_attention_vjp.json,sha256=Ro9iueXL72Nd9cWTJX5BAWYJC-S0M2sjk65C9pr2_aw,113771
tokamax/data/autotuning/tpu7x/dsv4_compressor.json,sha256=oETvyD-MW3w8UCUVOf6_T2r-8DhX7A3sPkTiZNs2Vpg,2871
tokamax/data/autotuning/tpu7x/gated_linear_unit.json,sha256=Kud3iB6UMMUKNJ5D0MyuBbCUO9dBhx106zBYR7MMSLw,120192
tokamax/data/autotuning/tpu7x/grouped_lowrank_proj.json,sha256=fldaH0VsXc3mHJ1DKDayuQR6lZJ0ufiNLKTXQP3D0m8,1564
tokamax/data/autotuning/tpu7x/linear_softmax_cross_entropy_loss.json,sha256=uHUXecHgiexovWrpHdF5wdSr2xJhtaHbvbL2vQc13c8,131794
tokamax/data/autotuning/tpu7x/linear_softmax_cross_entropy_loss_vjp.json,sha256=oBjpnxvRnR9PJ7kS64urDywOolWfn1AoCGdQO-RON0o,116534
tokamax/data/autotuning/tpu7x/mhc_fused_post_pre.json,sha256=I6U0TCzTSVJxFfVXOBr1aB6JxxSpTGWLKxH9OvYD0gE,1883
tokamax/data/autotuning/tpu7x/mhc_post.json,sha256=E_pj8Ay1Dezl6S5O0ltkDDUHUThLowHpIK5Ypa8Iq1M,1326
tokamax/data/autotuning/tpu7x/mhc_pre.json,sha256=GGNdQNAEbY3e4qPTswnMzqqY8UXMJy1VgDa6ovNvv7c,1597
tokamax/data/autotuning/tpu7x/multi_head_latent_attention.json,sha256=qKaexY1S9HqjQff1YBdgDjLTawQnpl0jb2zc9wbN3Ec,2339
tokamax/data/autotuning/tpu7x/normalization.json,sha256=JfzLw80F2So-Ir4fwjE94IR1VtCzBJViRVtDwj5H7Vs,4637
tokamax/data/autotuning/tpu7x/normalization_vjp.json,sha256=VUgS55frODqvt3XVwpSZ1J683X_laMbubkK72TZyC_U,2363
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_causal_conv1d_gated_delta_rule.json,sha256=-nCd1iaPmxKuos4lXxJwPlx6x2yjMROSKw5iNGFm7Ys,43065
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_flash_attention.json,sha256=rKpO-GQIV05lftabFJiC65msOVjOcRaLz02zU4l6La0,143875
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_flash_attention_vjp.json,sha256=zsQ-CQUWPAzlKPxFsIZEJoOX8Fo1q6o5tRnLS8pKYLY,94848
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss.json,sha256=sn-6dKaUSGVcp0vh872v0Rq1zwPLtVtpl6O-r_i95bw,163074
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss_vjp.json,sha256=2Upt2m0pWEVcYYiUxm9h-JzE4vbs4QB9n3zg9Rht6UE,91588
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_ragged_dot.json,sha256=_IRlHc_Zgg5G8GhaYF79L4a2uMeYVrjW8ltt_DPKepA,102998
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_splash_attention.json,sha256=tzf0hozTfzK-x58SJ8_jmTWp6jKygqcbpUtHoCixVc8,50080
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_splash_attention_vjp.json,sha256=uWWsFa2bkkkHK4RpmOTJpLJRtFyQj-oNd8uyJ7b0l9M,25222
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=oD0Ax-gdpMPB2GbwRstUZbDLNcWKpAsqnTYPTja8nDw,66599
tokamax/data/autotuning/tpu7x/pallas_tpu_batched_rpa.json,sha256=ep4nM1naMwUH8da0NhHlh2PJ6QyGlm7ABKh52hdngaU,56480
tokamax/data/autotuning/tpu7x/pallas_tpu_csa_gather.json,sha256=6b1O8ojrcQm8nA8EeXyAPvpp2fbTxOfLyJoqBSQeAXQ,1427
tokamax/data/autotuning/tpu7x/pallas_tpu_deepseek_rope.json,sha256=2jcje1cX_720eY8dOTJ-m__8lP7mb0E-rPbTVfRuhL8,1998
tokamax/data/autotuning/tpu7x/pallas_tpu_dsv4_compressor.json,sha256=SeNbzS_u87uXF1YYeR_rOran6WTTl_um9UmtKuaxcf4,2912
tokamax/data/autotuning/tpu7x/pallas_tpu_grouped_lowrank_proj.json,sha256=NPWnObgwlZYaOPAjoxDgLttBzL-ZyZ6tqQOW3gYwc7U,1698
tokamax/data/autotuning/tpu7x/pallas_tpu_mhc_fused_post_pre.json,sha256=b3TCK9uSEgn-2utWGzod6fvm8D66IWYHSxvSjX_6r8c,1951
tokamax/data/autotuning/tpu7x/pallas_tpu_mhc_post.json,sha256=gInuiiDPzhUSPmcc5qCd-I3tdDyVet-PrbKD3en1oI0,1397
tokamax/data/autotuning/tpu7x/pallas_tpu_mhc_pre.json,sha256=1_XKWnhYqplTy5FeM7Pjh8UkPyralzEK-GchARpfWbU,1670
tokamax/data/autotuning/tpu7x/pallas_tpu_proj_and_save_state.json,sha256=C3l7G0slzv7d4d5557nw4kBPBHj1msDzbBD6ajnfLTg,2222
tokamax/data/autotuning/tpu7x/pallas_tpu_router_top_k.json,sha256=-pBnLAXpSX-lesjRtD5x_iDuNGWsKfkgXTUF0DVRkls,1477
tokamax/data/autotuning/tpu7x/proj_and_save_state.json,sha256=3-puiaDEeQG8iZXNsTS8RzgJgLUaKEXAGPlNF0P9aMk,2098
tokamax/data/autotuning/tpu7x/ragged_dot.json,sha256=mz53jYD3fNHx9UfBWdcsKYW47ebbcoChWoIj6uULP_Q,94293
tokamax/data/autotuning/tpu7x/ragged_gather.json,sha256=OIGm8BqKL9zBKWTctipngO1RdrpcbYh5bsCcOrrwc-k,4076
tokamax/data/autotuning/tpu7x/ragged_gather_reduce.json,sha256=Ly8jmO7qaWCUqAKpA-Yg5kIlkEqc8L0aHQf7g5_lLLY,1833
tokamax/data/autotuning/tpu7x/ragged_scatter.json,sha256=6btLohvvbFZGOw33VoaUPBqZp-9PkEJCwY-fSzwyHxg,1628
tokamax/data/autotuning/tpu7x/router_top_k.json,sha256=Sim0N-vXZn_0w7mjEOHklWFpTQ_93YrkYje6TFBgJl0,1411
tokamax/data/autotuning/tpu7x/splash_attention.json,sha256=YQZyFsxXJTC3wkhqcavtV6VdlLJBHn5s2rhbNBoGBuo,3577
tokamax/data/autotuning/tpu7x/splash_attention_vjp.json,sha256=1aknf_-iD7KmEs8RHjZhJxpN-c_5P3ZiL6Lxo_yVbns,2137
tokamax/data/autotuning/tpu7x/triangle_multiplication.json,sha256=HgHMT-NCSsHP-lZNFEOganPgV0zJYYVXdIsMtcJE3os,12738
tokamax/data/autotuning/tpu7x/xla_chunked_dot_product_attention.json,sha256=STo-q8wRDwAVex7V5Qs5NnE4YyzTWsWSnJmnylQgpuk,144791
tokamax/data/autotuning/tpu_v5/batched_rpa.json,sha256=9JReye7caSR7vcVjrOaya8ggUlZBkXAzqi8XqYm7HNo,8474
tokamax/data/autotuning/tpu_v5/causal_conv1d_gated_delta_rule.json,sha256=Cl5YkD_-cbgf_ae9lHfHzoe3FiK-Mjzu1zlt9F4FxP0,25731
tokamax/data/autotuning/tpu_v5/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=ep1CUEYZNmhW7clavQZov3RTqwMGEz-GuitoZgnFfIM,117397
tokamax/data/autotuning/tpu_v5/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=B2jrShBOfGv3sxhNbyNdlmG9Iv_rxwveyP6vYHKw_D8,67562
tokamax/data/autotuning/tpu_v5/dot_product_attention.json,sha256=vQXxE-s28f6b-AjhNpMLZaXvTjTEkdIgsVR47vd3mTQ,159810
tokamax/data/autotuning/tpu_v5/dot_product_attention_vjp.json,sha256=9m_5E8o-VwyawTqcS2rsIPZznLPUM7eWbZW6mMDQ8tQ,45283
tokamax/data/autotuning/tpu_v5/gated_linear_unit.json,sha256=cQqkCrr8IH8UZ6IEK2Q8II-T86PLWvvHm_5n5iKNssE,121636
tokamax/data/autotuning/tpu_v5/linear_softmax_cross_entropy_loss.json,sha256=KExjGMfhPPKs5lEkP8J8VrVhrjVTYO50P8PpU9ePOF0,106645
tokamax/data/autotuning/tpu_v5/linear_softmax_cross_entropy_loss_vjp.json,sha256=YbnmZ5c65O02k4Q7avdigK1vuFoqR5dYZADnTXZYM3Q,62129
tokamax/data/autotuning/tpu_v5/multi_head_latent_attention.json,sha256=lBG8Ce463QEZcjw9LO1YYLNXpiOmX8y7xOSJsxS1_lM,2341
tokamax/data/autotuning/tpu_v5/normalization.json,sha256=ni2Lq0LbYE7GJ0Bc-F_2e6gA7XZ7fc23JeaSZOSKv5Q,3710
tokamax/data/autotuning/tpu_v5/normalization_vjp.json,sha256=Oq0LwdPVzuNLw2Z3q_wuqBM8a77BU0-vOGdxEZ_CqwA,2337
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_flash_attention.json,sha256=BmbHDDfHLVIH14I6Ds32lwSSDB1ACESDQuJzOtlVb2I,139426
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss.json,sha256=v53qXqkDyufDDic6PFpXxBaTHDAAC9hzYK0vcqcHaEk,128518
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss_vjp.json,sha256=ryfq9L6ij-xl5BNyyc6OTJIO5seOSY9Hc-wZvsN5U_c,69601
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_ragged_dot.json,sha256=AWcowgygEGXgMVD9QByBXESWh-Q4lsAqiEw0M6OT9ac,101748
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=HeDx0g_n8FBxkGjYmijUPxwzdYzgUx7V7Hp2J11oqBY,62129
tokamax/data/autotuning/tpu_v5/ragged_dot.json,sha256=fr1TG2mF3oHcxvJlDQXzggjx5Ow4wERQlDJrzNZe3NY,95263
tokamax/data/autotuning/tpu_v5/ragged_gather.json,sha256=i_YCc-E4fYsf76l33XjCmUd2A2WI4EvOjJUk8ELYFyw,3255
tokamax/data/autotuning/tpu_v5/ragged_gather_reduce.json,sha256=CnTd0oUVPAXrT7AL1wgWj7Ts1EUzBhRBDRFPuRTA5Po,1835
tokamax/data/autotuning/tpu_v5/ragged_scatter.json,sha256=RHg5ks0iD5mP1jSB5v_LUz1ZHz3MK3CeXnLQJNxYl2o,1625
tokamax/data/autotuning/tpu_v5/splash_attention.json,sha256=kh6i6BoLdw3B3-KAaswELtQuWLKIhIhF_QouwIuzn0I,2984
tokamax/data/autotuning/tpu_v5/splash_attention_vjp.json,sha256=iJRczLbwoLTbwghACu9NOO1pv4Jv16ci03xySPimVMU,2144
tokamax/data/autotuning/tpu_v5/triangle_multiplication.json,sha256=PKqmhIDUBquQppbhWMsDNpQr8ZjGjqo2c-l-JQpUExY,9523
tokamax/data/autotuning/tpu_v5/xla_chunked_dot_product_attention.json,sha256=7rZ9BIi6tm92TOvu2B48F6nTgGmmP2Co7tTOKKtKbaU,138008
tokamax/data/autotuning/tpu_v5_lite/pallas_mosaic_tpu_ragged_dot.json,sha256=ZSGr6YJxQvOaNW5XsZi2bwIdjmTGS8x3AGBjx4VcheI,36354
tokamax/data/autotuning/tpu_v5_lite/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=AB6uzRLDKy5dL9d68CcdP30wtpcITpR7EgH42ff8XN8,498
tokamax/data/autotuning/tpu_v5_lite/ragged_dot.json,sha256=d_0WfKfk2ajwsYIDgQ3CG_JoBfnihE5rPLxkUqr-2K0,155947
tokamax/data/autotuning/tpu_v6_lite/batched_rpa.json,sha256=Fyvlml0tCqSDGXNzWZ3JdgEdCBzbaaml5qZQpRv787c,10033
tokamax/data/autotuning/tpu_v6_lite/causal_conv1d_gated_delta_rule.json,sha256=SkZ7DUdlRxueg1fuPLTmHSYaBfTEKcGls9KicSJnFzM,30642
tokamax/data/autotuning/tpu_v6_lite/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=OICV1uapy64VHkdNfHK8ZVIp79vVD3PPiDk-IntJEac,149039
tokamax/data/autotuning/tpu_v6_lite/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=0EcRPrc7PjlXe6spAXL5sfiynxSbTiJhLdqB1RCZHaM,85550
tokamax/data/autotuning/tpu_v6_lite/dot_product_attention.json,sha256=8QnXVOpNLfIjQoDANNG62ATEKCsuj7FZQOWImPk4Zl4,165154
tokamax/data/autotuning/tpu_v6_lite/dot_product_attention_vjp.json,sha256=SZsrFShM2x_0XeJnPVUWg-Oqjo-7QYvDYHJcWe-lrlU,55230
tokamax/data/autotuning/tpu_v6_lite/gated_linear_unit.json,sha256=Udv01hgx4KDg6EXc60iZRsQg9Mhq1A9tBWeDLHIxOXw,104641
tokamax/data/autotuning/tpu_v6_lite/linear_softmax_cross_entropy_loss.json,sha256=-v_UwiS6b0GElvpLRRsW4G0tt7ZVyi42833UrLCmqCw,118127
tokamax/data/autotuning/tpu_v6_lite/linear_softmax_cross_entropy_loss_vjp.json,sha256=VPU_UZIVEC8GXTz8jkvsN-r6cQhqKXOSWCoMZMa5_HE,68047
tokamax/data/autotuning/tpu_v6_lite/multi_head_latent_attention.json,sha256=PRljbjfJun6dEutlqgNRzMz7zxH2rMl-gx7PgdQ9p1g,2341
tokamax/data/autotuning/tpu_v6_lite/normalization.json,sha256=4rsbGy8fsVqb1I1usBAXJVky6yg5Sy6ZJLtPrLB2qpk,4630
tokamax/data/autotuning/tpu_v6_lite/normalization_vjp.json,sha256=1DspvOteCu_WMRrUdwro3UcU6FE5yQ0sTOhfjdAMdvE,2339
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_flash_attention.json,sha256=w_tKRwPmOF4gjLFweASeV7m-0rqMegfbv8MnvC3Shto,141219
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss.json,sha256=28_iJsfQ7Ccp-aejC3AP-YkKT93xtPvhuAqJakHv0e4,162109
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss_vjp.json,sha256=9qOQzRBki1BJMcVqjAYm1h6M7zM-RS1KyNfspcne6z4,78788
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_ragged_dot.json,sha256=P7MRfqNW1QTcIwvOQeLfoHGs_9kE1GN1_Xf4lpQpBEw,94894
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=YYZ231ALoztpXcHMIbjG2ZUea0K6zqm2P_wp651KNaY,62449
tokamax/data/autotuning/tpu_v6_lite/ragged_dot.json,sha256=I5V3u4c5v2v9GAKAi8_AeDwyvxYjmGLS44YUBYgqEEY,94049
tokamax/data/autotuning/tpu_v6_lite/ragged_gather.json,sha256=A9gieprmFSVsmq3Vug90JzJii68_rgzo1_BCWHIpg5Y,4064
tokamax/data/autotuning/tpu_v6_lite/ragged_gather_reduce.json,sha256=SHDIaOI18J5Al4xoDnwsmtN8xETeyv9W4xJB5i3aspw,1836
tokamax/data/autotuning/tpu_v6_lite/ragged_scatter.json,sha256=SxJJQTO8R1k6ltS_wib9FmT-zNi0xMIcJpCNDBg49b0,1617
tokamax/data/autotuning/tpu_v6_lite/splash_attention.json,sha256=oxn6ugVxClFQyaTnsQRy6Ati1o8-e48jyPWOHixZK5o,2734
tokamax/data/autotuning/tpu_v6_lite/splash_attention_vjp.json,sha256=2FcVMN-zLWre_vHKuhNS_8r6ohiOv45R8nR3eiXmYrA,1565
tokamax/data/autotuning/tpu_v6_lite/triangle_multiplication.json,sha256=8tHJ7WKtI2u1iSNNye5mPmyPS4ZpF2UAQhjWjKPihpc,12716
tokamax/data/autotuning/tpu_v6_lite/xla_chunked_dot_product_attention.json,sha256=gU3lVcWciVScC8hU_74GbxjR-uxr5v4y2GHy5KdeCIY,151093
tokamax/experimental/torch_tpu/ops/__init__.py,sha256=BMN1NY2xJ9E6UfSV_YkyraeHE_H2KIlzk-kNGjuYr4Y,1887
tokamax/experimental/torch_tpu/ops/torch_op.py,sha256=_Tcc_RRrbSPmbkgUzZ675Gzparl_gMbSMtplf1D7Lno,23458
tokamax/experimental/torch_tpu/ops/torch_op_test.py,sha256=ezhCbm8262goo0-jSqkX_VoD55YNGZ38v4vSTyTonPk,40310
tokamax/experimental/torch_tpu/ops/torch_test_utils.py,sha256=coV8PeFYHNWYlYJgIUmdRfAbpyDsNfa9LFDwKg_d3ag,1505
tokamax/experimental/torch_tpu/ops/torch_utils.py,sha256=bnny9UhrMH3f40ihKUTV5F9SkZ9Dc3yjEfzvkYCAL1k,10936
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base.py,sha256=43ssgh0CQuSBE82mSzF0L5BZ-NbR5Sr36VEXP1_De-U,7895
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base_test.py,sha256=18SJBN9eUVC8eXGcW4vLygFHeNzp3B4GH1x8eTuH6-c,3940
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu.py,sha256=z35vgryirjjiijLuGpPJFB3UMpgcQAoacQgD7qNvJVI,9983
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu_test.py,sha256=pMwTkii3ysETDZtRMH_n_hbtQ2I5TFHmMM1F4LCaG6w,10190
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_base.py,sha256=tZxBu2gzDyT0MXiBNbDnOyinLokKbHTPiONnPIcq7uc,6133
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_base_test.py,sha256=GAq8fPOvg0N9J_3cIBD1l-GSHfdmyHPmFivHZbSw3AI,12697
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_pallas_mosaic_tpu.py,sha256=fmFmHyBMjLrHVu-Rdx7cKLg3_vo5wuhf6fQQPhik4A8,9692
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_pallas_mosaic_tpu_test.py,sha256=RPMf3rzr-NlvqtIJaNATJAA5kM0NcwIaQVF24umGkmw,17068
tokamax/experimental/torch_tpu/ops/csa_gather/torch_base.py,sha256=61yhigr1fIDNTyu3deTCP6LVBpGUWvk6QL7_Rphv7vM,3384
tokamax/experimental/torch_tpu/ops/csa_gather/torch_base_test.py,sha256=8zuCeRyCIW7uuj0jqETyjsYLOtA4WW6dWbt7q-63T10,5360
tokamax/experimental/torch_tpu/ops/csa_gather/torch_pallas_mosaic_tpu.py,sha256=tMI28CgKhpMS9rSzdg5nvnY8IhR2CP_aSanF7PUN74s,3752
tokamax/experimental/torch_tpu/ops/csa_gather/torch_pallas_mosaic_tpu_test.py,sha256=OkcKnqOEZQU60hgMC55NJODNNazAyOXlTdLzK8DAEpI,8704
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_base.py,sha256=9ud3J5YzqucTlkXzwXZyklj8ksYvCZl-TSyTGljEqhI,5649
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_base_test.py,sha256=G9-FjKJJW617o1IjsRT9oH4o28MBtJhq88eqzxIIsq8,10349
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_pallas_mosaic_tpu.py,sha256=5-0Lwin2oo1lyaoN3LOyQ17LgOyQG2527seaD3qFNJo,4802
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_pallas_mosaic_tpu_test.py,sha256=yMsDymQxtboN8m6W1fgIY_D8EhpdgxUIOjLz7Lvuq2Q,13789
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_base.py,sha256=YrBMRJt3RrAiwUWHauE76V75I-lyjvhGQwNjmyLj1S0,5658
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_base_test.py,sha256=dos2_mcVZl9J03u_7YXIbTQZkX-k62fk06BdY9IANlY,12550
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_pallas_mosaic_tpu.py,sha256=qJ1Nj1KzQ7trCbaPO15BPUGlV-x7rqTC0XoXSEZkj88,5641
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_pallas_mosaic_tpu_test.py,sha256=mTTRCRmRx2Ooep-_mzoWSIoUfxWohDy1rJgegLtlqbA,17613
tokamax/experimental/torch_tpu/ops/fused_moe/torch_base.py,sha256=xsfPKsY76B4gzIWC7NpOiryvg401nwijPL5Vm1ts5CA,3133
tokamax/experimental/torch_tpu/ops/fused_moe/torch_base_test.py,sha256=X6jtXYJbAFOKcGSaJJ860KJNjx1keGrDi-x22xBejIU,5109
tokamax/experimental/torch_tpu/ops/fused_moe/torch_pallas_mosaic_tpu.py,sha256=vXiXxDo_FNYe-yQWkc_cIL1SWPlyVRT-RU5HR0W-4cY,4975
tokamax/experimental/torch_tpu/ops/fused_moe/torch_pallas_mosaic_tpu_test.py,sha256=ZLMfahf3BSsoy5moaaOfEh_4TgDfY2Qbe7WPQabjFas,8407
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_base.py,sha256=OVV78HQT5ur3XPDm3GnTpqDY-ZWGTGlmLeo6HGMJqHs,3577
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_base_test.py,sha256=pgUDN3PhWcnXSLES3KK9hCEt4s-zuq-MTD85vXjkoAo,8774
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_pallas_mosaic_tpu.py,sha256=G6DjYrLLslYtE8LeTk8f9nLoOdY9pVKNKERVCsl5i78,4539
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_pallas_mosaic_tpu_test.py,sha256=fkfeWlmDMLOPEi6wlvMX2_lNxLlOrD-gg04qJKVXmog,14453
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_base.py,sha256=KQjtM4A3N46GqPUO64umOafQ9UkSlH03r_DQsV1-SbI,5293
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_base_test.py,sha256=u-JNNZop6DA37ZgvCWB6UNjS5nzgW-nJobSX4aRc6W8,11188
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_pallas_mosaic_tpu.py,sha256=XfjsXI8m4_aIryScppRjZUAfd9LwZtl27mRiwHwiKW8,6971
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_pallas_mosaic_tpu_test.py,sha256=nFj-2FFGA4ic_ug8oHP8wG-tM81xSNfaLWfP8Yiamxg,15859
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_base.py,sha256=t6VzHFTStBnbV8w78Tc9Y_TzS12nTwR6I_cjTelZBII,6977
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_base_test.py,sha256=qm56BmrTwXS5JslkYdIjfX17eagyxGyONg8qTXGQE-E,6015
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_pallas_mosaic_tpu.py,sha256=FTa-qRRAsF2OHBhR2uQzqei1uo5qxzlpMooNr7UPl0s,9530
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_pallas_mosaic_tpu_test.py,sha256=N-5RhKWnyOGmbWVKGrNTToNN-y9CuTuAUo3lFtL5hZw,15267
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_base.py,sha256=4o-ElTWzwKIVbp4mOnuL-u1jdkr92qrug5wMMlFl00E,4967
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_base_test.py,sha256=ygOK0fZNom2-jqdbMG3JeLKOlOWm5BQ6M1_iGqOnvPg,12160
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_pallas_mosaic_tpu.py,sha256=3UzUnOQnqKNZERMrd08ouTSQaTuxCV8isdcMOt2oZVo,6083
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_pallas_mosaic_tpu_test.py,sha256=OF6O3IgP6P40BwZn23r3GOVhPcgfbhnF3IwxhISkJBk,17863
tokamax/experimental/torch_tpu/ops/mhc/torch_base.py,sha256=B5sr_ITae5ox6jUx2QHIAZnVN6hAk6mPg5tM3W1rhVE,9874
tokamax/experimental/torch_tpu/ops/mhc/torch_base_test.py,sha256=4YTJeJf-QhgeFDkJIyAjWCNXzqm2ZF9Y0soty1w2KpM,15916
tokamax/experimental/torch_tpu/ops/mhc/torch_pallas_mosaic_tpu.py,sha256=sXLpfaRRTPMTS0xFPO8OAXlJZKiBO4vW_ZX0ab5tVAo,12121
tokamax/experimental/torch_tpu/ops/mhc/torch_pallas_mosaic_tpu_test.py,sha256=KceQw8ji0mUjaW_JEN0A4aVMOSoMpwHNMhkR8vTDIWw,25717
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_base.py,sha256=1oUUUT5QreMQGtF1Ndnv1sd8nOGHD8aMFFZw2sCOXNo,3645
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_base_test.py,sha256=QOJbh1LO976wHYsBtay6s8Srroy20_VafhO3XCMmZgo,9325
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_pallas_mosaic_tpu.py,sha256=j9u1pdv1bbKczI7u2e2ue1ViUIRTEPuWFxsz8Lux6qQ,4419
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_pallas_mosaic_tpu_test.py,sha256=OtcvJXdk3t60pWHlOeMUtgx2c6mS_QcMjtXx2uRpFFg,14395
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_base.py,sha256=-mTSX3Ci_H50idrjo7Es-cmnlBaavwPvzpOtqTIQpvw,20667
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_base_test.py,sha256=T3_83MbazqGmDwTiZjh-8i2kHftQAyFavlHxNTufYSM,8559
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_pallas_mosaic_tpu.py,sha256=zADFCE4RiZbyaOCLTQBbIuqQ_Nou3XpYz27Gpi6SlIU,8672
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_pallas_mosaic_tpu_test.py,sha256=WrKalLD7qsXwoNLKd9MgDKFWWPhbc4lyRexJkmdhPXM,11615
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_base.py,sha256=G8HivJMqc2kxSs5xkrL6GRUmqKyCE0V9wWudzyBoATI,2912
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_base_test.py,sha256=QCe11ftSKcypkVRfTTKz5I7odh6Yq-wZxDznVLib_qo,4439
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_pallas_mosaic_tpu.py,sha256=w0iRZzIenyHcQFK6RHRMe5pKnazySUNDULUbcF8eSPo,4451
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_pallas_mosaic_tpu_test.py,sha256=iXVGPHVNGJnvZWTB0PmKH3WQs3edLiHryIXkRpfY98k,6940
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_base.py,sha256=ZONWwoBhFcwF-Jpajo1MnicCsyPoqcQ2emwjgCj-QeA,3294
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_base_test.py,sha256=V5dwXjxM9RYtfBVmdUb-Gl8voEFiSqXMhWd3Ldzc7gc,7007
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_pallas_mosaic_tpu.py,sha256=dTRC_tKiFgpmH0wUS6TzQM8OCoAiW8YS6ErUey5lyTQ,4006
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_pallas_mosaic_tpu_test.py,sha256=hQDCExkqdicFilnCJlPbDOEoBfenkzNdeZM-8U1iFcM,13841
tokamax/experimental/torch_tpu/ops/router_topk/torch_base.py,sha256=CIeYTtHxqlloZjXlX4wmQRJUQYRQJ9nVmxnWEQtOnsk,2937
tokamax/experimental/torch_tpu/ops/router_topk/torch_base_test.py,sha256=WqPCPF0rlbQPDEE7xr46mNl2z1OiaFKfG0k98hX957Y,4477
tokamax/experimental/torch_tpu/ops/router_topk/torch_pallas_mosaic_tpu.py,sha256=jES1D-WNZudAJt_X_avSDB10gTRqzam7Br7ANvCjGck,3505
tokamax/experimental/torch_tpu/ops/router_topk/torch_pallas_mosaic_tpu_test.py,sha256=8JkrWyhQv161CPJ1fzWmWNBOk-eJchcjPMpa2UGptIs,6529
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_base.py,sha256=i2CyhcdxAsdOGI3om-RYFPBZGjj2Cz80_Usa4A1nAlI,4336
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_base_test.py,sha256=fQqQPBbP4xwJqiIfxnnDUqejgQETJZQntuKBuqtq-pg,11978
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_pallas_mosaic_tpu.py,sha256=uA_w6_gD7L_myVuQBKOdIMko-umYbUi8H10hc2e4-5g,5020
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_pallas_mosaic_tpu_test.py,sha256=NfWnZWdxv7nDAekpZqEp-U0jNT_I4lxz50bop7BHSjc,16695
tokamax/experimental/torch_tpu/ops/splash_attention/torch_base.py,sha256=vqaXWDENk-T5BXPltItzNryLAHXwCEcZu0ImpgcQaeA,21606
tokamax/experimental/torch_tpu/ops/splash_attention/torch_base_test.py,sha256=Mw-3k80dPS8pylUoorHZTxfZBDBxQVLG2nf27c5ufhY,8828
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu.py,sha256=iGxQO5hmsUcQdL2nkXapHNyxDhoc-qDN6y3Nf-U5-EI,9240
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu_test.py,sha256=SQfykRY45clBDgVeM_4NXfHXlJlGtRHAP5-rXeC_1yM,11492
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu_vjp.py,sha256=xvEUoDMZ1td6IwPQODOxK1Qy8tL0cFk7dfwsZh8lrpk,7028
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu_vjp_test.py,sha256=u0tHM0e9BwVrh5xj_DV53DMp3AE4yPnM2pjIPHBz59M,10058
tokamax/experimental/torch_tpu/ops/topk/torch_base.py,sha256=ppiKz9RR0G1ElMlYuwj_konTTfdaXWWIOeKyzwFLp1M,3552
tokamax/experimental/torch_tpu/ops/topk/torch_base_test.py,sha256=oJGGxob_U24ZMBzrg2b7-jqX_R5ZHGkHN7gUjB25uSs,4828
tokamax/experimental/torch_tpu/ops/topk/torch_pallas_mosaic_tpu.py,sha256=KgM4V6UQ3MznsLRaiA4ybZg7C5fFtaGZ6LufjJs8Z9I,4064
tokamax/experimental/torch_tpu/ops/topk/torch_pallas_mosaic_tpu_test.py,sha256=ea1Z7dtfSdsTTq4ywWPYd0xXxalgyqvXHNWDNn_JFH8,6892
tokamax/experimental/utils/tuning/tpu/ragged_dot_benchmarking.py,sha256=ve5LQUw-OdJWW0V7xkeJn-ryLAmgM7LvcxOClLhevu0,13063
tokamax/experimental/utils/tuning/tpu/splash_attention_benchmarking.py,sha256=CtX-KoW-Wkz9QU3QkUvtCbEluyUeP6sTPcz-TxsU0eo,10278
tokamax-0.0.17.dev20261016.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
tokamax-0.0.17.dev20261016.dist-info/METADATA,sha256=UBgLRPbkd_N51c8O_5IKwiw5Z6UuF85ksNSfTIzc9TA,17053
tokamax-0.0.17.dev20261016.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
tokamax-0.0.17.dev20261016.dist-info/top_level.txt,sha256=xHgO7qOuYjrLH3047bLQLnwrjI76i9-xlChxNQIZn-Q,8
tokamax-0.0.17.dev20261016.dist-info/RECORD,,
