tokamax/__init__.py,sha256=MK544SjNihQKWJba2q7KUg3s0pcFBZCEfGbOAEPz1mo,2743
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=VoUnYyUIjx9D9ZFEfWyPKzCOBnu4L-UQwBKHMsL2LqY,25742
tokamax/_src/benchmarking_test.py,sha256=V3CkD5SzEuRfrXG8AQhiB7rn9ZD6ZPoXaOGx4KxAvus,8733
tokamax/_src/config.py,sha256=dvHfkVdRxSq_j0eJqbFivjs49egNho9j2MUBfj43-eU,4538
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=U-up1aiLtX5onxifXYqvntDTgTJfVbgXmfAcjYbkvGU,10782
tokamax/_src/hlo_utils_common.py,sha256=ubC8ogv_uhKa5q3FyjVsSyf6jIX_zSVtggi5inCUXlQ,4381
tokamax/_src/hlo_utils_common_test.py,sha256=7t1Glm5GtrrrDxBIvdUJE-gMfZLFzCt0ePgY0e-bssI,1090
tokamax/_src/hlo_utils_test.py,sha256=hTg4OpJgm2OQn14Q_5Iep3X15z3VgrWQ3ObcbpZG35U,19336
tokamax/_src/jaxtyping.py,sha256=_W4COs9cj3GdNE0B_J2QNhn86tt2JPTUaVzp8g_rtP4,1389
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=M7S_oLwn_7fbvOE4PvjTHLjMBvmywPb6oHZ4aJs2bmU,5583
tokamax/_src/mosaic_tpu.py,sha256=_8b9rZyL9TTPOn_zfAlcCYJELitwL_9V_1rm_2mWGIA,10481
tokamax/_src/numerics.py,sha256=otocVizo21g99FIRAZcJcGgNeNs5p9zX39VA8QhXkOk,8486
tokamax/_src/numerics_test.py,sha256=e9ZacB1T09mrtMwC04FYUAFfZucriydV9jBZ5g4XVz0,6384
tokamax/_src/precision.py,sha256=xUNXV38Zf-Z2rdAZLCTaauh0_ccxps5d4bW14xItUvw,7336
tokamax/_src/precision_test.py,sha256=fFifAxvC3-hv7tY4qLXUj6313laQkeIB9AN0c4uFxC8,3630
tokamax/_src/pydantic.py,sha256=fMZHtXrSRR_iRwc63I0DB_pAJNc13BA-5WcoEddTvlo,13395
tokamax/_src/pydantic_test.py,sha256=nV1MuutbvKw-A-wt4UvgIbf-te79VD9VebOSCy4DMU8,10051
tokamax/_src/quantization.py,sha256=gyqo8qBXUMcX5f15--v_zpflDjXqXqLnvA5V6WZ36KA,2913
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=B5jnsACEfIzmcy9A5QKtU4pZuT63BNKmeDSnNEn3gT8,1213
tokamax/_src/autotuning/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/autotuning/api.py,sha256=rVAZbXDuyzU2O6gUwzTu92eRBaUSerxNMoDgITxsBfo,14810
tokamax/_src/autotuning/api_test.py,sha256=TU9WhwzL_k6qTtHIG7KT5mt_4CTnWgpVP_rES_lMq9A,17422
tokamax/_src/autotuning/arg_spec.py,sha256=nOVyBgJx8YaTZELfdD9v-1zNkwmqyXJfoC47aOsFUkQ,2016
tokamax/_src/autotuning/autotuner.py,sha256=JkKEdyLXDsT7C6xGx40itzaj6K5qyq7jtFmap3zRMQo,8187
tokamax/_src/autotuning/cache.py,sha256=lhvylZRrfFKtT1VKidy3HJijBHKO3K70LRuSgKdO1UI,3682
tokamax/_src/autotuning/cache_test.py,sha256=OGfC62g19Fxgh7lhDNwFJn8N6AKxCV1HZ30fj7JTl9Y,3606
tokamax/_src/ops/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/op.py,sha256=RX-lhtuJ0YZ3rH4han-pbT3XxlKQwxyotaoNYm9mzX0,25529
tokamax/_src/ops/op_test.py,sha256=TtYwy2jhRqZCV-WRaRytexyWSRBkTdlrwqQBt2v3uy8,11263
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=PQPkOIlIoSPx-CSZbyv-fle-9XI3oMYWzkvtFizP8M0,32981
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=026OvmLfhYYNG4FOCS00rIIpOcRbcbk17I8DEN0A3qo,38108
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm90.py,sha256=r41B6zXBXw7B8P38bgcqOhauKasFfXFKJMg_UJThqwY,23915
tokamax/_src/ops/attention/pallas_mosaic_gpu_test.py,sha256=Cl343KQuoreot_vv7LmIt_Idd2i9r3lFhOb2aCWKF7I,8993
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp.py,sha256=DgHayI9i4G2bDhbRWOqnar5qoU-1wwr_u49EvT2MCZc,7286
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=ag1d1gH8rH2468g4r1Ah4XcwPBt-Cfj7tpYlskWKtGo,34974
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_kernel_sm90.py,sha256=Y5HrbhiEoFshhVnfgPi8AKCmdi7Xog_r_iYANbWxwUI,24987
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=bsbMLo-npsXzBoU0cIf_bYO92GcYyRBRKQOInCpW9Ew,39275
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/base.py,sha256=aLfJ8YSIoap1aoECN64OMxFBB-aohMsCZepHs1OgkzQ,2516
tokamax/_src/ops/causal_conv1d_gated_delta_rule/base_test.py,sha256=8_z3ZWY12_xsjGaFUOAIQDedhZk2YfD-9mPJj-xeytU,20648
tokamax/_src/ops/causal_conv1d_gated_delta_rule/compute_conv1d.py,sha256=5e5wqpVqSx73DgPX-DYF1ho4t3yOtABpv29k7GoOBmY,2459
tokamax/_src/ops/causal_conv1d_gated_delta_rule/compute_gdn.py,sha256=ejim_lItVBEVb5q3aW4167fhNJ2w-hulkIckv_obmAM,17514
tokamax/_src/ops/causal_conv1d_gated_delta_rule/config.py,sha256=fWwx_JpfWevqj1hrvKEMfVHHImCuAeAmqIw79M6fTsw,4153
tokamax/_src/ops/causal_conv1d_gated_delta_rule/memory_ref.py,sha256=BfTox-Jtyjlcjgb7fleqJuBTooeoct-wZJGawaSDSto,13142
tokamax/_src/ops/causal_conv1d_gated_delta_rule/metadata.py,sha256=6KZeQeVrkdwYl65qtoS75LIPpxusfhuBHE9k3ewBTDw,5453
tokamax/_src/ops/causal_conv1d_gated_delta_rule/pallas_mosaic_tpu.py,sha256=jTeocB8wdQo_FSS37SbPClLbS9Vz9wX1_ktUKbaVwXI,3305
tokamax/_src/ops/causal_conv1d_gated_delta_rule/pallas_mosaic_tpu_test.py,sha256=bFVYOvKcYyOVP_tQIIZGPJrgfglpLU2Rt1IuMQ_sgGw,20271
tokamax/_src/ops/causal_conv1d_gated_delta_rule/reference.py,sha256=YGoKupq3jdflNCyIYA18uxDczfNIpsUkGAW9pV2asZ8,15116
tokamax/_src/ops/causal_conv1d_gated_delta_rule/tiling.py,sha256=G2b8RCIv6PcFFBv2l0YxC4Bv2l3MfKgUjyMOrjjxUIA,11000
tokamax/_src/ops/causal_conv1d_gated_delta_rule/vmem_ldst.py,sha256=yCjYi3XN4QNtnn8kLD0V3FSBrLYwX0WYmcuSAIieDek,9738
tokamax/_src/ops/causal_conv1d_gated_delta_rule/wrapper.py,sha256=DKFNI5teTenRpmCHiz7-s6fslPiIwsnzScrsXXSEQHM,18768
tokamax/_src/ops/experimental/batched_rpa/__init__.py,sha256=ZXRlLcMbY-OQGSiroo2YyieNsy7ImsY4E9g_R20dQds,696
tokamax/_src/ops/experimental/batched_rpa/api.py,sha256=A4cAmBaEt3nFMd7uDvfSaVsibyfgLSiQh1Wyi1zJ1P4,3965
tokamax/_src/ops/experimental/batched_rpa/arg_specs.py,sha256=sZej609WzXROH6pGQ2Z_Gn7CX0G4u5vlPlgiV3QkAw0,6653
tokamax/_src/ops/experimental/batched_rpa/base.py,sha256=nKuacFJ8KEl5RgeO8PTlNwKeDvuJqhb8LwpE81Jc1NU,7505
tokamax/_src/ops/experimental/batched_rpa/base_test.py,sha256=smawEWXSzBDusUsLePEMDk5BbOLm4LNaFgWMx0n7464,6684
tokamax/_src/ops/experimental/batched_rpa/pallas_mosaic_tpu.py,sha256=e2bSJoXrOOqxsQD_rvHKNSrVSCKlOVsPTSvFIDLwzRc,7630
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=TsevwAij_IYd9hUk53syOT27y08ASCar0NllXU8ytY0,7945
tokamax/_src/ops/experimental/batched_rpa/types.py,sha256=CX0bGzYqsNN91BihTA_mSwIkZu1POEq_4EuBiXxv8pY,3706
tokamax/_src/ops/experimental/batched_rpa/kernel/__init__.py,sha256=YKJW4pOeOddU5njid32X3hV6RxrTb93syWZgZv4FHEc,950
tokamax/_src/ops/experimental/batched_rpa/kernel/bref_override.py,sha256=HG1yz5T-2w-nBkEmap34prlqqCm3UVVKxuoNukIvmPs,21727
tokamax/_src/ops/experimental/batched_rpa/kernel/configs.py,sha256=a5c_tSnLJ_VvQBpeiBTVkxrk-B_JtV5cLjV6yLYJ9Hg,14385
tokamax/_src/ops/experimental/batched_rpa/kernel/configs_test.py,sha256=rSUCvq5e7UjWXtLPpHQ0BO9ycdTwSh2NLb1MdjxCT-0,3825
tokamax/_src/ops/experimental/batched_rpa/kernel/flash_attention.py,sha256=LF3tnLXwk5GtgPk2QbwFWG9TPGnhk-WwkrlnsJItmGk,4912
tokamax/_src/ops/experimental/batched_rpa/kernel/kernel.py,sha256=xMZlZ1bqiZPyM1vN83Bj1bNe9a7aP2jfan4nFRUU7gk,30955
tokamax/_src/ops/experimental/batched_rpa/kernel/schedule.py,sha256=7B2fNHYa20MUS3w2cUqo-c_0LU43azscf0Sm82tBRPM,32849
tokamax/_src/ops/experimental/batched_rpa/kernel/schedule_cp.py,sha256=b7Z2j0CZlyxJY25xjWgJ1grlqHxG1ZcvSug_hhcVhRM,10657
tokamax/_src/ops/experimental/batched_rpa/kernel/seq_along_lane_parity_test.py,sha256=IcCHur9s7wFjXE-73-BLH9pUakEP9MoQ-GXITFZD_IE,4669
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=gllQKRw-F5BR5RLWkrz1IulpEprLuw-kKSEzcXQh0A8,6442
tokamax/_src/ops/experimental/batched_rpa/kernel/utils.py,sha256=uLn7V7xi8yNYrUbDIoQsUkunU9EP9y9_fPJBI827Exo,5349
tokamax/_src/ops/experimental/batched_rpa/kernel/wrapper.py,sha256=aPuas6K5HZVwjr4B92yo_kAF9KyA-9C97boJjUQOS5o,27987
tokamax/_src/ops/experimental/fused_moe/__init__.py,sha256=WevLGtYKhjC41DAhdMPtMfRRqm6czh43uMeSZK24uLE,2500
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=eed9hCXuakxcJ27IILijzOfgjOrCqXni-2klnXbcRjM,67019
tokamax/_src/ops/experimental/fused_moe/kernel.py,sha256=Zjv_f6-1-2wc4Dt9uYN_z5XEwbFhIxMXlrq2o_ZThS4,99356
tokamax/_src/ops/experimental/fused_moe/layer.py,sha256=wSRcm36knnd0w80GLGHJJnJNHj4VB68s91snWirSojA,34781
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=4N3UMSmzvYPx9mOXZXF5etXX30o41mAelbr4sPGuUrM,5147
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2.py,sha256=_lVDUL_nRlgwOdw27cY031ZC2hadf1ZDuULE5X3trF8,24902
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2_router.py,sha256=nPNCULvsL-fXj8UevBrDCAD7GgyZ7ombYCg0jlpvFC8,6206
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2_tables.py,sha256=kExI_7g-nfd5rWZnrQZ0kMu76xQQgyGF9Ec63gw-h08,11940
tokamax/_src/ops/experimental/fused_moe/test_fused_ep_moe_v2_tp_tokens.py,sha256=s_pd7VzspHNLD3ztIb42slXCHOBQE-x84x14IYyhDeU,13854
tokamax/_src/ops/experimental/fused_moe/test_tp_token_layout.py,sha256=KPcvhnOQPmNJ79voZ1TJgHbDQ2nTP0h1OGbLEl4KEWE,4831
tokamax/_src/ops/experimental/fused_moe/token_parallel.py,sha256=0nrC4KBorzry732jc-p48g_-9HpFPFPFfkNbVgh76bI,3983
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=8a4KycHPTDDmbIqSxCvcjQOfs98dVKAjpONyuDOy4g8,78000
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=BQQUbkwhuq5uFNwnS0roOp7cRiFk_dJNUiELFl3SmxY,62037
tokamax/_src/ops/experimental/gmm_v2/gmm_v2_perf_test.py,sha256=Q79QdcfUiZGKr0L9kvOgrZmbOzSeBgRlh8IjD2yLrFo,8763
tokamax/_src/ops/experimental/gmm_v2/gmm_v2_test.py,sha256=nzEJ3DjVvrQSRSftv4QrGz0XgiZw-Ngrdc8jC5T4xzM,55200
tokamax/_src/ops/experimental/gmm_v2/tgmm_v2.py,sha256=6E41q9BQI3OA9KN9YJN64jb8Q6RYAvjFvnetScXLGs8,31200
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=cL_RIVaEAns0zPrHvEZaFMqPgPObKHCuZOWqXxCtavM,6513
tokamax/_src/ops/experimental/kda/arg_specs.py,sha256=gcUBrAurRHoD6UnP9vqwIkfTXP0GWw2EMPRlUZ6EY7g,3326
tokamax/_src/ops/experimental/kda/base.py,sha256=7CtZ4GzqUg0NaJ4lKexx_mrY-W5vvt3yFx4EkjN4v2k,7323
tokamax/_src/ops/experimental/kda/base_test.py,sha256=0oYQSJPSk8-i6vB-nCEyE-USj8NYEc62p8zOj0u_unU,10500
tokamax/_src/ops/experimental/kda/common.py,sha256=QT6S7E6PZFv6rshZpgCDF2EHJjLJs_B8obM4Ds0f32s,30364
tokamax/_src/ops/experimental/kda/cp_utils.py,sha256=7s4u7U7v7XK_lK2sxfIEEhnW67U0u2BGRkOC5LOipBw,17448
tokamax/_src/ops/experimental/kda/pallas_mosaic_tpu.py,sha256=f4r7zZHlE6FMzJQ8z_bdHl1RQ5yWspbTB5KPjOIIccI,17451
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=ODtAMpn5u6awWBDqvEIUqFxSdALHQOk3lKn8CQJidSw,7518
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=vUQUeJ8j8w-cjR4e7G7zCdCjw4EsHBxhI0vIRPvrLOg,14004
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=ntZgrU4MQE4YhWUn5Gwhr7YNMFlDzw6Sd09nGtV-UtQ,4862
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=Sx9mHRmZKRvBdp5vAi1Z1vMvS1MNb_2ya7r0toenIP0,5897
tokamax/_src/ops/experimental/mla/pallas_mosaic_tpu_test.py,sha256=arYhkQf67fx7i0PQpeMjDinvhAZX6nCGxcXkV-AQLDM,3772
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/splash_attention/__init__.py,sha256=fmfcX8cskB_eEmnhvsaai-tSLNc3gaGwxL_A9UNTEn8,688
tokamax/_src/ops/experimental/tpu/splash_attention/api.py,sha256=RBDWF4mCfUnHMgA7kfsdnemWJMuRuMRICSKKJZMRkiI,4617
tokamax/_src/ops/experimental/tpu/splash_attention/api_test.py,sha256=yKEs2saR68RlIpQ1Zar9x9oWLsIM9Wni-x6R16Yfuxs,12736
tokamax/_src/ops/experimental/tpu/splash_attention/base.py,sha256=UyWu0oUhYNOUi6HPX_e1Y2cV7qmvUsG_yht7meoN9uM,5919
tokamax/_src/ops/experimental/tpu/splash_attention/base_test.py,sha256=ooT3xRrk1zvGpIafaxiCtQXTgDmziuEFIbOw0jOZG1s,6689
tokamax/_src/ops/experimental/tpu/splash_attention/pallas_mosaic_tpu.py,sha256=mBedsSmDjEXYUHXNButmsXke6WuqgyYMfa34I4L-ZoE,8469
tokamax/_src/ops/experimental/tpu/splash_attention/pallas_mosaic_tpu_test.py,sha256=EneSQpX35qg_QxENqkDSPa2p_fK2TVGf_-Tx960Y_g4,4830
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=qhtMkFhAL5iUDFq6qduu6BcyhKy3grKPeC8Yum8klgM,92014
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=rKYNVjfa9DACsT7AbH0CFKqkuIo8iRVX6NuXAFBA54k,51872
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=0UZtIRWm_jUabgyLU6yWQgp2dZE6y-Y1dqjMup5dKbQ,19865
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_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=bkpa0xe4bqsAs6c3OrcpNzjU5ajgbxA2WiFwASB21Dc,4128
tokamax/_src/ops/experimental/tpu/topk/api_test.py,sha256=FasILGncb2g70PrKonUOB-XlvG15V2PeX4njP_A62YE,1962
tokamax/_src/ops/experimental/tpu/topk/base.py,sha256=CUB5nYsUZjLnjqMcESc6cZ4G1_o7aGUgzJLRh9UgIc8,3760
tokamax/_src/ops/experimental/tpu/topk/base_test.py,sha256=zOfTgUdpN4zU8KTf36NnVGeqgwl5xeq3XTJOVF6Ou4Y,1789
tokamax/_src/ops/experimental/tpu/topk/pallas_mosaic_tpu.py,sha256=jh4fdHaZ3neKHCWaeuc5jT8U60Nqx0R-30A8WMl6T0Q,3361
tokamax/_src/ops/experimental/tpu/topk/pallas_mosaic_tpu_kernel.py,sha256=-TVEpMFKhirbZxERNjQd5LrWEeT4z2_6ez7iigtD6sc,34843
tokamax/_src/ops/experimental/tpu/topk/pallas_mosaic_tpu_test.py,sha256=33ZuCvEb4dNoXAUiMQGin5nWmJ2ZDuo_jEPpNx-O9Xc,2078
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=avryAQHoZA4IvyKX5RvAjai_pzVbuS80DH3n9twL4sY,3507
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=7I6ItGPWSzfz_2X28mROjKXf-wl47WN1jZzPic_pnrA,2961
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu.py,sha256=5VzxmW7H2pfYA0Zcf9UQRGwnVE4S_ilabbmYtE0_Rmw,3864
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=zUqRv81q-tZvdlDeX3lPFRpe4aAoKkPx2jcyDNMhdvw,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=6HK1SlcHcKzSIELW-Xn10JzxTmsijIiN6cxmvYngvzQ,1666
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=dse-2OgNuOE7hwWFGr7JKef_BRuVwMTvevV4deP6q5I,2074
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=1lliqGrJiAbG1B41qN7wsoHXpdBECxH--xcMnXiPM4o,7466
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel.py,sha256=pesXcm_x9g1hAaB4I0QJnKz02mmaI18mpPzsJkPCMUo,51936
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel_test.py,sha256=N_x4QTsIwhp7bxNx4nIkBskxZhQoLI0L9VjyUxSfCJ4,25762
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_test.py,sha256=HGNd15tZ7N_wGo3vsvleReDCFLC-Q1ZBornYZPtzqqY,3400
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=cME1KGCi7yxlepZd2YH6zGZOz5wVvfbi9bePk5Kr3Jw,2193
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/ragged_dot/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/ragged_dot/api.py,sha256=J4LOXK5BsnVjl3V86liuKv0jksOD6i_gL0rL1Khl6Gk,10461
tokamax/_src/ops/ragged_dot/api_test.py,sha256=5PLGk44Ym2rdy_4GTRTsofYpbnb50n7YAMBuiatSlhQ,11920
tokamax/_src/ops/ragged_dot/arg_specs.py,sha256=cjV-pVkWkYQEgJ5MJWjhuUfWYu0JXs6NOKVoD5GyIPA,9621
tokamax/_src/ops/ragged_dot/base.py,sha256=7He4C3fX0Yp2nps7ercHJaG4K0N3refonwcNOP_je1w,20055
tokamax/_src/ops/ragged_dot/base_test.py,sha256=AKSHFalUo4y4UFhKqytLu6hfZOObdYpBOuLI2jOrqjE,2151
tokamax/_src/ops/ragged_dot/bench.py,sha256=u0gwZJVrOEtKFbBQwqkvPOtBm7dduoII_5nVH0rGS00,3766
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu.py,sha256=2Ivv3RL-Nhtw2AS8NzqrruU4WcEdnNBkr0Y8QSupcWo,20939
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=ffDAqKzVhQXvtPMg-BHfF-VTQqXg1sNPY3OhNkDz20s,6570
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=Qm0Ep73aXfQxvSiWYXRCkl-fmDH2sFKOOzKmTqmyRO4,4505
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu.py,sha256=eHw7MBnPojG_P_wnzt2ngoGsskKFbPtmWQr49UOrXuM,18432
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=Vg-UDiyC8aseYSIsy57m6n0sCYTqDPVSm3zIhEtlguE,12764
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_v2_test.py,sha256=3dWZXYGsFPAYyUYoe-QkPjyN9iiXSUpMkAYpkLr8CS0,21305
tokamax/_src/ops/ragged_dot/pallas_triton.py,sha256=YTkPNChIe9AqwA645TYvXhaG8sMsmtpnKGFzGSWXSEA,16739
tokamax/_src/ops/ragged_dot/pallas_triton_test.py,sha256=LgRlTZ6ze41xVx_Sm5cCpwJ2QvtLXHScMOR5gIJ3Zk0,2708
tokamax/_src/ops/ragged_dot/test_base.py,sha256=yshoZV8Wi7F8HxhzIdNCBL_7n8qruFwqUhn0ooEuvHg,11806
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=YVr_d1Ve5dCKPk7IKl_2PwevVw5oqlbslzF2kfSMOaY,2135
tokamax/_src/ops/ragged_gather/arg_specs.py,sha256=8rieM0BEMweWQgBwLRDFf712ofa2LTUlUYRi66rJcrI,2216
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=aeuYnVLR28YvIrgOluMefAwD62sOOgkZj6me_8HR14k,10472
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu_test.py,sha256=cw-lYs_GcxbHoa0LbKGeCua865l411qICJcLqEoPt80,2230
tokamax/_src/ops/ragged_gather/pallas_mosaic_v2_tpu.py,sha256=U_0RrfjkKyZ-v3xSFaQhMcVid-4klPx-_PdfwzOp7YQ,1782
tokamax/_src/ops/ragged_gather/pallas_mosaic_v2_tpu_kernel.py,sha256=ddKMzKgwxmjc1gbd9RaTDVY42RIeu-yjq6H2DSTFrdo,9656
tokamax/_src/ops/ragged_gather/pallas_mosaic_v2_tpu_test.py,sha256=0o5SxhEQibC8gmz34YLHdf49-LZC4-McrqETTO29lPo,2243
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=NuPs_gQSXkrVlSoHduWFtXc1XQfnLlpJrQ1t7OR3wPA,19306
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=5Ka9eK6l37dprqzgfQqyb6CVOLXD3PFU918xRXsKPwo,13698
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=KCsQGbMBpawnqWdQHRkAhosIt039eyL4Av-x5z2p2YE,4620
tokamax/_src/ops/triangle_multiplication/api_test.py,sha256=-JsctbYVRCjw-B-GVtTs0u3UM5vYCMQUnRB6xymXVSw,3203
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=G1Rf0ZJ9bJcy_yuARIDJBl8_5DxtKF4Mnt7JMgXnd6Y,3608
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=9NPGjGVxti7Ui3vQAbx-pQI3pvWkvX7G1DLL9bXuETw,10131
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=Rluuozus41DIGYQIPjFs42ZUgQF26GEYlGggBcPU4aQ,25942
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/ragged_dot.json,sha256=cOeLVwwTLt3SlOShABvnwyirKmzr0-QyRF1E9f0dMyg,69661
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=1IvSE8gH6n6e-8E3twORI7rnIjqHohdzHbDR4KgtvSE,82099
tokamax/data/autotuning/nvidia_b200/pallas_triton_ragged_dot.json,sha256=btDmnN1DIfbrfdKAkQJfCrofifQbS_5-3P4neNWLCrQ,110937
tokamax/data/autotuning/nvidia_b200/ragged_dot.json,sha256=XnFrgdIZWNMz9klzSsklGZFxhs3Y_KZLTNjD2b0GpiM,75195
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=7s1jlYDsI7yHkuU2wsV7VOWhGPluTERS2J2DKBopTeg,52004
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=SKOFKo4pvWwJbreeQZXY5CfL6sgdnSj3JY_USeDqWa0,57645
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_dot.json,sha256=HSQsiqdAnv0k5WJxudo6VDzRnpQVi6Wz5LDBLTSe99k,62735
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triangle_multiplication.json,sha256=qn20NBXlI-eWllghw-0zM3cQ9Dieu6O8h-85Xn6DJNE,12641
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/xla_chunked_dot_product_attention.json,sha256=BZYXyM_VDcKQDjjZOiZ8h2BiSa-NV4oyGtm7cH1vbBo,149162
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_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=-4rYLWjS9JNu-xaSgMAsOmEj8BXxEY4UruYyb4E0r0U,78113
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=Ie9YysKA5BnTJUhox2WTG8C3gx7qqHBmKwSNrmoCLQ4,103732
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=UNL7lb-9tlv30ib9GQO9V6FOp_74UGckU5KNHbEkvk8,73643
tokamax/data/autotuning/tpu_v5/ragged_dot.json,sha256=Xey-RobmnHw3P8W6eqPLbJkg-WvodwiBlzxKiHxywZ8,95401
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=lV4qDu8HoUDgh5PkqqfbEpGyVBWF4Pg2qjVwd8fo0s8,40172
tokamax/data/autotuning/tpu_v5_lite/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=3CMg1tCDhrzckXuuzKQnUB7yODeKGeGU4Dwu3_dApro,646
tokamax/data/autotuning/tpu_v5_lite/ragged_dot.json,sha256=ttScI0i4IXTcIgzrrEce8r2JmGLrL01Bygf2dWJWlF8,176045
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=tgt_5H685keMjlCo2PRzj_RGEKXx-9VU0NtAqMKGm_U,95668
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=vNrRMKtLUZTVLQSyvSCi3qktPChqLF2jA29kCo6NcOM,73414
tokamax/data/autotuning/tpu_v6_lite/ragged_dot.json,sha256=juHJYZ6LTiy6CUbZjcyn_1tkCmlXjvJz7oSPoHDC1is,94186
tokamax/data/autotuning/tpu_v6_lite/xla_chunked_dot_product_attention.json,sha256=zrcD27axfPv9rpRSH993YfFVtGxUcT7xQknDX0ea5Dk,147358
tokamax/experimental/torch_tpu/ops/torch_op.py,sha256=wukK8RTo2vSWRx-DZngik5J-u2S6eyPzX_O81kJrcAo,15432
tokamax/experimental/torch_tpu/ops/torch_op_test.py,sha256=-lm0wLN8tjxI7QVOreMqfjd56PygoCJlli2vaanroOE,18007
tokamax/experimental/torch_tpu/ops/torch_utils.py,sha256=HR4gg4hhE3ITtMghZl9eR_86JqsatFpEXapyqDNPc9Y,4145
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base.py,sha256=fNmljXwnCgDuHzReiJ2GH_g6zI1z189sR5LFEC2Pye8,5848
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base_test.py,sha256=LKKZhCwRC088L56tWfQchQd8ECei2MfsLwsoYSciLRg,3994
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu.py,sha256=ETU33za-8Naf5aTIXrBoycDungQ5zkOytC--VdLnmdI,6456
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu_test.py,sha256=74FleS5u6ulur-qc9JcbJUqMUab-KZOaVS3mKG_B7Bw,5028
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_base.py,sha256=Xwq1zbNLXPFebMYr_0xP1H8egAGBIsVn6vDtWH4zPSU,6819
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=RMVCsFhbElOk28DEsIYWSew10nWzEqJ-1hUMipBC99I,6072
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_pallas_mosaic_tpu_test.py,sha256=Dp0aOnctY00_g7_3TNcjjQggTgk0LvVN6se-8N9YreY,10065
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.15.dev20260922.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
tokamax-0.0.15.dev20260922.dist-info/METADATA,sha256=146uMivTmY5AttawL3M6oLHvian4tnS_SLWKDaMaa78,22241
tokamax-0.0.15.dev20260922.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
tokamax-0.0.15.dev20260922.dist-info/top_level.txt,sha256=xHgO7qOuYjrLH3047bLQLnwrjI76i9-xlChxNQIZn-Q,8
tokamax-0.0.15.dev20260922.dist-info/RECORD,,
