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=cuPp3H7BqY9aylFXlSvgcX4i3Zmxns9jFPHmsUvoy7g,900
tokamax/conftest.py,sha256=qjs1VrvMMZeefevbOkpj-TtXSqT2sZGJj-nQHH7Jrrg,1269
tokamax/tokamax_test.py,sha256=_A7IMcPpkz18NSnqjYz87c4yL9ErxFPuyKBi4JqLUR4,5595
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=TuvBNuRYMgfZzGlDIJPytRii1u4MfLNUG8knevXyFdI,27518
tokamax/_src/benchmarking_test.py,sha256=mmCEjemEFXkS82D0IfzYu02eJpwMe3_Yr3mFJ6oCeOI,9757
tokamax/_src/config.py,sha256=vrPnOxDUnEGKiPuM60s5B4T3XrgtLD8WniDmX4A1wWI,4163
tokamax/_src/config_test.py,sha256=ZShG9IQj0YbyWp4ontfZi8c3YWTyxotsd4eYlEfE0OQ,3946
tokamax/_src/gpu_utils.py,sha256=suxOk_9Iz8zyCxXQ3P7d2F3kW2wncsf9uNe5Ca-AU08,3024
tokamax/_src/gpu_utils_test.py,sha256=5QMPfeky8zHTUwx6e9a2OBYC7T_olAiFAlj0GCxR5mY,1687
tokamax/_src/hlo_utils.py,sha256=BaxMUnU_broZhwRxl8XnzyC4fCdMvXlQ6qNS8KDIxVI,11417
tokamax/_src/hlo_utils_common.py,sha256=61tNEN3GYBJkupCvnx-aXwfc5nshRrHSb8aD6pQFAtI,4265
tokamax/_src/hlo_utils_common_test.py,sha256=AQnIVl2h5KoApga-tIypEBWKS1N5RXuLSBHI3ISm7B4,1974
tokamax/_src/hlo_utils_test.py,sha256=FJKXNLm_kRcwbO9_mEtBEB4l7HJQE6N64LuIX_ga6Nw,21025
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=xUNXV38Zf-Z2rdAZLCTaauh0_ccxps5d4bW14xItUvw,7336
tokamax/_src/precision_test.py,sha256=fFifAxvC3-hv7tY4qLXUj6313laQkeIB9AN0c4uFxC8,3630
tokamax/_src/pydantic.py,sha256=s4Z32AhHNqmGkaFX15NG5ETZXiYAbCTC1QqV120mugo,13450
tokamax/_src/pydantic_test.py,sha256=RMIOLoc-Epg03YIIIe8YFw3WwW57Ie6_kV2xJ7f1PjU,10214
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=ZV0stuomkT0WxTdIiXhOA9fRfMkImWOu3jgQ7tQ5X9s,5066
tokamax/_src/utils.py,sha256=M99bvel6cptkiocbsR2WHEAwe-QuqBbZ7izFhRcl9is,1898
tokamax/_src/version.py,sha256=ZnWOhI-yALJAdn85gxaasnxO1xh870vCju_JYITqJiU,1213
tokamax/_src/autotuning/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/autotuning/api.py,sha256=M1awCQxTXeZV31IVzbC40rxWR0WF2Vctl6EfVkDYejg,16115
tokamax/_src/autotuning/api_test.py,sha256=pxAVlE2uQeBBm-RFHnsd-cv2oVVH9dI0JFM6k2SQowI,18431
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=keOe4imeYo9jUeF1kRrJ2xc5Bnm1GtZUdpG96tFSVG4,4124
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=BWoEh5p8TSe7udXwnqMeGqh344Bu792kssYhxFaQJ9o,7053
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=GlW8jcHf0qR4chPFMaUslvsOE1_TYh_xz-sItW0_vIk,33502
tokamax/_src/ops/attention/base_test.py,sha256=L9gAlnSReP0oyNL-lpQ7i9yrXcyfMoPV2NilwdCYjCo,5286
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=27QW8M-tzZ28awX-4YOWVrvI9h4698PCElgCCT0k6dM,8798
tokamax/_src/ops/attention/pallas_mosaic_gpu_common.py,sha256=xsMoN88SZqnJ2zZeSZWeABZR7uY0f3p0HrNy6xH_xnA,10526
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm100.py,sha256=xC-EvaBF7dpbgiGBK2s1PzxtoqGayCGlVaOmFYn1XeM,38477
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm90.py,sha256=E2tMgYOPy01oBw3MXGZ8k45MHHuQAlsRLPIvNefv55o,24344
tokamax/_src/ops/attention/pallas_mosaic_gpu_test.py,sha256=Cl343KQuoreot_vv7LmIt_Idd2i9r3lFhOb2aCWKF7I,8993
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=sWlSS2OD_XPj-aRK3sGKtWw8KkT4oh6a391OMhfmpn0,4124
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=QrRuigbk_86wSU8wL1zczgrwY4bR4t-CYroaH3HJVfs,39295
tokamax/_src/ops/attention/xla_chunked.py,sha256=4wgO3SZHNcVuh19thHzi2TOZnVEBg57XMvQeBaFAJqI,12963
tokamax/_src/ops/attention/xla_chunked_test.py,sha256=Fvgab-Trrk_dMGrDmaBYoSZFvXvoiK0VRDbEaEJ0W7c,8693
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=QzHfFgpc7hubL0ekpGccpAJvds1sbi641LoIkhGG2GU,15298
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=vmMmHQMhaWxGdigIJLKepS1rOLVQfy1pMABWYe5MRkE,7914
tokamax/_src/ops/causal_conv1d_gated_delta_rule/gdn_attention_test.py,sha256=c8DMD2eZcu3Gvij1PVktTubqZTjKAVi-2yhvVT5VIOI,46029
tokamax/_src/ops/causal_conv1d_gated_delta_rule/kda_attention_test.py,sha256=DgMRiFSi6lGuLMSeyGIR4O9tll_YJe6qctL0l0PM-QA,20210
tokamax/_src/ops/causal_conv1d_gated_delta_rule/memory_ref.py,sha256=gYIjYqk9njaywz_l5LuqdrULTse8Tu8I9qVnsRgIL9o,19565
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=rT5WNq3U8OiVT-C8DGNmQiyfHv4yxksPChDyxlcfY4c,4899
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=so5v4fDii09_-F-tFXzL4uQtUAo98Vk5JqtQyNZ8IVc,11075
tokamax/_src/ops/causal_conv1d_gated_delta_rule/vmem_ldst.py,sha256=NfDlkHch781kjufk_rXwK8EYEyeBiMJPsvVDwpekwNc,12235
tokamax/_src/ops/causal_conv1d_gated_delta_rule/wrapper.py,sha256=XSgRmaoymfZd0AOOVdDAIIpoTRmnD2GwzWlS_e2U5jg,26017
tokamax/_src/ops/experimental/batched_rpa/__init__.py,sha256=ZXRlLcMbY-OQGSiroo2YyieNsy7ImsY4E9g_R20dQds,696
tokamax/_src/ops/experimental/batched_rpa/api.py,sha256=kSV4gOC_zJBV8cIPjzfpT7Dyj4WksN2EAIRune-56ZQ,3983
tokamax/_src/ops/experimental/batched_rpa/arg_specs.py,sha256=sZej609WzXROH6pGQ2Z_Gn7CX0G4u5vlPlgiV3QkAw0,6653
tokamax/_src/ops/experimental/batched_rpa/base.py,sha256=0C7yk9k9R4DPmdlxHy4Za8IgWE5wdL86z6YHVikDTdo,7575
tokamax/_src/ops/experimental/batched_rpa/base_test.py,sha256=smawEWXSzBDusUsLePEMDk5BbOLm4LNaFgWMx0n7464,6684
tokamax/_src/ops/experimental/batched_rpa/pallas_mosaic_tpu.py,sha256=J8njdToSwzMl51ntWJmvKc5_HZqUEofJGilXJkqULJM,7566
tokamax/_src/ops/experimental/batched_rpa/pallas_mosaic_tpu_test.py,sha256=q0LCO3QmkvA1-7nd0G_hf4woUPDyVs4oEDgAQJfCZpo,4288
tokamax/_src/ops/experimental/batched_rpa/reference.py,sha256=huA3pRn-Px2E8edPHP9SLYV_T0V0CynqVe2j2Y9tR7w,8029
tokamax/_src/ops/experimental/batched_rpa/kernel/__init__.py,sha256=YKJW4pOeOddU5njid32X3hV6RxrTb93syWZgZv4FHEc,950
tokamax/_src/ops/experimental/batched_rpa/kernel/bref_override.py,sha256=0X006V4W57EbDZC5xV4yoaXCBAgvV9LrvBn9Hh_7Vdc,21477
tokamax/_src/ops/experimental/batched_rpa/kernel/configs.py,sha256=8F6aRrdeWQJkKzM1jBYvjdK-8xLZlt_gk_bqK6WTeIY,20211
tokamax/_src/ops/experimental/batched_rpa/kernel/configs_test.py,sha256=rSUCvq5e7UjWXtLPpHQ0BO9ycdTwSh2NLb1MdjxCT-0,3825
tokamax/_src/ops/experimental/batched_rpa/kernel/correctness_test_per_token_quantization.py,sha256=gijSXLxcnD_oGT4iQ4K2H567DonFVVTnuvUJngcC8g0,43623
tokamax/_src/ops/experimental/batched_rpa/kernel/flash_attention.py,sha256=rG483zrZGoJA-xhbUadUblTh-S3R_gl2acmYAnjHuiE,5680
tokamax/_src/ops/experimental/batched_rpa/kernel/kernel.py,sha256=9tDVYbV7aUtVdnXByjmu1i3-Mu50ZlBttZuK6VkfCzY,34974
tokamax/_src/ops/experimental/batched_rpa/kernel/schedule.py,sha256=VeQNmMG3BSHIjsRL5N4UooALZemskdpfgj9PhEAlmUM,32675
tokamax/_src/ops/experimental/batched_rpa/kernel/schedule_cp.py,sha256=0S1jxkutB2Iv-yEqbpW5VJLWW_HS_BhysuWFUTnw4KQ,10411
tokamax/_src/ops/experimental/batched_rpa/kernel/seq_along_lane_parity_test.py,sha256=F7cgIN71Y_9_S3qCjlaViq4r-Na3wm6Gt8eFVWo8N3E,7302
tokamax/_src/ops/experimental/batched_rpa/kernel/speculative_decode_test.py,sha256=82g_avGR2wbebaKecCuG0nhXGa0qdbXFytUZ-evDQMQ,6884
tokamax/_src/ops/experimental/batched_rpa/kernel/stacked_rpa_kv_writeback_test.py,sha256=J9huP7ASFbMWcVZW8y68qdF1tHz95_0OTIPDr8StxKg,8370
tokamax/_src/ops/experimental/batched_rpa/kernel/stitch_utils.py,sha256=IAU_BXGvasB4f4ayOlL75MGzkEOnshbPgDbGAxCApKQ,7855
tokamax/_src/ops/experimental/batched_rpa/kernel/utils.py,sha256=M3fSX3Z1ze2PPvOtpk6Ig3exVovg-eMdislEUWKf4j4,5287
tokamax/_src/ops/experimental/batched_rpa/kernel/wrapper.py,sha256=3EAaByuiEV8aZmWhbeQwwtHs6k0JaBQkQENUdaxp8c0,33009
tokamax/_src/ops/experimental/fused_fp8_matmul/__init__.py,sha256=TGvQ6johv4CESH1L9Syn6UwXDTtp3DTXDLAfvvb6pU0,696
tokamax/_src/ops/experimental/fused_fp8_matmul/base.py,sha256=FSN0Lhi4d_Wc08o29sCOuao2Kk-5rV_o9KTYzUW23_U,9148
tokamax/_src/ops/experimental/fused_fp8_matmul/base_test.py,sha256=T0M22yFqBcb3eptWbFoEioYXgURqDKxNlurfmqCMZv0,2946
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu.py,sha256=xHcT7t-takau__olwHby2cUw0yBkCYCE9Ps3imq6-i4,4679
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_kernel.py,sha256=LF7v9Q0qxOxD4PZry1Dr4T6mrGXBs0SBgcTqnM5DzL8,18039
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_test.py,sha256=o82G749DkV8mhYfx93lfJsZfvasTEnq-MJJsULm4f_0,5148
tokamax/_src/ops/experimental/fused_fp8_matmul/test_base.py,sha256=L2FDMklWUYzHSw7D9reevcUydlGD3vno0x0c1ne8SJg,9837
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=rJb4uvfYb1tPmj5tDtwDKVJKxpYWKYxmGlhmOQugynA,107021
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/gmm_v2/__init__.py,sha256=OtXRXu-czwR1bk_ECYH3uJ5bcOomEIkjsyLFNJ4QZQo,655
tokamax/_src/ops/experimental/gmm_v2/gmm_v2.py,sha256=b7D7y-PX_67ifkI2_ofbZrEM5zZSe4jztmzCOonoM6M,61798
tokamax/_src/ops/experimental/gmm_v2/gmm_v2_perf_test.py,sha256=qRuU0Ot2P5PWSeY7GLEOpw5WQC9VO-4u1HJvCsPOw3g,10821
tokamax/_src/ops/experimental/gmm_v2/gmm_v2_test.py,sha256=Z3YbkhCnQ6KsQ4gzG02Kn4WOKe5WNKHp3XXfYh6tWZs,64211
tokamax/_src/ops/experimental/gmm_v2/tgmm_v2.py,sha256=RIyP3Hk80Zp_3lVQ3_Q132LJXGRTgfI6bk4Ux2RobW4,29466
tokamax/_src/ops/experimental/gmm_v2/util.py,sha256=A1tpYOIhCjHvupDTCDlnffAmjrZ5UNQ8rNZ1MNaX2J8,1899
tokamax/_src/ops/experimental/kda/__init__.py,sha256=0besMHT5_wyFfkg9TCCtoJDffeMgxOR1P720KBe5QoU,878
tokamax/_src/ops/experimental/kda/api.py,sha256=wH-P4jGuyloUwng5yFvC15tSpAQyjtHUisr0RGvvjsU,7067
tokamax/_src/ops/experimental/kda/arg_specs.py,sha256=IkAioLlUEje9BEyM8oFYnJbenHZCFH3cCQSWkvAK19A,3427
tokamax/_src/ops/experimental/kda/base.py,sha256=7CtZ4GzqUg0NaJ4lKexx_mrY-W5vvt3yFx4EkjN4v2k,7323
tokamax/_src/ops/experimental/kda/base_test.py,sha256=Qj9F0JrpxCDTiiH7feVFCN9cFVzV9iVg4ZC772178PA,10496
tokamax/_src/ops/experimental/kda/common.py,sha256=ul2jJnK_z_LfCcYtht997MphEE7AHsWce-jNFzIFKus,28170
tokamax/_src/ops/experimental/kda/cp_utils.py,sha256=7s4u7U7v7XK_lK2sxfIEEhnW67U0u2BGRkOC5LOipBw,17448
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu.py,sha256=kbm3IfuHTfq-xsuvHu0bCNb1lI8qsr5OcTeBE521sxs,18553
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu_bwd_kernel.py,sha256=L11Ua12i-OMQTEILpm7wXyTxS6akA8zNbAz9dfeR9m0,60754
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu_fwd_kernel.py,sha256=CkfUV4VDvPQNp6I3TJcRkNaowc9QiJ737aX9fUKBWQE,57432
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu_kernel.py,sha256=SgJANWlcTNDe2Nfz9TtIzuRXeDXan6vtabdYn0nIEMA,1026
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu_kernel_test.py,sha256=LvinqfRSrqtqdlz-ocX4-c53ILnCDyLobnN_NExC484,63821
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu_test.py,sha256=QgK0xNVQcHoylUmxMWJPDIWsYCRJQTaIGdadb7esF24,9582
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu_types.py,sha256=OjfVHD_YBUxHMK6kqKi9cxJpJ8xTYSvQQR1_Ww82eaA,2256
tokamax/_src/ops/experimental/kda/reference.py,sha256=Q_KfVk25kqfYaWiD6qIwJfgzgzW4zkkuACsOPml-lik,6838
tokamax/_src/ops/experimental/kda/utils.py,sha256=fuoov2n0z86kOnUdRGkK7qtPHhlEQSD8MzoNxb6dijc,16543
tokamax/_src/ops/experimental/kda/xla_chunked.py,sha256=-orks3WcZ9KAJgBpTL7bKOBaRww_jcgJtoV0aS9NTzk,8504
tokamax/_src/ops/experimental/kda/xla_chunked_bwd_kernel.py,sha256=cXe2IHgvFd3dXkv6aygc9tBucwpMxbwr0gK92RrLy0U,2425
tokamax/_src/ops/experimental/kda/xla_chunked_fwd_kernel.py,sha256=GJPDCW6xfVTWf3gSe9hJ9dM4u9rV7g1T2s6MPDIFdhM,16351
tokamax/_src/ops/experimental/kda/xla_chunked_test.py,sha256=XTZTHdEVi96UfecFNglFvCROswBIencACIWqR10HC8k,8068
tokamax/_src/ops/experimental/lightning_indexer/__init__.py,sha256=Zx5fOsw9LvX-EK0a3SIACWiHqds0wZ4N-i0B8UxDxyk,689
tokamax/_src/ops/experimental/lightning_indexer/api.py,sha256=CUyCKaEbW3a2_5lh6CxPR7gAgHWp7BAsN7pSlzz1d7A,4911
tokamax/_src/ops/experimental/lightning_indexer/api_test.py,sha256=2iPgwqHlE6RCTro836-FHV9KH30hbRCDUiYvdl4Wv0g,3385
tokamax/_src/ops/experimental/lightning_indexer/base.py,sha256=w2JdTPO8XxiVrpCjl-YUOWGvcaj7_b_vAUC76UtzIqc,5016
tokamax/_src/ops/experimental/lightning_indexer/base_test.py,sha256=QYX5-LicgXDJCzaqhBIzWhnCcXwH4QrxD9YNraGdR7g,1920
tokamax/_src/ops/experimental/lightning_indexer/pallas_mosaic_tpu.py,sha256=l4CNsFvBntebc2dGLYyVotvOY_Rfi5J81PZTsBBThjo,4965
tokamax/_src/ops/experimental/lightning_indexer/pallas_mosaic_tpu_test.py,sha256=n-VktqqMM2ViN0Fl50OZYtx8bdMCxFCzP5tnj-mXRF0,2531
tokamax/_src/ops/experimental/lightning_indexer/reference.py,sha256=aJVAkabNtU1E40Fm01UHJcOtl_qw5QHLUs9hTxSzPcA,6917
tokamax/_src/ops/experimental/lightning_indexer/test_base.py,sha256=HXGfLHyPaIUmZ3J6ORrPKBoBx2gqFuqZ5f8Mp9P8mYk,7455
tokamax/_src/ops/experimental/lightning_indexer/kernel/__init__.py,sha256=ipixnHwBtMoXGa9jiYV5WmQfKpZrfA4tiI3GN9DCzdQ,705
tokamax/_src/ops/experimental/lightning_indexer/kernel/bref_override.py,sha256=zuwpKMITjp3NTx39LkRDsHqZHvhO4fLBC3zZov8-nhQ,19845
tokamax/_src/ops/experimental/lightning_indexer/kernel/config.py,sha256=8qoDaHqs0MI2L8nU0nRh1aAzQwhGG1gFqTasPWE7Eyc,5321
tokamax/_src/ops/experimental/lightning_indexer/kernel/dcp_sc_compact.py,sha256=Gk5EWt4C5OnGIIZOPK1xgtGrnjiHhuOaUMTxjvAY0C4,5575
tokamax/_src/ops/experimental/lightning_indexer/kernel/metadata.py,sha256=KfaVgVaWHPctbI74yLX2_tg-EXS2w1GSC25v2a8ksLo,13720
tokamax/_src/ops/experimental/lightning_indexer/kernel/metadata_test.py,sha256=VXVU5aZbD6g00ucgxJx8ofeZAEzxQgFJ-tqkY7CvGSw,18173
tokamax/_src/ops/experimental/lightning_indexer/kernel/streamindex_topk.py,sha256=pbtA6vhTWJEPy1cwD4PC-rTL-iIaG3Y2t6-zHn8y5Tw,50948
tokamax/_src/ops/experimental/lightning_indexer/kernel/streamindex_topk_dcp_test.py,sha256=CpB0kRb9gmxM2IGXeFBkaYsx1CrYcry3m_vHoH4NVsM,12709
tokamax/_src/ops/experimental/lightning_indexer/kernel/streamindex_topk_test.py,sha256=TEsFbyAc3GTEWcYL2bOBVNuG_tm18wQjNH1kR91t9ds,53129
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/tpu/compress_store/__init__.py,sha256=y_CpGGJ-P_GduSsHNwM6P3svzalHDduN9n3XqKhpCKY,698
tokamax/_src/ops/experimental/tpu/compress_store/api.py,sha256=Lkf-vg9laLG9_5wsyvKZ6466Rz5T_oPqJo8d2lYuTvo,5705
tokamax/_src/ops/experimental/tpu/compress_store/api_test.py,sha256=dsEu-1LOWQu8vXXx9qplQOpyPtS9FVrRBpQLTCPTa50,2588
tokamax/_src/ops/experimental/tpu/compress_store/arg_specs.py,sha256=0izOHLqBTrWQkVpqrcvRcrx74bOwdTJKEY74bHgClLE,8909
tokamax/_src/ops/experimental/tpu/compress_store/base.py,sha256=av3oOoHBMkTK5sZE651BvchtmHOYJc2dn3ivskTtgu4,10265
tokamax/_src/ops/experimental/tpu/compress_store/base_test.py,sha256=oCk8cKpYareDhlnjz21u38yobumcH0xhDw58z7vZ3Ig,2761
tokamax/_src/ops/experimental/tpu/compress_store/buffered_ref.py,sha256=zmk7sfQTwQ_Ws8pO4On8boJgVWe670Z8U9Oaq-g5MJU,17098
tokamax/_src/ops/experimental/tpu/compress_store/compress_store_test.py,sha256=HTMRLx0Mge1jVmdX7b42H2AWs7qnma4G3udGdQd2SAQ,20287
tokamax/_src/ops/experimental/tpu/compress_store/compute.py,sha256=OieYX3MuOOHpwL8WFucjO8_LyngPqya3CPZzGKpxy1g,15189
tokamax/_src/ops/experimental/tpu/compress_store/config.py,sha256=N5Ai3ZmaE-Fv3pdCQLX7hI7Wvtfju73JdO7zHFc9vpE,17411
tokamax/_src/ops/experimental/tpu/compress_store/csa_cache_layout.py,sha256=txVjf-kX7OUttuDDBJOsa5xX8iZFFpRvdvGjcNLlEW0,4832
tokamax/_src/ops/experimental/tpu/compress_store/pallas_mosaic_tpu.py,sha256=r6jObSiJxJaveWbjolkbiU9lmGR8vpnR1PGB2Z7Sxvk,6319
tokamax/_src/ops/experimental/tpu/compress_store/pallas_mosaic_tpu_kernel.py,sha256=3a-w3PoedUeZX3pFnzNG9M_K3_48_lCs72dK1QUGYYs,17633
tokamax/_src/ops/experimental/tpu/compress_store/pallas_mosaic_tpu_kernel_test.py,sha256=fHJKmMPLyHtCWlDEQpoKYjpcJSK1Wz2MvjfRUFb6lE8,21414
tokamax/_src/ops/experimental/tpu/compress_store/pallas_mosaic_tpu_test.py,sha256=A_OB4HgSw4p0MIQCTE9XezWKexPT0yFQGUWWunYgUPQ,3999
tokamax/_src/ops/experimental/tpu/compress_store/reference.py,sha256=cXdXh5Q0eqkh2Y6FsH9Wul97ods4nAvXxUIbGC9tbCs,22162
tokamax/_src/ops/experimental/tpu/compress_store/test_base.py,sha256=q4gBEnoR8dg7YuDUluT0zJ0LgN-YQ8R-aS5KcEQDZqY,18463
tokamax/_src/ops/experimental/tpu/csa_gather/__init__.py,sha256=Fyu1RjA1-ZGi6EbDO92VOs7ahZ6HNBDlSh6cgcJEteY,690
tokamax/_src/ops/experimental/tpu/csa_gather/api.py,sha256=pafenezvoFBHPjI16UwOhWI2HirILrZJ08G6X6e4Wm0,3865
tokamax/_src/ops/experimental/tpu/csa_gather/api_test.py,sha256=RGGXufzxwF3pH0P4_9Hrbn7DALU_fevhkOQ_7bt3NRQ,3411
tokamax/_src/ops/experimental/tpu/csa_gather/arg_specs.py,sha256=Trg5zXJdbFEcv4f-nptUScfvQhsEDQPPt174S1WAtcI,2141
tokamax/_src/ops/experimental/tpu/csa_gather/base.py,sha256=9xBsH9q32Odr6_DOFM30a1WM28mj8c-pOwpZTfex7EA,3174
tokamax/_src/ops/experimental/tpu/csa_gather/base_test.py,sha256=DMC1AtAzDeD1Tsw2MshthW3UpPMozEPYm3Qwulu01yY,3949
tokamax/_src/ops/experimental/tpu/csa_gather/pallas_mosaic_tpu.py,sha256=O8969OySdj3L7S32dlWNOtMehURK-FA2pYOg4HIqbug,5670
tokamax/_src/ops/experimental/tpu/csa_gather/pallas_mosaic_tpu_kernel.py,sha256=WWDDeqGkIpZD43gurGM0jlE_UFRAnscQuC9Q6J8vMUY,12985
tokamax/_src/ops/experimental/tpu/csa_gather/pallas_mosaic_tpu_test.py,sha256=C-BHR52qW5Vw3kKe3tLJBylE4nBII0VE5yhSnKdEi_s,4987
tokamax/_src/ops/experimental/tpu/csa_gather/reference.py,sha256=DXGfwmSfL3j-8LZhyGAT1zRk6gHzy2CiujA7x4cGtlo,3581
tokamax/_src/ops/experimental/tpu/csa_gather/test_base.py,sha256=flwl3A4kFEWwge29B6f6XrdUH6Mz51LPD3ErAMjsNuI,3605
tokamax/_src/ops/experimental/tpu/mhc/__init__.py,sha256=xpTpj-P_eNfir6pCWMzDwzPUzd7Iw7l7RJ2586sGcR4,736
tokamax/_src/ops/experimental/tpu/mhc/api.py,sha256=Wlj6gnlUP2NJNesDM10m6CLr412apUdVF0RJu-FvxpQ,9747
tokamax/_src/ops/experimental/tpu/mhc/api_test.py,sha256=leiYnIQvyztXvnZOj6NvYgoJd0WfaQlSrU6NjhFsxQI,4585
tokamax/_src/ops/experimental/tpu/mhc/arg_specs.py,sha256=RedvXy8bSrY2RfFiPnhMkzmR2O5LHF8RgecchW1zvXk,2855
tokamax/_src/ops/experimental/tpu/mhc/base.py,sha256=C9EvSs1q0BeFgQxmg47ZFXN1ZOCrBNbxajj9BdmxcjY,9372
tokamax/_src/ops/experimental/tpu/mhc/base_test.py,sha256=aPnFETnCkvqLs0H_VGT3fav0D-jIFfYk3-yCQl9zZI0,7175
tokamax/_src/ops/experimental/tpu/mhc/fused_post_pre_kernel.py,sha256=vMZWdLUDdj-EWwdaD29QQFOYhcupdnT7hah-DWCL8TA,11064
tokamax/_src/ops/experimental/tpu/mhc/mhc_test.py,sha256=QickTnta1AIgV4e0nPnc5yJ_OpQtJIUwJyPKXRUZ2as,19714
tokamax/_src/ops/experimental/tpu/mhc/pallas_mosaic_tpu.py,sha256=XN-ga932uWnDIHRrlszwi3CRpBJ3bQCDXOPfqW4iLeE,8272
tokamax/_src/ops/experimental/tpu/mhc/pallas_mosaic_tpu_kernel_test.py,sha256=0ZkqmijZ8drw-3RJ4P63i2ZXC9eCEapUcks8ekEmrr0,5682
tokamax/_src/ops/experimental/tpu/mhc/pallas_mosaic_tpu_test.py,sha256=RdzhlgnzugcBtcgS_k2zg6Ksbo0pdDYFDEjLcV1KpXw,4102
tokamax/_src/ops/experimental/tpu/mhc/post_kernel.py,sha256=etHaj84oRotg53zPnVL2OseEQfRnpWhUe_fN2jvZXoM,5915
tokamax/_src/ops/experimental/tpu/mhc/pre_kernel.py,sha256=d4sWqmvnUGHjCqg4MH6TWkeonuYxnLSYFQB1ump6K9Q,11939
tokamax/_src/ops/experimental/tpu/mhc/reference.py,sha256=Zkd72suUXdnGo-sORUU4X66oHF60f7DrIi7Kgvh6nLs,8442
tokamax/_src/ops/experimental/tpu/mhc/test_base.py,sha256=-fV2-SVayYOb052BybesRO8PuFQ6sSx6T_nGyoy8dw0,9094
tokamax/_src/ops/experimental/tpu/mhc/utils.py,sha256=4MDYnB3j23bHp2G6ouqAa9giJbQXvequx19Nh5BdgaY,5623
tokamax/_src/ops/experimental/tpu/o_projection/__init__.py,sha256=P0isQ1Ev-qzsu1ahtJ6vzc_8biYHZolgNCffD8IGPHA,735
tokamax/_src/ops/experimental/tpu/o_projection/api.py,sha256=TjwHrpqShpCX2lF7u__n7hsZMwXB2n7xxI9wtzjgVDI,4444
tokamax/_src/ops/experimental/tpu/o_projection/api_test.py,sha256=ovuq_U5PeHFCCNd-mqp1LtsyE6ETt_jcjAtpPBxHQUE,2991
tokamax/_src/ops/experimental/tpu/o_projection/arg_specs.py,sha256=T23iR58pKbxAW9P4ybX0vn45ImBcffyqTb0ZZEYzWGI,2750
tokamax/_src/ops/experimental/tpu/o_projection/base.py,sha256=xykg0LG2wPflGCoURbpjzhJFyPI0Mvf95ZzhZ35FDz0,6801
tokamax/_src/ops/experimental/tpu/o_projection/base_test.py,sha256=o2MYvgkTVs94MCVD2wihjeBtDnEfW_FeUoKvu9O2pvc,4022
tokamax/_src/ops/experimental/tpu/o_projection/o_projection_test.py,sha256=0GY-jEIJeO9jK32wx1ovII0aioUfLnrDpoAkagTAfWU,6670
tokamax/_src/ops/experimental/tpu/o_projection/pallas_mosaic_tpu.py,sha256=OI5VcAaiXSPtoJjD3TOv79qRsu3SrtQJIXh53qGrUMg,8945
tokamax/_src/ops/experimental/tpu/o_projection/pallas_mosaic_tpu_kernel.py,sha256=_BTFc5CzZcANHFFkYOiSfqnO4bmXvApXCX5Ae0angQc,11419
tokamax/_src/ops/experimental/tpu/o_projection/pallas_mosaic_tpu_kernel_test.py,sha256=YUv8Rt0X7akUyfW62XkJF8R8bO_qrKtaYUZwT_NC7C4,9371
tokamax/_src/ops/experimental/tpu/o_projection/pallas_mosaic_tpu_test.py,sha256=j7gm-44_OeeGbygh2Zx2slLylN9Xy4l5KvevGRXSpEM,4820
tokamax/_src/ops/experimental/tpu/o_projection/reference.py,sha256=ak7iYX6a-5emI6RX_xyJoeuILujgQnjDECBdWz20J_8,4105
tokamax/_src/ops/experimental/tpu/o_projection/test_base.py,sha256=87gbDbOnbtDNv8FyZv4iq4HaS9CHgdyD2RcE3Cxeulc,6395
tokamax/_src/ops/experimental/tpu/proj_and_save_state/__init__.py,sha256=EyGX1VT2fUTgHUF2DrOYOTHZh_jU5Z4W-32TKxlAJx8,735
tokamax/_src/ops/experimental/tpu/proj_and_save_state/api.py,sha256=c_WSHrgVkXFyX22CoaiaqRKbYnIzNSisZNlh9s0PDJ4,4189
tokamax/_src/ops/experimental/tpu/proj_and_save_state/api_test.py,sha256=eDi7PE8FGL4aDiDQ2p6umJes9XqzxdqxpBM67C-4xN8,2796
tokamax/_src/ops/experimental/tpu/proj_and_save_state/arg_specs.py,sha256=rl92EvdQuY68y1sNFkq8FCC_DuESsUSPb4UKADDMpb0,5097
tokamax/_src/ops/experimental/tpu/proj_and_save_state/base.py,sha256=Cg83aj2w1kFSj2oLF0nR-0vJcdRcwHQXe8m3Iq7lveQ,5417
tokamax/_src/ops/experimental/tpu/proj_and_save_state/base_test.py,sha256=-rPPfiRLU8CkYUVugoYcIFvFXQmMVlJVX1TnoJcgyUw,3930
tokamax/_src/ops/experimental/tpu/proj_and_save_state/pallas_mosaic_tpu.py,sha256=GYmmmOarXo7vTvdBmkqVyFv-SNvQTlOyLTzTKzOnd3E,7720
tokamax/_src/ops/experimental/tpu/proj_and_save_state/pallas_mosaic_tpu_kernel.py,sha256=-D_D9ZuMqk5GV6b3nBQhpsdTrQ4Wur6iR0jHpPtAv4E,15666
tokamax/_src/ops/experimental/tpu/proj_and_save_state/pallas_mosaic_tpu_kernel_test.py,sha256=gqqQJDLhR3h03sl9hJ2EGLBucNOvjHomm9qsf9QOgts,13057
tokamax/_src/ops/experimental/tpu/proj_and_save_state/pallas_mosaic_tpu_test.py,sha256=zEaZGm_gVt4PaCte_YvdA6CQvv57VRNWCcO7__Aj2mU,4412
tokamax/_src/ops/experimental/tpu/proj_and_save_state/reference.py,sha256=EUmM5mhilmHGlfqAD0K5x9jXe2bko1R42RBuEZnpBpg,6486
tokamax/_src/ops/experimental/tpu/proj_and_save_state/test_base.py,sha256=PcleTy74dAePFBSVCJVGqo3Ww51AmVSX4LTE6cJOYVY,9645
tokamax/_src/ops/experimental/tpu/rope/__init__.py,sha256=OW12E6mnjE0S4irbGbdaMcRo0_Ip9SCR_b5-JkHyE_E,696
tokamax/_src/ops/experimental/tpu/rope/api.py,sha256=V5PUj7zgrW02Syyw3UvUC3v9y-RVQQTdHkk5Vg_ft_g,4974
tokamax/_src/ops/experimental/tpu/rope/api_test.py,sha256=P3pycwwOjwKCIJo588mYVRqtSGithLgzarOA6GG_Qg4,4861
tokamax/_src/ops/experimental/tpu/rope/arg_specs.py,sha256=PIRsxhgeDq8eS0UW8CtnIR6LUuGYx11zYBOJat6EhL0,2568
tokamax/_src/ops/experimental/tpu/rope/base.py,sha256=jfG7hoeWV5mXdOlZi6V1A9SzFcCKgDfsowZvyBBNXRE,7455
tokamax/_src/ops/experimental/tpu/rope/base_test.py,sha256=6oRtgA7kWfODOPuSze2gt1mTGOfxLAfmhbkFynwmCko,3817
tokamax/_src/ops/experimental/tpu/rope/pallas_mosaic_tpu.py,sha256=34dOwq5qZoYqkcz2MrHTy_jvOzX5N5fmq1_HyAit0qQ,8063
tokamax/_src/ops/experimental/tpu/rope/pallas_mosaic_tpu_kernel.py,sha256=N0EjCYBC_LevQ-4KX34qXUiV0KgrKrY_tjRUQ7oqkAY,20499
tokamax/_src/ops/experimental/tpu/rope/pallas_mosaic_tpu_kernel_test.py,sha256=oemow4PdKldkqs_DxMZHn5PwsIkxapCrxYSF7Hb_NIQ,10319
tokamax/_src/ops/experimental/tpu/rope/pallas_mosaic_tpu_test.py,sha256=aUSuGpyWsvV8L0rXVdjEsYFieQ1Pbe7R_gvhC0K8PQ8,5873
tokamax/_src/ops/experimental/tpu/rope/reference.py,sha256=dgnvIr-8eTaoDiHC89fk2wicCzgNr6IpZGFi9J2_Y0E,5408
tokamax/_src/ops/experimental/tpu/rope/rope_test.py,sha256=wlu61rkDKfEPfTTCcCloJ9hmuFI7L9seLTGrzCqjAug,9277
tokamax/_src/ops/experimental/tpu/rope/test_base.py,sha256=vIbidQkvPSyYAOVO2MwYPE8dQ3rrpFZy4Af0QYS3XFY,6277
tokamax/_src/ops/experimental/tpu/router_topk/__init__.py,sha256=N4NCVxareHoVJ_e4t5ohe_0niltDrq2564ujDjMbASY,706
tokamax/_src/ops/experimental/tpu/router_topk/api.py,sha256=SYQJuphsjLJXvJ9qZyFhmT0AlMDAMNqTCr3PEDzuaw0,3754
tokamax/_src/ops/experimental/tpu/router_topk/api_test.py,sha256=7ZPt6wg89XCpVHf5Yiw9NCyjwUyed3zLlr7BeD2EpHk,3391
tokamax/_src/ops/experimental/tpu/router_topk/base.py,sha256=Hesq8Kunfwhjyw0lUPrQ65-_YNtzjF2W154ABJn2PSI,4252
tokamax/_src/ops/experimental/tpu/router_topk/base_test.py,sha256=IPU8NMc8QRk7aMpvKy68APe6Tt8ImB3YMb7npkiPj08,2572
tokamax/_src/ops/experimental/tpu/router_topk/pallas_mosaic_tpu.py,sha256=UmwTurKOg5B1WyM3iTrSrFVqI90xuT1rwloDD-KB4V4,6275
tokamax/_src/ops/experimental/tpu/router_topk/pallas_mosaic_tpu_kernel.py,sha256=v3MRJ86caxh4jZdxtbK_GWmJAXlBcIn2kaJZ0xDnKkA,4399
tokamax/_src/ops/experimental/tpu/router_topk/pallas_mosaic_tpu_test.py,sha256=1gg71W2juydoQeMJV1B9pwqsxjLF5C1sdzOGZSFDda0,7963
tokamax/_src/ops/experimental/tpu/router_topk/reference.py,sha256=4_SHnCV3v2E8oXVTHYyC2weljdHoYx9Z1rC8Lykt_aU,2811
tokamax/_src/ops/experimental/tpu/router_topk/test_base.py,sha256=dJzO4lbOnGchh8rx-NczJPf1N6wl35hjCDpGlXIq1FM,8347
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=borNIGFMyy4-o4_TwM2JF5J2gFwI9O5DqfrR_awHdiY,4618
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=baVrNOzdRwxyVnoPqE_9oDLqVWeD95LIMSWQExG4gXQ,3134
tokamax/_src/ops/experimental/tpu/splash_attention/base.py,sha256=4e3SLO99xs1hPhIdlR0pX7xa7tULKoA3thTFAeSesto,9957
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=xfjlEmmA_Tltooh63eFSOR6wYoBp3M7VU1h3vYaEmWI,9191
tokamax/_src/ops/experimental/tpu/splash_attention/pallas_mosaic_tpu_test.py,sha256=kHheNK0Z2TzEFdViDZ3SogKO4kafgnngoLtU-Dr4MV4,8876
tokamax/_src/ops/experimental/tpu/splash_attention/pallas_mosaic_tpu_vjp.py,sha256=cVDN8mdmm2St6bVaiS7-np1Qo9VSX_lLoGIZ32MXRgQ,7688
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=tLtsRpQ-x2jrokovHx4zTDTiLBITQYgw2-4BfdtDa04,101909
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=CL6Qk9vPGYdcGNQecM0cMt9YDkoIEj6PoLFflQoPaRI,52424
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/experimental/tpu/topk/__init__.py,sha256=T249jb1MwC6WFuwr1-Evu-bDB4OG_3NmhYqqkvhpbwA,676
tokamax/_src/ops/experimental/tpu/topk/api.py,sha256=956_3444UnWQrINv2jhwz5XZGf2EYWlPRowRRS6FlIM,3411
tokamax/_src/ops/experimental/tpu/topk/api_test.py,sha256=6jTSTLMVBEkZnKficfMlWRm0Ysfg4o2kIe3PGRy6N6U,2335
tokamax/_src/ops/experimental/tpu/topk/base.py,sha256=5xhIvX-U63309ERhRkKDfHOKCBULFtZEk8uPshkyMSU,2666
tokamax/_src/ops/experimental/tpu/topk/base_test.py,sha256=oxm3II8f0HlqSf0Bu2RjJCwOz0BED-PHGrqPNS29bt0,1907
tokamax/_src/ops/experimental/tpu/topk/pallas_mosaic_tpu.py,sha256=jHqEBfSmoQVeqc9HoJC9mzOFMQU_sTtZugdZsBpmJRY,2716
tokamax/_src/ops/experimental/tpu/topk/pallas_mosaic_tpu_kernel.py,sha256=7nlal7nL4jirAZeULtPDlAmYQ_PfbnZijEGOOv-d5XY,19317
tokamax/_src/ops/experimental/tpu/topk/pallas_mosaic_tpu_kernel_test.py,sha256=iOnxMyKNNdcIoUGJRSkCJQDAXi9d4EYUWULGeP9B-Us,14317
tokamax/_src/ops/experimental/tpu/topk/pallas_mosaic_tpu_test.py,sha256=bMd7PjLDGM1zdvPEMn2jFcBrlpFgp0InnMUyz1e1UUI,1429
tokamax/_src/ops/experimental/tpu/topk/reference.py,sha256=2FMQc62K5Lxk_6jqwA4GBEJwIN3Ni6z-i2H1a4dRJz4,2797
tokamax/_src/ops/experimental/tpu/topk/test_base.py,sha256=qJM-8GGswT8eWo6p8hRbztaFsBQfNNHRpMR1wWfrQB4,5937
tokamax/_src/ops/flex_attention/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/flex_attention/api.py,sha256=oHt0SMiuiziGEtOTJ9Q3Ex1unHjg5bRr7T3ucANwE-4,892
tokamax/_src/ops/flex_attention/arg_specs.py,sha256=DECEEmyM-WzMoD7z8nd48JBDKUK9zyhY2InIT76AVaU,820
tokamax/_src/ops/flex_attention/base.py,sha256=noxY4QSQ1o3hg4dTB-YH6FF9j2HgTD6Cmu6dHO9EUQI,12340
tokamax/_src/ops/flex_attention/base_test.py,sha256=l4y8geGYw4AIJ_HnNOfXIyHXw2LKVc1xYEBlwNhIws0,1475
tokamax/_src/ops/flex_attention/test_base.py,sha256=RmAVNi29wphSWCX84Miv6sdnAtlJeNYTwpUsBn_catY,2865
tokamax/_src/ops/flex_attention/wrapper.py,sha256=RBC2bJ9edmLb-z4fVFJ-2J6B6aVm-FmT37QYk7jgYIY,3538
tokamax/_src/ops/flex_attention/wrapper_test_base.py,sha256=yMK2Hm8M2QyCSkf2x2iFRlePbb5WQxR3bZhV8M9fWG0,1335
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/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=vQT_5NC4ixu3Mx5JwMIdMTUW4CwMMj5AjXqgJ9kQW-8,5456
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=fTDtoyImXaKdMqnES6DpshLg0RUtg487mRRAE7_DXsY,12112
tokamax/_src/ops/linear_softmax_cross_entropy_loss/chunked_xla_test.py,sha256=sbr4ihHYEBr8WUvO93I_oD3B_PoGthON5A_zNDFjjh0,5420
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu.py,sha256=lBJHJKc1XW_bzoo5VSp3Ft-5XjaRTcMmrxMncjZisIw,7752
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel.py,sha256=wNKG9vQZ5UJeb6GRVVtlFBLae-dIprzvi9pTcJ0_xo8,56916
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel_test.py,sha256=9M607clmE9f4rmZGF4JTcRQowSk6BNJc4uRflbWJzss,40142
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=2oz6idp1kvOMnJD14BtaYfxLrKrLqi2LG68n9SKA0H8,2961
tokamax/_src/ops/linear_softmax_cross_entropy_loss/reference_test.py,sha256=EnZlEG_sQcA0FO1Ot3P0FKws9dUoshOBru5G5fHoqgY,4453
tokamax/_src/ops/linear_softmax_cross_entropy_loss/test_utils.py,sha256=0gjGXVuLWy7dEePckg4zNTbjwmYrnQCNz2KQao3JUpA,1963
tokamax/_src/ops/normalization/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/normalization/api.py,sha256=TUe2_ylDC1LGlkBt26bMmV5XJT3xXac2g0X6jf1mbC0,4536
tokamax/_src/ops/normalization/api_test.py,sha256=9cBM1R41w5wTm4ajZWkgug_SVJh5AwrgLYVsPtuH8xM,5454
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=ViSrGvFc946n1TWLPUqw6Hg4LeBJpOaAchQvkzO6n4w,2328
tokamax/_src/ops/normalization/pallas_triton.py,sha256=62BISUSYLM1b67DZK4EJlPjOYTF8M0JoKeF8sO8g8-Q,7940
tokamax/_src/ops/normalization/pallas_triton_config.py,sha256=1n-XBomBeEwF8NPogVO--YPWbZ4L_Wq4BzUgB_6R11M,4164
tokamax/_src/ops/normalization/pallas_triton_test.py,sha256=d_mBlKftcWWv_lJv6fn4QUf-KwKOHzOacryL-Ybal90,4853
tokamax/_src/ops/normalization/pallas_triton_vjp.py,sha256=GKlF9mEclG4Go4ikfvX2__fiY9F5ClRWSgPkTfZtJtY,7210
tokamax/_src/ops/normalization/pallas_triton_vjp_config.py,sha256=bRA3urHYYZwIyB-1o5-JQa9yxXBuNTHHLug4j_--H7E,3078
tokamax/_src/ops/normalization/test_base.py,sha256=ODAR0SG-83FHYDIlGxR54ZuJkEQ-1T4hqOH7NDocdqQ,6684
tokamax/_src/ops/normalization/triton.py,sha256=IygDouAK-JrYxWVAchW9wBYsxA4Ude07yHtVVqioRbo,18912
tokamax/_src/ops/normalization/triton_test.py,sha256=DBnUgpO_uwSY-qLV1-IeMKwawc7RKRoRwpuCJcCdgd8,4984
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=jZaJQRHRSKbBhKly3HMPu0xWJ3vt_Ho18btqpAGA_Gs,9929
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=syiA2WQgnw9I81IIccINIR_5ChS39g8ovEVGEFqa1cA,20293
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_common.py,sha256=sLBEFRqJnHpvvRjLUrdbfzhOb6vLUB0BPJtd_Nbl1wE,6419
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=BfvRCZoEQ0Mvyx4d6UyJ7Okrb0UyWoEaC6qDS4aYcNs,9502
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_fp8_quant.py,sha256=jMo0ZC3Xk3-wlee4Jhm8u4_2Ur9KDWKnWfRAbX8EE-I,35752
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_fp8_quant_test.py,sha256=rjEeMLk6akpYey1O_cewko6TlXjpO_bnecjAvTkXwFI,24464
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_i8_quant.py,sha256=VI9pcwLKl7UdOJEhhhtBAmkpuATylX3WJ93L1joGrG4,21764
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_i8_quant_test.py,sha256=M1ToRbySbD79e_ki9kGuml8UjfsenFrj3dNAV4nHRxg,6486
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant.py,sha256=0kieT7OrrSutTPkqdWchguAcc5Vi-QWja5ajFYz4u6o,18084
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant_post_scale.py,sha256=lmXgnnzIYkfh0AxOgcPpJke0AJyPhOaEqgBPkpUB8j4,16867
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=yAwpaLPmEIds1YtyyHWlLXOcR1YO1vZqyk68qjteVYo,9702
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90_quant.py,sha256=ZwC5EggBbvI7LxYxuZgIl-EG1EZTC0-J5-PYkfrioE0,13947
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_test.py,sha256=Va1Tc9GI74N02Yus4d0gohnkG5DxFFHmC6qLiHK5iDY,4504
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu.py,sha256=xE8OX-aDPKHdIDrik1z2iFX2yVuCYoNumiUrM6zDfjk,17786
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=XhredXDKjMcClpC0j2XHY92CXr6EvZ8i16twLTbn3lQ,17423
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_v2_test.py,sha256=MG1YNnQs5qUt8v5oTkC-E9iBPcvbY7uK_AdZthvSUqA,22221
tokamax/_src/ops/ragged_dot/test_base.py,sha256=tkN_V64DKMZT6odcNAQikW5MXY5peFCNeVuGYiJoYVc,11788
tokamax/_src/ops/ragged_dot/triton.py,sha256=hXzjccNGSWirIFcKukjqhXuOBhch_NqWWiQe5FZzzFE,21580
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=v4r8JdvQS22fwIEBEdZuGdNGIn79240OsOBLswWoyzc,3509
tokamax/_src/ops/ragged_gather/api_test.py,sha256=PVqvddhsxD3FpjD_dKnlaGk-wCzO0lvUkyCQtgHD6ow,2949
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=T_x9Bty-xj3s5Jeb-pfyfZUHxnumfWh1KBg__UNyBpI,1746
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu_kernel.py,sha256=wUu0dYFDJ865Lbs3Liaar6Kq-YJjj9hzRqm7vwuXcaE,10476
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu_test.py,sha256=_hY8kQhkHoB7ki_Iwoc2UnU3rwFEBJWH__SeELw_5wQ,1579
tokamax/_src/ops/ragged_gather/pallas_mosaic_v2_tpu.py,sha256=1rx7TI2lRjCN-6mikURcRYAoPF7dy8NErMCLlsM8HY4,2657
tokamax/_src/ops/ragged_gather/pallas_mosaic_v2_tpu_kernel.py,sha256=lKzhMIvmEUP60-4IDr2EAo1NSQBh99r_ktrpn-zVnXk,10363
tokamax/_src/ops/ragged_gather/pallas_mosaic_v2_tpu_test.py,sha256=EYaiL1eQ97NEXcc4BQYgy9AVOJTa7p_NOy2XXWbu70I,1697
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=q62ZS1oTtbiLnV_7n5s87MkHLZJ6L07NvCFqEeKDg_g,3360
tokamax/_src/ops/ragged_gather_reduce/api_test.py,sha256=atJRpcOksUaG1CwIMIuGvEfdWDHQiC8qGI1uh5-RRSM,3672
tokamax/_src/ops/ragged_gather_reduce/arg_specs.py,sha256=FaKGdzUxdNAEw2Ce2ZPUmmy60mIy7iNwGZ5oow1NUaU,2130
tokamax/_src/ops/ragged_gather_reduce/base.py,sha256=YSOhWSn8wDBJ6Q7tzIes_qLjVI_n9hQRAwoxTUDyRjM,2750
tokamax/_src/ops/ragged_gather_reduce/base_test.py,sha256=E12OJMwSLMc7QU2UugOZPFSe56pfl0v64tSjeJie64E,1759
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu.py,sha256=oyK3unVp4B4H6M7jnoi5eh7yA1m9BxyTkI_PoRL_a6o,2057
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu_kernel.py,sha256=zDGrgrxAvs__An4UrDhYj6DtrTX10kDhNxdvQOY2Gls,19310
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu_test.py,sha256=PVwKLcGY7Bobl3qP6ttPsAJA-lk01rH4CDGT-2muhBE,2455
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/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=N5-TDICT875SxFi_gcnAES2dpU-ECfjKWexAsCFPZPs,3623
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=bGnO72R7J8YMq6qNflP4pnAZOX8gmAb6EkTLgsuL27Y,10260
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=U6z1WxXF6mxXB63v2GCWbOkSCkxQCVA71zhi8Po7FI4,152036
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=zgsnHyt5t9S0-kL0xVHIPtyD76w-klSw7e0uM4-rVNc,130289
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=rljlQLXiYdhFDhChu8Mc3gj28eMjgasSgoIz8jkzYE8,183287
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_normalization.json,sha256=O-C7zYu8x3h7HJ0McOJlZ33AIW5i7r6wAqzXwPqhH_U,5226
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/pallas_triton_normalization_vjp.json,sha256=rHQftC_-vmNpBbY7g15iV-gbiHj5nDul80zIOQ_mulg,3264
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=NfbjzqYinNd_8i7FSa1J3pDremKmDSAzYP9yWps3t-I,136473
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/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=6D7vDYw6XAK9XraTN5rTRNgU3dMJRG4toUOKZ_ad70Y,94151
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=LalGRoyQ0rMxSLDZuvApB2CKlHcA6-O6CbzVvHxjWJU,51708
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/dot_product_attention.json,sha256=bWxYXIDB_k4dMvgQzYLPIjjfg1kHBPJsSwrCY6mNKpM,166778
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/dot_product_attention_vjp.json,sha256=mD2zAIT9sVubvGDQ9lPeMwIugyx_nR_wsgVtUbmn4OE,116544
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/gated_linear_unit.json,sha256=UxqXFJS7VqOmfgUMD7hfhQwU-VQyeSRaefWHJnAR0Ww,120347
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/jax_nn_dot_product_attention.json,sha256=8yHi1mFA8nc5aeRITDWVBRT3-uJrrmBqRg5qYeMlkJ4,151420
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/linear_softmax_cross_entropy_loss.json,sha256=RWCg0cgODe-bPVuFeBATLuLIda4m38puY9TUVstRhGY,111456
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/linear_softmax_cross_entropy_loss_vjp.json,sha256=y2kJ22m4RdHK_zxRClIZooJ5W36HkSudhtXwcgSNUc8,67083
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=1p5s8-yQi-C6uBgRXME7uGrHu1pPRMi2L7yAKCVe3HM,194112
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_flash_attention_vjp.json,sha256=cBniYs3Pni3DmLpH27XaPII6mPCqFgp74kJgIX-GiuU,138300
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_gated_linear_unit.json,sha256=Y2GUll1YWeGSGX5tuOTIM7ElBCKbi-bO8ZL1iQnm-T8,111381
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_ragged_dot.json,sha256=bXtpWNzJ1zooMgUGwxIENM_s_v0gVKx4XVN0SiDnJk8,45344
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_flash_attention.json,sha256=3Z1tEbwfkQj2G5M_QGZT_u9APvT0hx_K59fVrJvIU_Q,199826
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_flash_attention_vjp.json,sha256=sD7fv6O7XcUOF43BCbzv84YVChuF3DQiENpdejc_LfM,136428
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_normalization.json,sha256=eARcW39khJwZC-1L_NgL8YzzzsloJ-RqLqGldFWlbso,4853
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_normalization_vjp.json,sha256=S3slqeVcl-Ppbg5qdJUygN9W_eJ3m1Fy-uGwlfI24VA,3249
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=R2obqm6Qxhy8hj0bGcMY2CIBw0-z2LWLQkKU7MDq4Zo,49119
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triangle_multiplication.json,sha256=qn20NBXlI-eWllghw-0zM3cQ9Dieu6O8h-85Xn6DJNE,12641
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/xla_chunked_dot_product_attention.json,sha256=BZYXyM_VDcKQDjjZOiZ8h2BiSa-NV4oyGtm7cH1vbBo,149162
tokamax/data/autotuning/tpu7x/batched_rpa.json,sha256=KdF9NtHatJtPDyiOXkwULSWsIzLlAj8-oxM_MlKQrbU,43917
tokamax/data/autotuning/tpu7x/causal_conv1d_gated_delta_rule.json,sha256=1NbqyUXiZndiylL6j4b4yLt79ddyrb9jjCPYNAF4hLU,36480
tokamax/data/autotuning/tpu7x/dot_product_attention.json,sha256=LFZ-2NH6evDpo1J0oko-gQqul8DsqRwnNozaYIpR2uA,163907
tokamax/data/autotuning/tpu7x/dot_product_attention_vjp.json,sha256=0Yjqm0wXB5XHi8XQfjAKqCvL_vliiWolFRh7KKUIBOs,112435
tokamax/data/autotuning/tpu7x/linear_softmax_cross_entropy_loss.json,sha256=3J0IZBXOdMQIueQIDf6BCM6GcdkYASpAJJKSqQzdW34,67332
tokamax/data/autotuning/tpu7x/linear_softmax_cross_entropy_loss_vjp.json,sha256=JarV6oUV66KnvBmstnjwyZBxBGVn7mIFtjqPEPulA3g,78669
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=C8b5yLDXY1RcefF3IqQoBasCxZgEdfXCzNnpfuwH_T8,141243
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_flash_attention_vjp.json,sha256=pyQBUoVDGg7eJCdrtmcweejVTTgUXlHYeNc36T0pJEk,93512
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss.json,sha256=LozbNPd97Jns1IIQ-vMy2zeTi_HzJ-4boyZO1KZStio,46378
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss_vjp.json,sha256=JtC8D06Q12db8QwrIjckOVi6pEC-gGv-XbHImoIS4_g,57682
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_ragged_dot.json,sha256=_0ObBHnPZHw6OGmvLMBaky82oxtX5FBJiqXpwWfEt44,78299
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/ragged_dot.json,sha256=LGD-HCJ0_scofmbmlZZxR4kVe6L8v6l--UmCGWKeCn4,68685
tokamax/data/autotuning/tpu7x/triangle_multiplication.json,sha256=mMsYoqbsVMIbUHtaifKeOd4iKOPVOCYDWy-W9JANvR4,6200
tokamax/data/autotuning/tpu7x/xla_chunked_dot_product_attention.json,sha256=7n5F7Tkag3nisU_pGYPSzkPoktzea9u8ik8PYQjagxc,142159
tokamax/data/autotuning/tpu_v5/dot_product_attention.json,sha256=LfTTjon6KKyH44bVkHV8y_RUNPLGRQ_r47RsF5LrF3o,157186
tokamax/data/autotuning/tpu_v5/gated_linear_unit.json,sha256=tvlW77QyTqZQD5_8chuHa_pmfAc1rN43fLFN8_uATHk,114050
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_flash_attention.json,sha256=KC0rdHbhwy0Fj4-V8b6fO2mDcgoUYoDqylRJ1V5xuoA,136802
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_ragged_dot.json,sha256=wTneKqQJlicns-khxtn6q2MDOjn2K26E_Zepw4sTzVQ,90316
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=1FMWsTcPSvDUTVS7tq7RGdUJPa66a3iFFRk__HqzuPo,81985
tokamax/data/autotuning/tpu_v5/xla_chunked_dot_product_attention.json,sha256=sROxCJYCVv8Rt_VUEiOqf-Ea6WB9k96nxHZBGPDTFME,135384
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/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=YNJNMbf0u4R7Rw0qKwrVCIRR-G79psV7QaV5QMOTxWM,17759
tokamax/data/autotuning/tpu_v6_lite/dot_product_attention.json,sha256=ESIYZGC_dfxeAK1PYsPLXfcuqFa9gFgr_OyjvRo_RnA,162514
tokamax/data/autotuning/tpu_v6_lite/linear_softmax_cross_entropy_loss.json,sha256=jBH9xKIAdmSCteapWwqC_Og97t4970u2iw1uNEKyRrw,98631
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_flash_attention.json,sha256=fOQTQAma2XmXWbZD7qPVt9CM62xoukEzaSERFcDXAsU,138747
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_ragged_dot.json,sha256=K1L2kMxqHumAOgS7Dk7VSyf33BoOngKrakfF-d6DVIo,83155
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=qwiFgDYYvJ30GCrVtBB8bsyd6p1ogzDmTSPm6UyPBSs,80770
tokamax/data/autotuning/tpu_v6_lite/xla_chunked_dot_product_attention.json,sha256=zrcD27axfPv9rpRSH993YfFVtGxUcT7xQknDX0ea5Dk,147358
tokamax/experimental/torch_tpu/ops/torch_op.py,sha256=SIMX9JrL5KN58v4fTdmmx5dXOQb2UF_dXxLObl-p66Q,23386
tokamax/experimental/torch_tpu/ops/torch_op_test.py,sha256=aRFio500dPQRaUe3-zwDCRRbHGM0yLCjpaevUjWqA2k,36156
tokamax/experimental/torch_tpu/ops/torch_utils.py,sha256=zUtTrVoFDLSDhW3M-3fL3grLpmzzfrjCLOfurpT_RUU,4809
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base.py,sha256=GOhXhfAhDOIjrHPGInp-KoZjBP1Oqb9E6V7PeZoH1zE,8135
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base_test.py,sha256=MCHZZZb0udknA88hYTp92ar2dx1y5yrEC2oUvu7eORY,3953
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu.py,sha256=ZH9bEzU1u__dSabiAoy6w8q1_VFaSF9oNGCRqf9F7Do,7918
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu_test.py,sha256=5PwWpj3Be4WN7awFmGDYTW47RvPg7NNzf6bFxABWbAA,10117
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=lIeYl_3_XEWBWDCUOpOh-FeizG68NhwCL09k3GJ4Oq8,9392
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=zq0t8k-Lr5dzL7sxbK-89hzynXfD014dp-BnSvdeGgY,3401
tokamax/experimental/torch_tpu/ops/csa_gather/torch_base_test.py,sha256=Kp6YJLjAo0rv8JADbAb8Zj3kRt0kR1zxKV9ArXk-638,5377
tokamax/experimental/torch_tpu/ops/csa_gather/torch_pallas_mosaic_tpu.py,sha256=DOE7FWXResCwwFr3sYFZHM4itDIuLjjQxzEwHl1d_WU,4248
tokamax/experimental/torch_tpu/ops/csa_gather/torch_pallas_mosaic_tpu_test.py,sha256=pKXm6HsOYcXvg8d5oaQSIbzqbG0xAnLmrgH4wBj0olM,8738
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=U3sHtUNJTWDRuaKIxsLbdzVQAsyKp9X93X7awpWc2h4,5371
tokamax/experimental/torch_tpu/ops/fused_moe/torch_pallas_mosaic_tpu_test.py,sha256=ZLMfahf3BSsoy5moaaOfEh_4TgDfY2Qbe7WPQabjFas,8407
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/ragged_dot/torch_base.py,sha256=eGpXsQasnEYToQdYjyhzLOTiLJcWQh5XgujXvsO8ZxY,14332
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_base_test.py,sha256=XUpPxkUQjkcjA5HrNmFUR7WKdxySmDGtaDpPo7fWS5A,5695
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_pallas_mosaic_tpu.py,sha256=B2-X7Fr5Sd66xNGx0fm9i61MQn-m0L8FTuYSoAhOnJ0,8975
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_pallas_mosaic_tpu_test.py,sha256=MGqybnhTQUaQZ-8TvR1ag0D_nNDtZlCtaMqe_7_0pqU,10694
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=zjg3xi8TlnnDFhS1J5EUE9IgRCdNwOzWNJKbs1VoRLE,2926
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_pallas_mosaic_tpu_test.py,sha256=351pBsUXvnpOOShAYf8mk3MpxcPJS-Le1WDg_OdzdoI,6040
tokamax/experimental/torch_tpu/ops/topk/torch_base.py,sha256=TNRctyqAi4UEzRFaIhihBVQHfumm41kAr0uSVdo3MCI,3569
tokamax/experimental/torch_tpu/ops/topk/torch_base_test.py,sha256=oE1Ap2kc3lB4uUnEJxXwWg8he_Xhqzh7SH7woeZztkM,4845
tokamax/experimental/torch_tpu/ops/topk/torch_pallas_mosaic_tpu.py,sha256=aqdTX1R2yJ2JQJBF_LrcYey7Txwe3e1DtUuCycZXDQo,4560
tokamax/experimental/torch_tpu/ops/topk/torch_pallas_mosaic_tpu_test.py,sha256=JxiR43XH8gThnbW3Vb4tHER5jdQy02ni12fdi_cAV-Y,6926
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.16.dev20261009.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
tokamax-0.0.16.dev20261009.dist-info/METADATA,sha256=jOvEJHzWNC56sfrZRHwxtbl7blm8fhD6NCHBpNMjBJs,21988
tokamax-0.0.16.dev20261009.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
tokamax-0.0.16.dev20261009.dist-info/top_level.txt,sha256=xHgO7qOuYjrLH3047bLQLnwrjI76i9-xlChxNQIZn-Q,8
tokamax-0.0.16.dev20261009.dist-info/RECORD,,
