tokamax/__init__.py,sha256=hDHHrk1BXv3YmzawR3U9f2n7PW9SFRCMAiOOW93M94s,2841
tokamax/autotuning.py,sha256=Z3KLjS9wXkd49fTz0jyH8EGDPh7tTARSkcq3fk4jMUc,1059
tokamax/benchmarking.py,sha256=sJxtRSAuHgW4JbGMyHw07vvUwRR_z8NRVYGVHCZrehw,1097
tokamax/config.py,sha256=6p8RoBezruw_UOP8AL_PS9SOF4wSjH2Z9wv4TsOrNY8,1024
tokamax/conftest.py,sha256=GVumgMTyrVZH7cR2Tj7UKjKQbEoX3L2gQKIpaTNYRhE,3884
tokamax/tokamax_test.py,sha256=y4CYurAN3ypr-HajbeGkG42zF-BbS9AHtpaK4htbTcU,5609
tokamax/_src/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ad.py,sha256=ztYw3M7ZfoMcjdl4MowW7RHHtnoyK2ZU6lfAKnRVTTY,4334
tokamax/_src/ad_test.py,sha256=DmhH5K_PeUsuqblEvCoDtwVCYS2i0wqdsirRzkp5JYE,3947
tokamax/_src/batching.py,sha256=zzALo24O680-Wni4Vys3ZxPbf8gyncjhjKCsE4tjQTo,8245
tokamax/_src/batching_test.py,sha256=PPdNyzeaPWHTxZmYgpHtaSKzL_thN-LXnW7_Gys5-PI,7793
tokamax/_src/benchmarking.py,sha256=vP7lbC3urS7si6fl5GW07jTi146IPh0XfoQXlvj1cf8,28276
tokamax/_src/benchmarking_test.py,sha256=ZeFCiUeno2tgxYlCaLww3O76BEROfsucudKaOEdaPAY,10617
tokamax/_src/config.py,sha256=3W7Cs8W7zFT81QkHSYLg_oqtWe2wtZoVFaBn42SrS-c,5553
tokamax/_src/config_test.py,sha256=ZShG9IQj0YbyWp4ontfZi8c3YWTyxotsd4eYlEfE0OQ,3946
tokamax/_src/gpu_utils.py,sha256=o4x1YMhLW6F2-cTbdGNeowz6q2VbRtiZq2n2BUTmMGQ,3576
tokamax/_src/gpu_utils_test.py,sha256=xZCfpeqxe87_1EVqlpGq5dDobi5HPHdT74vrsSxWUo4,3104
tokamax/_src/hlo_utils.py,sha256=EVlrlp6QlZU14jOuXsK0HO9xa2pmljcKQyPsizhB8RQ,11265
tokamax/_src/hlo_utils_common.py,sha256=25DVWKAOO6j4d-PnjsgTWSATHHdAOORE00Yd0JqViuE,4173
tokamax/_src/hlo_utils_common_test.py,sha256=uSNEu6beRJ3hNrPiyh-mpY_qwqTHJi3N0m72IHwdqqU,1752
tokamax/_src/hlo_utils_test.py,sha256=XuJihTw7OugJvirWe3PjvsktqHCnCSScE858rSfPr_k,18909
tokamax/_src/jaxtyping.py,sha256=mXl08fcQMMmiZVQ2I7IKgDONrL1GMBcl3f-WWLemabE,1863
tokamax/_src/jaxtyping_test.py,sha256=vDDdV7JyVds-SpyReKiqZr1-WbAT-3qRnXloYLVRYGE,1558
tokamax/_src/mosaic_gpu.py,sha256=MuWca6LEi8ugH-VLhCeIQr98PFfK6PssD5HWLSuRnY0,16919
tokamax/_src/mosaic_gpu_test.py,sha256=UI8BCwfJVRURXTO8a2vY4Rbp09werG1CYiAtsdBn8S0,5393
tokamax/_src/mosaic_tpu.py,sha256=FUms_ZmlcIV9R2ZGGO_ayRj6pUbs7DUTV_8WIOgmw5A,10558
tokamax/_src/numerics.py,sha256=ORyxMFxwPIb6Vx4s1M0hTBS17B9jlLurXOx5_xeRHmQ,10214
tokamax/_src/numerics_test.py,sha256=O5muLl2mZhPz4QiT7us4vnO4Y0z_afG52Fa6Wr79_X4,7653
tokamax/_src/precision.py,sha256=7y0HcRF6wzgtJLYv57U4JX1XePE4ZORaSck7sopgcVA,6422
tokamax/_src/precision_test.py,sha256=fFifAxvC3-hv7tY4qLXUj6313laQkeIB9AN0c4uFxC8,3630
tokamax/_src/pydantic.py,sha256=ODJiK6GmVyONNyr4rnL75wtoNtDIZwaIbngkzYe5Ocw,15809
tokamax/_src/pydantic_test.py,sha256=17_ylf_sF_TKCPBGGp4lOrVFNi9l0iJcX5tc1cUPJvg,11827
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=bPnIjI7eJK3PBK4Ne7DXLTm8UtKpEjecv1y-IbOX1Hc,10704
tokamax/_src/utils.py,sha256=M99bvel6cptkiocbsR2WHEAwe-QuqBbZ7izFhRcl9is,1898
tokamax/_src/version.py,sha256=10RRbqQ0XOX_64Y7YlC3HHrFT5NX_LL2hNqAsuOdsH0,1213
tokamax/_src/autotuning/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/autotuning/api.py,sha256=UbO5XinCglgMjRG3dquJbBnbT7jl4aHggD-6lhog3K0,14051
tokamax/_src/autotuning/api_test.py,sha256=iIJIPAhvYrI3yea0gSfQEzZVA1aR3dqmpw_AU4j4Pr4,26521
tokamax/_src/autotuning/arg_spec.py,sha256=SuYiSxuBI1PRzr4AIXdWs-ZnvwBrpCMqg5en-xL4ImA,3076
tokamax/_src/autotuning/autotuner.py,sha256=TgZs8aVv8IqOyyxfwQrlog7dO4dYKCfBq7wPeP80TPQ,8988
tokamax/_src/autotuning/cache.py,sha256=5SAjGxOD887eW4PIVi9ZEe2-zZNhYP_HKyPHSSOw5Rg,3778
tokamax/_src/autotuning/cache_test.py,sha256=yL7ooFbvWUHY1x1TCpwOvCzUjpYNS9XzXt6jVDTtL1Q,4877
tokamax/_src/ops/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/op.py,sha256=TtnJiuVIPKaqvzHK-a6FIfxnHp2_aoJwuHYsvygwlCw,26129
tokamax/_src/ops/op_test.py,sha256=th3EgwJXVC80eK6s3ulHcJw4HyGbpGr19uiLRlELn1A,12323
tokamax/_src/ops/ops_list.py,sha256=RMqhjslP19BBl5r7O5WtTY2E_fEEytxN6U9LTkXmiYE,21628
tokamax/_src/ops/ops_list_test.py,sha256=WS08f5chlTMrJR2seS0MizQztjRoZxdhr-9yl_2LnVA,19982
tokamax/_src/ops/attention/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/attention/api.py,sha256=pzprhlL-6G9fTV9tVQAkjOLPJ3PNKR2vM58GqcspSSc,7009
tokamax/_src/ops/attention/api_sharding_test.py,sha256=bTecmL_es6iZK5zAFg9h6wMx8oVeOucY-SlP9qwMgKk,3344
tokamax/_src/ops/attention/api_test.py,sha256=O2lZgEwTMpiIeFQzQoX3jWChlFOU76AVrLBAbtCTwqg,18466
tokamax/_src/ops/attention/arg_specs.py,sha256=1lR-ABTzWcBePhBGXb5HEySf1F1N9u4Py3m6pf0APDY,6593
tokamax/_src/ops/attention/base.py,sha256=sIIwxX1DJ2mRsh0RxJR4mD7QXPEE7E7BaIA5tMUFGzY,38559
tokamax/_src/ops/attention/base_test.py,sha256=IWE370cFZsO261x0Q1dz1Yl_Dujh5jJaE5jYtY7HoOo,6754
tokamax/_src/ops/attention/bench.py,sha256=W7_KCHrQsxUPiNeNupRAbYNkPq1xM_USF6mgX3xFiuA,2489
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=HZVoD1J5w1s28IP7SpZh6GZzqpq3qscvmlm-0r-J_ew,9729
tokamax/_src/ops/attention/pallas_mosaic_gpu_common.py,sha256=LS_2x67kpFEFhqa7zPrQMYaZbdx1CPekJZFWIYgqP3A,16967
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm100.py,sha256=9G6uswkWX7R0fsxrqOhJjBP958Rqe9dKhu2lX9FKLZQ,38719
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm80.py,sha256=cKTy_sRHwom41_gMvyZUgKY7rTd2qc-SVyI58jsoioM,58260
tokamax/_src/ops/attention/pallas_mosaic_gpu_kernel_sm90.py,sha256=nMJwSigiec2J2huos8rrnInDPkmOJQAh8MmCdCWOVwc,25030
tokamax/_src/ops/attention/pallas_mosaic_gpu_test.py,sha256=mnmQxkRLSORRvgkqabZLJQKKZiRwSxtzBfxiPmMu53w,22646
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp.py,sha256=Ztj5IAe_kKGWxJd3QxzgRldkCdmUQKo7mAo6e2MK0PE,8777
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_common.py,sha256=uqInKkTtrMQXAMUp0eA-csnqoentCqtzWscLEkP1nPo,1685
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_kernel_sm100.py,sha256=ZVZBbe1Ev2oQo2rQc-KsVJwFOsbNiD-uIP4eZ5fLbLU,38409
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_kernel_sm80.py,sha256=EgmL6TrAAO1BlYjxwcW7CFn_6ruA_Oyfwx1_qAjHBVU,64506
tokamax/_src/ops/attention/pallas_mosaic_gpu_vjp_kernel_sm90.py,sha256=dEdmTxqthy-9h4dV7vDGLektrU0wL6YnPyHuZgFw7sw,27225
tokamax/_src/ops/attention/pallas_mosaic_tpu.py,sha256=v4OJohJ2QkPkJmkyiq-aWQRK8wr4BY6PziS-Il89dxo,8008
tokamax/_src/ops/attention/pallas_mosaic_tpu_common.py,sha256=dJtsQlvsmTu7rUmd2ieuZYuCP3bChXZ1yDZHilr6kQg,4244
tokamax/_src/ops/attention/pallas_mosaic_tpu_test.py,sha256=HVcst01ECr3tbkyFbqag6Ru8zIf2eaemg7USmM0G_R8,6928
tokamax/_src/ops/attention/pallas_mosaic_tpu_vjp.py,sha256=Hg6PIlVQcyjF_ncg3L85TBK4FQIh4ZjwS11zMAOw7X0,8415
tokamax/_src/ops/attention/test_base.py,sha256=n3HKw7KXWPa2Hp3wWp23kZZK9MjbdJEg0IiMy_3epR4,39318
tokamax/_src/ops/attention/triton.py,sha256=PA9RVftUTiJnStf2dg2asVTNL6tkugYwbzayEzGTNxU,33691
tokamax/_src/ops/attention/triton_test.py,sha256=efa0gxtT-RF_wig-4b4qA9XfsC0xu3Xc52vzA5BPRoE,17270
tokamax/_src/ops/attention/triton_vjp.py,sha256=8zgy8421iYaaS3axeLrlWxNToPloxhYxJ_uojM9OEsM,32075
tokamax/_src/ops/attention/xla_chunked.py,sha256=NudbnrDY4UZrg9WXp73NC9_VT5CwFuwxuuiNN6eWgkc,13649
tokamax/_src/ops/attention/xla_chunked_test.py,sha256=aQcFiI4yVhIUDGbOv-2dLWfaUuoR6c26E2sXmsz0-3s,8750
tokamax/_src/ops/batched_rpa/__init__.py,sha256=ZXRlLcMbY-OQGSiroo2YyieNsy7ImsY4E9g_R20dQds,696
tokamax/_src/ops/batched_rpa/api.py,sha256=Ntq5WDOpqKS73OH1uNuhsbJL2v_GIoESDhxnqHPdj3U,3944
tokamax/_src/ops/batched_rpa/arg_specs.py,sha256=13Zj-iTTqld9PmLjIxNRlDi8StGNFEFdS7GTXa7b2ek,16331
tokamax/_src/ops/batched_rpa/base.py,sha256=DoAKAZZ-on2kVqm4aExsWlMUm86Sl8-HGfGvRWs-VdQ,7510
tokamax/_src/ops/batched_rpa/base_test.py,sha256=AnUPdYPZvshSsB94dKoCJfbTh1qFRtlLi7_ZeVA1xLs,6642
tokamax/_src/ops/batched_rpa/pallas_mosaic_tpu.py,sha256=nI6jl5Xtq6V0OWguBhEyGhP6AGhss6EDwjPLyXlCplU,7278
tokamax/_src/ops/batched_rpa/pallas_mosaic_tpu_test.py,sha256=_fhwmNQGsBHhvL_sz9O1sehJgiUrMaZ9zgn-w7ylKJM,4248
tokamax/_src/ops/batched_rpa/reference.py,sha256=4JdxX3jaSiAxCEi_HllTaOnMELcbbK2XRtJdIwNDc2g,8751
tokamax/_src/ops/batched_rpa/kernel/__init__.py,sha256=YKJW4pOeOddU5njid32X3hV6RxrTb93syWZgZv4FHEc,950
tokamax/_src/ops/batched_rpa/kernel/bref_override.py,sha256=DTUoQXNdeDbio6yZKMsD8kYcCU3Kv4-PUgNCvdCo9Rw,24531
tokamax/_src/ops/batched_rpa/kernel/configs.py,sha256=dFmA_CJIkq-rtDwCi59_jj4KzDWRt2SK9lb7S0g4JQU,22512
tokamax/_src/ops/batched_rpa/kernel/configs_test.py,sha256=Cry3l0XJwgWsz99-4DPy65HRIDXfk3ivE4gUKbng6SE,4019
tokamax/_src/ops/batched_rpa/kernel/correctness_test_per_token_quantization.py,sha256=bnKanBNAc5wuqtJdPNgBKgGoziPbNNLb--B_hOuOUmA,43478
tokamax/_src/ops/batched_rpa/kernel/flash_attention.py,sha256=9Eu7ZAj8LDhv3dhh-wfMJhU2uCI-rOO3PzCV47AkALo,6706
tokamax/_src/ops/batched_rpa/kernel/kernel.py,sha256=DPz4rqlypnOIOxLCfZRgMtLhtQChBqX5KNGvQv3Z4WY,41622
tokamax/_src/ops/batched_rpa/kernel/pipeline.py,sha256=LqCi-E7uV_bRgjQUZtI1xYDJfIM066yc_NyGbvpquHc,4561
tokamax/_src/ops/batched_rpa/kernel/schedule.py,sha256=Im1Iy0mC1p5149Q0AIkVOPNuZZcaDkqwz1J-6niDML4,31306
tokamax/_src/ops/batched_rpa/kernel/schedule_cp.py,sha256=eVqbCqxeaIsWqxVYVfczzeWseciS0MZFAli8RhddVTI,10455
tokamax/_src/ops/batched_rpa/kernel/seq_along_lane_parity_test.py,sha256=lqlC0BXVSryD05u35c48T8dYzr1CIis7EJQhFvpFevE,7244
tokamax/_src/ops/batched_rpa/kernel/speculative_decode_test.py,sha256=JSQu4LZ3cDkA9njrp08zFH9Md-osrTpYB7gC0u5sZX8,6858
tokamax/_src/ops/batched_rpa/kernel/stacked_rpa_kv_writeback_test.py,sha256=doMJqddIRmZSjRGBifnXIAVWNtNksC2Zs2GdWPo17Vc,8328
tokamax/_src/ops/batched_rpa/kernel/stitch_utils.py,sha256=WwlToiUb294uVwF13pnG_LzrcQ46uU0Ys9kydmrF0tI,7259
tokamax/_src/ops/batched_rpa/kernel/utils.py,sha256=iUdrx0piNMuOSnDMsdS5XqedBNQcTf109tPH9oWewS0,5561
tokamax/_src/ops/batched_rpa/kernel/wrapper.py,sha256=W8jXWQLXa0zoZ2IYb1LNQIsNtt4misUT_NtLfqoAoEU,33177
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=BfdevfJHGS3B1vXh8r5M6vJAvcErVntV1wXJ01VyPug,31062
tokamax/_src/ops/causal_conv1d_gated_delta_rule/base.py,sha256=qFh8yIW9T3O7sVffEKcd6OriYrTUaqunvzY5e777kr8,3177
tokamax/_src/ops/causal_conv1d_gated_delta_rule/base_test.py,sha256=6nUcNhop-sZIp2DwogVMQ7X_HNOBeD94-wwniRLblNU,7046
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=IW_r50cctirLcaca2uJPJOaH93XjeHVuRJX8Jxr4eII,19757
tokamax/_src/ops/causal_conv1d_gated_delta_rule/compute_kda.py,sha256=bYnfJimlJpawfKRFRk-wpy_6rxVn0CIFH3xKHTDJpag,17156
tokamax/_src/ops/causal_conv1d_gated_delta_rule/config.py,sha256=a1E82_creMyaYMtQzgOUVIEkgM6W89XpJyhK6r7s890,8818
tokamax/_src/ops/causal_conv1d_gated_delta_rule/gdn_attention_test.py,sha256=b7HY9ilGqGWVb52TD7FTgkEUcKkZirO66EDFk9SlBZ8,114839
tokamax/_src/ops/causal_conv1d_gated_delta_rule/kda_attention_test.py,sha256=teT8v3eRASA5O_jH7xLoFymmO22ktnmhY2eQYVEOf3Q,36502
tokamax/_src/ops/causal_conv1d_gated_delta_rule/memory_ref.py,sha256=KzTCotCsLa1O8fg_1M9RpkPnQqJ_Bk55ZgpJ26YpZVo,23120
tokamax/_src/ops/causal_conv1d_gated_delta_rule/metadata.py,sha256=9X061ibGA6LCNC1EU9W9OeJtD-xEuuPku5OyeTFvsMQ,7027
tokamax/_src/ops/causal_conv1d_gated_delta_rule/pallas_mosaic_tpu.py,sha256=4E5ZW1OgxeU_9WBumTCRfxRIRy5tiyN93Jh2mUXFFDs,7929
tokamax/_src/ops/causal_conv1d_gated_delta_rule/pallas_mosaic_tpu_test.py,sha256=lxsscbQnGg5uj1Q_uCcKUsnpT3nvSTbJwqmTXH5UwKs,5112
tokamax/_src/ops/causal_conv1d_gated_delta_rule/reference.py,sha256=Izigxc4adGIzWzGvkPmOuuAI8rH63PJKkkKyGlIhPrE,16081
tokamax/_src/ops/causal_conv1d_gated_delta_rule/test_base.py,sha256=JKWe5DIjGQah5jLBDAZajeveXq9ZXpCpkYsEmq5FdwU,29020
tokamax/_src/ops/causal_conv1d_gated_delta_rule/tiling.py,sha256=iSuOkH88MGY-occl54CAdRoHC77unWQMaGRM68AUCIE,15948
tokamax/_src/ops/causal_conv1d_gated_delta_rule/vmem_ldst.py,sha256=XXvgMYqVinNjI7ztciZHGlXasQmxcUgnhFN7CGz1USY,12337
tokamax/_src/ops/causal_conv1d_gated_delta_rule/wrapper.py,sha256=Wk-5oTs8vGgnU81jMiJHljp3J9j_gD5EnC_XJ4ULR5w,27858
tokamax/_src/ops/csa_gather/__init__.py,sha256=Fyu1RjA1-ZGi6EbDO92VOs7ahZ6HNBDlSh6cgcJEteY,690
tokamax/_src/ops/csa_gather/api.py,sha256=iqXjJi22BnPvqLJj6Wdo5cTlQ6rby93Lalqz2eg_Yug,3831
tokamax/_src/ops/csa_gather/api_test.py,sha256=jwvesAu85shG0cv8qyRKMfS8Wteo1LOwGrJiq_c6Wuo,3360
tokamax/_src/ops/csa_gather/arg_specs.py,sha256=acF3fKMMnzbi9tYh_JHxWcqwnbQohPVCAZ_pvGBQRVU,2281
tokamax/_src/ops/csa_gather/base.py,sha256=HAyK9ZVl6GFLheDEfBmtXP9-dVtvHHeVuBBoI59YeC0,3157
tokamax/_src/ops/csa_gather/base_test.py,sha256=6Si6mLiCblpINkA1EzC6zcK81uyOfJFpfsT46sCbJ0s,3899
tokamax/_src/ops/csa_gather/csa_gather_test.py,sha256=XnCAT57PHLML6PNrd_MbpvnRiXGy7n8IdgCYXDQ-KtU,5217
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu.py,sha256=9kvaoE_hAogXAQSVYdpeKygOt1ndu7pboZuwaLQPSko,5636
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu_kernel.py,sha256=Jf8ZjUBH5KQLKLlFVFsnqdVoFxkiUnTdKMJju2NfkQA,13063
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu_kernel_test.py,sha256=qsaImrnUIbss9sEOAiIKRgV8Dv3dpH_ohboBjTxlKSs,7206
tokamax/_src/ops/csa_gather/pallas_mosaic_tpu_test.py,sha256=qlmcq7OjfBxNCIpvpJ4vhHKAQKSVHsgu9pPW8qlxPx4,2575
tokamax/_src/ops/csa_gather/reference.py,sha256=quGbL2JVqImrWu9SO3A7zXUukj710HWtCvglceChTOY,3565
tokamax/_src/ops/csa_gather/test_base.py,sha256=2grsGGqZu5imrIRXuqJci5DUetUlvAGkHZUcyc4ur2Y,3588
tokamax/_src/ops/deepseek_rope/__init__.py,sha256=OW12E6mnjE0S4irbGbdaMcRo0_Ip9SCR_b5-JkHyE_E,696
tokamax/_src/ops/deepseek_rope/api.py,sha256=YAunda0BHiGBB5fr8DnnEj_y-rE4y-nir9GFa9LmKl4,4991
tokamax/_src/ops/deepseek_rope/api_test.py,sha256=CYvYR1xfneZT-fK5J-vQ9213GgLRYYsCzPwGXg1z9mw,4878
tokamax/_src/ops/deepseek_rope/arg_specs.py,sha256=sxFtOv2MvUUq3L3Eo-8MVfNFHlvmeLp2pi4wcG_pHzQ,2754
tokamax/_src/ops/deepseek_rope/base.py,sha256=-ILWvR8w-wAprsnoxXo4piiWNns4FwXzRWmuZ5ciekA,7487
tokamax/_src/ops/deepseek_rope/base_test.py,sha256=OnWkDjMRaDb5QB71yFOzYUHMyuIwdWMZd0xs-9Hu4Ns,3962
tokamax/_src/ops/deepseek_rope/deepseek_rope_test.py,sha256=OT6UVbIejJiQSJOMVOqjNaFDdWBkEwtORLIIKtnvx78,9269
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu.py,sha256=WTIDEhFC51lWTEx1NvVEpSA_iHwK05cs2jMGhrTx6G0,8613
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu_kernel.py,sha256=HaYaMU8Vz4C5OUTKl7AYYE7FyYicbGVYQo84LXVN62Q,20888
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu_kernel_test.py,sha256=IkPvPVdOFWEMKzC3JNHpI3Dei6MXVQE3hVAdA3IPdfs,10303
tokamax/_src/ops/deepseek_rope/pallas_mosaic_tpu_test.py,sha256=goNdmjYM4DIdnLfVOKLJaL6NGAGt6SkLtXf6HDhBhDE,5999
tokamax/_src/ops/deepseek_rope/reference.py,sha256=dgnvIr-8eTaoDiHC89fk2wicCzgNr6IpZGFi9J2_Y0E,5408
tokamax/_src/ops/deepseek_rope/test_base.py,sha256=wEakihTX72KwjGciMa-nneg9bUXwpzkgCud0MqteOss,6369
tokamax/_src/ops/dsv4_compressor/__init__.py,sha256=y_CpGGJ-P_GduSsHNwM6P3svzalHDduN9n3XqKhpCKY,698
tokamax/_src/ops/dsv4_compressor/api.py,sha256=7xMmKrqJnu6r9OUwwHwLR3MqWFyT5g_JcH3TDA1w6XI,5676
tokamax/_src/ops/dsv4_compressor/api_test.py,sha256=0wd9gg7Afa-gMtyasaFzbMmCgUk163zSMOiKQEXsi0s,2557
tokamax/_src/ops/dsv4_compressor/arg_specs.py,sha256=mRDsHqjeuCja2Dm0EGylnDpt_5hkJmq0ZexdsdeCUus,9087
tokamax/_src/ops/dsv4_compressor/base.py,sha256=m27nK6xHqqyUKpWjIxcyo6by0oyt-BZdfHOGHgO2N8E,10250
tokamax/_src/ops/dsv4_compressor/base_test.py,sha256=eAF1o67_NZ_dJeVjc2yycCipbewX5WyusWwwLpKDuiQ,2735
tokamax/_src/ops/dsv4_compressor/buffered_ref.py,sha256=-3pJ0Xd3-W_W4_1WcABaNZ_GvTArYOMH4zZ0nW11mjg,17387
tokamax/_src/ops/dsv4_compressor/compute.py,sha256=NhmWBU0JtWc_DeYpJWSa0kcTeETMwLI4YqOPRzZQIIk,23518
tokamax/_src/ops/dsv4_compressor/config.py,sha256=Ful-ktNWn6GyJ7wBaOhtjrCjxd9twZOmcYDJs4PWiyk,20840
tokamax/_src/ops/dsv4_compressor/csa_cache_layout.py,sha256=txVjf-kX7OUttuDDBJOsa5xX8iZFFpRvdvGjcNLlEW0,4832
tokamax/_src/ops/dsv4_compressor/dsv4_compressor_test.py,sha256=zPswKVzLrosKwGkscrRgvDGgXYVbBySF9Hj0QWOi058,29911
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu.py,sha256=aakvrjAcPbUaY_Iw5LiXzP0XESboQuTolCeDx0_RwH0,6290
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu_kernel.py,sha256=PdpBMsASl2BCZvRjEI0sRybWRUwyBraj2A5Heb1d4L4,20540
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu_kernel_test.py,sha256=dWLA1RbRZgOYnz19VCQiASwl6hv9jeCQTNYHuhZnrwg,21337
tokamax/_src/ops/dsv4_compressor/pallas_mosaic_tpu_test.py,sha256=Fna0x2XxfQd7YGHMipAuin1qkKjMIIu_h9jz80W-jQY,3974
tokamax/_src/ops/dsv4_compressor/reference.py,sha256=cFP13n55wjZsk_Q1XixPxRMDF93URMvqEp8XPvT9T1w,22146
tokamax/_src/ops/dsv4_compressor/test_base.py,sha256=tr1Pd1o7FoR3pSTCwLA1FDrz8O9rZvWfhixVOaXZ-Ps,18411
tokamax/_src/ops/experimental/dedup_moe/__init__.py,sha256=oVsB5t5EJ1-GpTqf9KMUYD_GP3-MNozjwfiSQ6bUADk,2565
tokamax/_src/ops/experimental/dedup_moe/common/arg.py,sha256=g1xbh-0VNgazYFYb9SFZjOCZ2NJuLkfBLUdJpLkMVzA,2737
tokamax/_src/ops/experimental/dedup_moe/dedup_plan/dedup_plan.py,sha256=ZCHtalV2690PBddCNn_SX1u1vdUBi-P3bYIYadswgx4,48505
tokamax/_src/ops/experimental/dedup_moe/dedup_plan/topology.py,sha256=4_kc07x8xw6bwqEg6qp8L-6C_ICA5-MVfrtkyGvOAFQ,11192
tokamax/_src/ops/experimental/dedup_moe/dispatch/source_plan.py,sha256=n3iH_Fyt3tDqepS5oYp9EnUpXPh45_TstEj7Facyoso,21369
tokamax/_src/ops/experimental/dedup_moe/dispatch/tc_sender.py,sha256=714BIq16ggVq3FlN7_jsmaiDuigzy4hBttW5dnpneI8,19176
tokamax/_src/ops/experimental/dedup_moe/dispatch/tg_assemble.py,sha256=-9SWA6AFZKdTe2Wg2PpoxIr7cnfi2uEmOvRAYO5aNBA,14689
tokamax/_src/ops/experimental/dedup_moe/dispatch/wave_dispatch.py,sha256=92r-cEbwxIuoUhH-g_aajpytNUMLDI2RGHpIkLbdJSA,27126
tokamax/_src/ops/experimental/dedup_moe/fused_kernel/config.py,sha256=mrBZ2h6qKuXYuXcg6Rqk1zmPPOGTh90C8isW68sFx84,14988
tokamax/_src/ops/experimental/dedup_moe/fused_kernel/kernel.py,sha256=NzD44CHmeThcmegTEDCzVQQAJ4EQFTFKW_yOsMT-k4Q,15872
tokamax/_src/ops/experimental/dedup_moe/fused_kernel/memory_ref.py,sha256=ZUq2RaK1l0-10N92lIALapmNV6GzSaJVELvM9fUb8aQ,12574
tokamax/_src/ops/experimental/dedup_moe/fused_kernel/pipeline.py,sha256=oKE8LhrCHOfU_bVLQB9GRe_PIBfgosTHaHFHcnusRSU,34118
tokamax/_src/ops/experimental/dedup_moe/fused_kernel/plan.py,sha256=f5BVQJLSiAINzqJAeeh5DpWDcXF7ULP2hTqYpo8QMI0,13134
tokamax/_src/ops/experimental/dedup_moe/io_relayout/input_pack.py,sha256=2RbOjimltMRb0-9DejW9NElKxwp3FSINB9AZ413hq1Q,3168
tokamax/_src/ops/experimental/dedup_moe/io_relayout/output_reduce.py,sha256=k5NZgLltgHBayw2VR3m7wrkgSi-SNHHPorjioGmD5pw,18098
tokamax/_src/ops/experimental/dedup_moe/layer/config.py,sha256=JCtC-yF-ZV3KMskkFSnDUbxv9XexuFgR3UB5sHoUF8M,9729
tokamax/_src/ops/experimental/dedup_moe/layer/mode_registry.py,sha256=i_G1cYqACh3Mjjv4_XlolIpnjAxY4LnLnT1dVp4Px_s,4762
tokamax/_src/ops/experimental/dedup_moe/layer/modular.py,sha256=rcTOtHoPH4q27NMCQYAG36vta-70tVwdNDQyXkw5t0o,9149
tokamax/_src/ops/experimental/dedup_moe/layer/monolithic.py,sha256=b4sYgAivZJO5PsuMULVQrF-LrGv8-pV7VdcmOmC8vPI,37670
tokamax/_src/ops/experimental/dedup_moe/layer/plan.py,sha256=y5SKlYnBhnV2oD5vYOOJiL45caI9LUCemAHAv7CCvh4,11759
tokamax/_src/ops/experimental/dedup_moe/test/common/hw_util.py,sha256=o8jc12EukLRY3bKi-oycKZQF2oQzgBnknu--ao1XEP4,9220
tokamax/_src/ops/experimental/dedup_moe/test/common/interpret_util.py,sha256=mvg7t_8LAi2EPQR-L2DslNNXGgNT8z5yk3AjQPBomqk,16386
tokamax/_src/ops/experimental/dedup_moe/test/common/moe_inputs.py,sha256=dwv5LrX33jyV9aD9FZuCrNUO_ITeBwWNAt6hoapQ3T4,10396
tokamax/_src/ops/experimental/dedup_moe/test/common/phase1_ref.py,sha256=aLSUmPOvKus9ZLvwoS-YSvqtlAepKEAPZNjdlYI1qwg,5812
tokamax/_src/ops/experimental/dedup_moe/test/common/reference.py,sha256=VXAGWzv4L9y53HAuHrMpxkFMwvnUCIe3fyyKLrfUwHA,13157
tokamax/_src/ops/experimental/dedup_moe/test/dedup_plan/dedup_plan_test.py,sha256=vVhhEeGXryyieO8JUy4XEsOtlfhLlSGGPuj-CMZvnnI,49557
tokamax/_src/ops/experimental/dedup_moe/test/dedup_plan/source_plan_test.py,sha256=yHH9qIeVeLOzmMm3E831_4cpZUgTPd7A_ocPk9Ug1bI,20769
tokamax/_src/ops/experimental/dedup_moe/test/dedup_plan/topology_test.py,sha256=DOz3f32P0IKEjkskAxgLuWiT00B1L8bjBBve6F3qYpI,9699
tokamax/_src/ops/experimental/dedup_moe/test/dispatch/wave_dispatch_test.py,sha256=z5Y5Kah3Q3_GtkMMDzS9cZnBIdfuaVrrJqH570JImmI,16786
tokamax/_src/ops/experimental/dedup_moe/test/dispatch/wave_order_test.py,sha256=TKEDU6RuCZTP1scck8M1j9t5kkRyPoqNUrriSil0KTs,13597
tokamax/_src/ops/experimental/dedup_moe/test/layer/dedup_moe_test.py,sha256=lBC6rFIN1Bmt0qpBqgEqCTtJl-3SY26U-Geb0UhPTbs,22174
tokamax/_src/ops/experimental/dedup_moe/test/layer/layer_interpret_test.py,sha256=BlNFV6YTBTUmBNMEIT33q44Bdgq2Kv2eRhX-pUeBBiE,11321
tokamax/_src/ops/experimental/dedup_moe/test/layer/mode_registry_test.py,sha256=NfUn965_XsDZ-R23ORYujiztbCQ7SzAOLGOc16T6CrM,5005
tokamax/_src/ops/experimental/dedup_moe/test/layer/modular_test.py,sha256=d6xhhF-Iltt0FPxljw5ritH40kHIERI--BncNKtuG2g,12337
tokamax/_src/ops/experimental/dedup_moe/test/layer/padding_test.py,sha256=0mPl70cwCLYyf5jHheevXwt5S2fsgSTbeKsxBBN6irE,12706
tokamax/_src/ops/experimental/dedup_moe/test/layer/plan_test.py,sha256=pGGdUwkKo_hR6s6sgU2DGLc2nmFCqaqjrnFKykWvY7Q,16844
tokamax/_src/ops/experimental/dedup_moe/test/layer/tpu_correctness_test.py,sha256=OgcLaXHy_nI8HaFcy60_0cNcFpjmWKAe97pkzfrGO_Q,28084
tokamax/_src/ops/experimental/flex_attention/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/experimental/flex_attention/api.py,sha256=oFr6b3izddfYyQJ33r0i2-CmSa5zWEWOOEWvS8fvm-4,6025
tokamax/_src/ops/experimental/flex_attention/api_test.py,sha256=68214QUDUqyJgKAaqlwpEfE9ATqVIoux6MafZwttQpU,9188
tokamax/_src/ops/experimental/flex_attention/arg_specs.py,sha256=DECEEmyM-WzMoD7z8nd48JBDKUK9zyhY2InIT76AVaU,820
tokamax/_src/ops/experimental/flex_attention/base.py,sha256=noxY4QSQ1o3hg4dTB-YH6FF9j2HgTD6Cmu6dHO9EUQI,12340
tokamax/_src/ops/experimental/flex_attention/base_test.py,sha256=gUx-MOXnycUOHnHSitSZWqtP8ZBaclTIf_cWpNpcPNk,1514
tokamax/_src/ops/experimental/flex_attention/example_masks.py,sha256=BD7tuu-6cSh6kWSCd2Rhz4DL0uaYcOu8VD0nV6f9134,2299
tokamax/_src/ops/experimental/flex_attention/example_score_mods.py,sha256=10guykqny3SkHhT1euaXE14ZfpZ74sfNE3Z7b0hU-Kk,1328
tokamax/_src/ops/experimental/flex_attention/test_base.py,sha256=RmAVNi29wphSWCX84Miv6sdnAtlJeNYTwpUsBn_catY,2865
tokamax/_src/ops/experimental/flex_attention/wrapper.py,sha256=_op2yZKYZVTwmA79mLvYmreAY1ennHKf1prAPkFUU8Y,3551
tokamax/_src/ops/experimental/flex_attention/wrapper_test_base.py,sha256=Xs1hf1AtQJ-529oxyC7wlzm6lvcbaO-2YGP850os6vI,1361
tokamax/_src/ops/experimental/fused_fp8_matmul/__init__.py,sha256=TGvQ6johv4CESH1L9Syn6UwXDTtp3DTXDLAfvvb6pU0,696
tokamax/_src/ops/experimental/fused_fp8_matmul/base.py,sha256=HUzmmDZyCgUkmqjcCWibawIv3QxaiVCpKQvFEUZ29Hk,11004
tokamax/_src/ops/experimental/fused_fp8_matmul/base_test.py,sha256=_YInpZgkLNcTXiPR5jMUmKfjlQ05sWR9siG2YjugG0c,5932
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu.py,sha256=3p8-96EwAmfpTswbQhaZC_oUWGZ9iogdePcjQH7AcPk,4784
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_dlhs_kernel.py,sha256=HgQYLnWtFOSmAzvHe345fardUQRBx1eK0bKyiWreshA,8982
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_dlhs_kernel_test.py,sha256=Nuf5mJ-mewRfXBNGzB8MZns6EekGyisFIWWisN41wfY,5814
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_kernel.py,sha256=Ez0lL_5y5-y3AefydKDoS5SSS4Ch6BbZo0n-Kb1R6as,18401
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_test.py,sha256=o82G749DkV8mhYfx93lfJsZfvasTEnq-MJJsULm4f_0,5148
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_vjp.py,sha256=porrpRpcmor3pMrOdzDAWnokEiBeYN-EgdWi9wMzoVk,8514
tokamax/_src/ops/experimental/fused_fp8_matmul/pallas_mosaic_tpu_vjp_test.py,sha256=sadOD1QBdiptJ_2CahXHe3ELqq4iNLUx66nXWV1XF7k,7678
tokamax/_src/ops/experimental/fused_fp8_matmul/test_base.py,sha256=X_KiGAMXaQXpmZKs6zBaVVeZKKSbRGE5r3M6EbqQUi8,10269
tokamax/_src/ops/experimental/fused_moe/__init__.py,sha256=81z-IbB9dS9jFjt3kyQ4AFuMcGWoY5QQAumD5Vx86UA,2711
tokamax/_src/ops/experimental/fused_moe/api.py,sha256=fEiwMgwySKaEP-M1xllbnGnmEzXiR-XALB-9K-Ibmp8,4653
tokamax/_src/ops/experimental/fused_moe/api_test.py,sha256=TpXR1zmHW24PzJ4OS9TDTKUCUdqaSf-BXikWbeulGdQ,2816
tokamax/_src/ops/experimental/fused_moe/arg_specs.py,sha256=8nEZDoVn_yAf1p3f8DFZwExISohV1yCYPAByNLVYivo,5654
tokamax/_src/ops/experimental/fused_moe/base.py,sha256=ZU2rkQyMZs2yyVva6D9RPpx8geu56pW1FDAUr0mGdNY,2847
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=3pZkfYcPpInM2BfEMxxWrMgpCRjC4oGAXo5-TdwKFkk,107159
tokamax/_src/ops/experimental/fused_moe/layer.py,sha256=-lFheT9mSfbPqMXy8mS_F4ilaT6EOb1fnZI5RkmfRq0,32694
tokamax/_src/ops/experimental/fused_moe/pallas_mosaic_tpu.py,sha256=sBb1si4ohXeQ4LouuP7CZHCBcbr_uJFC1uicLYSs9Jc,3491
tokamax/_src/ops/experimental/fused_moe/pallas_mosaic_tpu_test.py,sha256=dWoViwa0UCqCZUSS0W5279DzJJvy153u-nXPb5RjZA0,2459
tokamax/_src/ops/experimental/fused_moe/reference.py,sha256=_Ui5yeoV_ZHRhc-NeGco6qBhrzpeTQBNolRdz_Rl3lQ,7219
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_decode/__init__.py,sha256=NRAP_YV3o7vCuQ0p6B3MmcoEC2IMR-fIAmE1GnH92bo,737
tokamax/_src/ops/experimental/fused_moe_decode/api.py,sha256=4ae4qBptPIQRaC8kHJkJ0EzphlysDKBBr-cB6HrdeZ0,5867
tokamax/_src/ops/experimental/fused_moe_decode/api_test.py,sha256=X77DZ1N1f_lDDQ83Dx3i36_MdqbuzuaQCPTWXjhi2qM,4832
tokamax/_src/ops/experimental/fused_moe_decode/base.py,sha256=x7Na-AAUYcqRVnDooW3WLgfRtuK9hIDIrAw5JEQKs3w,7746
tokamax/_src/ops/experimental/fused_moe_decode/base_test.py,sha256=oAUJ4nXCCCX5hKLq_XH5B5KV25TvRCuqnxujhaT1Nk4,5797
tokamax/_src/ops/experimental/fused_moe_decode/fused_moe_decode_kernel.py,sha256=XAOE9SSddNj-Gf6D-Ft9w60e9VPwh3ExhOUMUuRlryU,29662
tokamax/_src/ops/experimental/fused_moe_decode/pallas_mosaic_tpu.py,sha256=hq4gmLm0_L0VN9H_DErknUfy5VVjRMg3QeEUlZjDBu0,6350
tokamax/_src/ops/experimental/fused_moe_decode/pallas_mosaic_tpu_test.py,sha256=SQwjSnfzDd5PAUCUIIOSDminpOEyPUBQHmoDx_RmKLo,9535
tokamax/_src/ops/experimental/fused_moe_decode/test_fused_moe_decode_kernel.py,sha256=EOe4e191g7xUF7-4pqbMHj06iDarsT8JCsVHV9ZGVUE,12210
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/gdn2/__init__.py,sha256=oS3rVXUl0FaPIGN_DeZbJYem2qqyAz9ZGF7y44cBrtY,727
tokamax/_src/ops/experimental/gdn2/api.py,sha256=X-M8RosNhT0YMP3fiZ02buyxRU7PWSaNZUfAgATO8D8,4541
tokamax/_src/ops/experimental/gdn2/arg_specs.py,sha256=4nKyNbNX-6dtt-gBbv-DrDYBr15vzfza8lFv1pKyYIY,2293
tokamax/_src/ops/experimental/gdn2/base.py,sha256=B85nOW15Czt06jTvGLPzapG9rI4jb7xpTvLvRoQRGmQ,9452
tokamax/_src/ops/experimental/gdn2/base_test.py,sha256=sd3roQFM67QBH7ZOOsihPAHssADgdPpCEXMqtxDvCh4,3055
tokamax/_src/ops/experimental/gdn2/pallas_mosaic_tpu.py,sha256=A_EU28q7bJo1SUVOy-yrs2PbwBJkqbeuXok2xVllEvE,6295
tokamax/_src/ops/experimental/gdn2/pallas_mosaic_tpu_bwd_kernel.py,sha256=4iMINH3pIIm1Xip5JDOsmtAxikWZgW6fIHU29LfWMmg,14699
tokamax/_src/ops/experimental/gdn2/pallas_mosaic_tpu_kernel.py,sha256=hrwE2Aa-Z01GARUr0vGxVW4zK00Qg3cAePIhw9WMsFs,10497
tokamax/_src/ops/experimental/gdn2/pallas_mosaic_tpu_test.py,sha256=fjKAlgSts6U5LGGEw8nfEWC9QxWy7kuSfmtREHNu8R8,6715
tokamax/_src/ops/experimental/mhc_lite/__init__.py,sha256=OtXRXu-czwR1bk_ECYH3uJ5bcOomEIkjsyLFNJ4QZQo,655
tokamax/_src/ops/experimental/mhc_lite/api.py,sha256=Yxl99SoqCe26B6uhroHtyfNspOE9Zl81_o_qv-com3k,8326
tokamax/_src/ops/experimental/mhc_lite/api_test.py,sha256=OOR6DETeENpXh6us4Hv8J3-r3gXhgQewunhwbB_vmIU,21625
tokamax/_src/ops/experimental/mhc_lite/common.py,sha256=2PKjNZGQqgkqT5QWVPA6O143aHGLSXLIiBRMoagpJA4,9900
tokamax/_src/ops/experimental/mhc_lite/hc_xla.py,sha256=Py7o13h0If4DI5dAeDxRTPRRPhRCpV9WrBD4rJNXFfE,5219
tokamax/_src/ops/experimental/mhc_lite/mhc_benchmark.py,sha256=U1tT4Xpzh7WNQFzf4PbTZgX5To1oRGKlhj2N2zDgppw,13564
tokamax/_src/ops/experimental/mhc_lite/mhc_kernels_bwd.py,sha256=ADRH3LMg6_P9upRtbm_Vb_YdV7jayV0yhDSqLs_0dFk,32836
tokamax/_src/ops/experimental/mhc_lite/mhc_kernels_fwd.py,sha256=UF1sAMa-jbNYyWOZLSqiVfiAgoLXK6Es-gMG93yQzJU,16998
tokamax/_src/ops/experimental/mhc_lite/mhc_test.py,sha256=k_bkPkWHv_k4RuPo2A4r0NlJ1o5H0zSsWdrYvrYDLQs,13683
tokamax/_src/ops/experimental/mhc_lite/mhc_xla.py,sha256=r_XWRla3uxH7p4jwhjzQxF6BrKdU3P1lYpwmMoIsE-Y,5243
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=dT_U7acypST31ovQkDXum3rjduuZDSc0ZWjT-Q4xzbQ,10323
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=1p3EZDWU4f6XbXRj0WnogDezkTtrSdT9Z1JQDKEO0vw,11393
tokamax/_src/ops/experimental/mla/test_base.py,sha256=I1_U51YxYK5Z_KZFYM--25B_HFSjGcj85PamMZhsa4w,32811
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=VBjAPlsI3RC-3IkNPUreFXLKkLuwWdGvTud8IDjkgow,104662
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=QtkN1-arERUoj1yrNtsx-_Hoy8CcCXOjL42qikIhHxM,6659
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/mla/v3/__init__.py,sha256=dHrMCOg1HWExWWczemCH5TSc6sXJo8qz6aNk9PhtFWE,2410
tokamax/_src/ops/experimental/mla/v3/bref_override.py,sha256=RRnok53xIXIG9pEvuWfNsyYT-1LNdkAbBoqhlWnVE7Y,21285
tokamax/_src/ops/experimental/mla/v3/configs.py,sha256=QDB3ANKb-gztxnmGZwKt_D2ieo2bcnQH_N9haJFISW0,19227
tokamax/_src/ops/experimental/mla/v3/flash_attention.py,sha256=WVp3XeIkKq9Hp-93Ly2y5uf_FvEP1r9JgR055eP0CKE,20112
tokamax/_src/ops/experimental/mla/v3/flash_attention_test.py,sha256=7apJL7SWuhLmCab7XlDLaFyDrutYpkzePmYwSpiDGj8,22182
tokamax/_src/ops/experimental/mla/v3/kernel.py,sha256=TZaQiCZ5K0ItjN7Gu1HRhBHjfKPxLfeZwtMaHTP2ISE,30632
tokamax/_src/ops/experimental/mla/v3/mla_kernel_v3_test.py,sha256=dsbEFIq2briaDqkP-N4jyjpZ7JtcQM4Z4l4i2MTQPL4,10668
tokamax/_src/ops/experimental/mla/v3/mla_wrapper.py,sha256=YMmmGtsIjtlNiOOyLgnrSaFmyISGsMb4Vq4oOhqM5mw,11807
tokamax/_src/ops/experimental/mla/v3/schedule.py,sha256=zR8yEe8kSP0_tJOP9UbEUrNXI4twL5tAnxh5vyXLQvk,23192
tokamax/_src/ops/experimental/mla/v3/schedule_test.py,sha256=Ve4kmEomitQDh2FF-5xJqj9U8eFVG7awJJ_QfaE91H0,15552
tokamax/_src/ops/experimental/mla/v3/stitch_utils.py,sha256=QcU5nm8jUxpI-PeM8gakRQEqW4p4SSdrp4uLcTdR4_s,4848
tokamax/_src/ops/experimental/mla/v3/utils.py,sha256=cP4VHvrORTMzjeeNGtgiA_LsKCWwltsYTIPxmh4Kdjo,4750
tokamax/_src/ops/experimental/mla_swa/__init__.py,sha256=wED8qQf1icxD9jVjA_6kQAxcPLupoxkNpmbyMiEAisE,724
tokamax/_src/ops/experimental/mla_swa/api.py,sha256=ZiDE4UJvG5LNOZxSRVc4Tbna_GCw1frY9IZTgw-VKq8,4717
tokamax/_src/ops/experimental/mla_swa/api_test.py,sha256=QCnqmBOlT5Mnptmji3LRSsQllSk4ZsclvNRYNjJ3gB8,2232
tokamax/_src/ops/experimental/mla_swa/base.py,sha256=-9qNXPx4iYQm3KeZGkkaVcGWoRvd_CTsIVp10HoLD_A,5689
tokamax/_src/ops/experimental/mla_swa/base_test.py,sha256=HJMsLEOylD_hhMhVrgKGbFoHBPM1LAFT60dcNJjtu_c,2121
tokamax/_src/ops/experimental/mla_swa/pallas_mosaic_tpu.py,sha256=GgNC7sapOx72eqMkMx8nFxEcO5qFoETnpmqbONgnZn4,4688
tokamax/_src/ops/experimental/mla_swa/pallas_mosaic_tpu_kernel.py,sha256=PPsCte9Uv_ddzLtPqeiRtyS43m6C4bU4Rvpxgb8UNow,65086
tokamax/_src/ops/experimental/mla_swa/pallas_mosaic_tpu_kernel_test.py,sha256=OZZRWfuBZGO-358O801t-y8eXc9M72pysIMggTZAbhI,24743
tokamax/_src/ops/experimental/mla_swa/pallas_mosaic_tpu_test.py,sha256=C-amUP3WKb3vqZRTT2bJPBP6s5vSvjxiWJb4n8Zq6lI,2171
tokamax/_src/ops/experimental/mla_swa/reference.py,sha256=bNHC9E_tKLbhzQXHJx29m3LFmlkxUeIo4X-2t3Q4BRM,9035
tokamax/_src/ops/experimental/mla_swa/test_base.py,sha256=v_f7bTpnsR2MhlbzqVljbc7J92GdDdyzq34hJjYrYxE,7761
tokamax/_src/ops/experimental/moba/__init__.py,sha256=JUrmkl9d4ImNto2fGK92TuodIhibpUSzsHDL8pivigA,735
tokamax/_src/ops/experimental/moba/api.py,sha256=3M1QLtyj-QJF4p85erpq587zR_iLFTGEz2gBq61gTlc,3479
tokamax/_src/ops/experimental/moba/arg_specs.py,sha256=43Ft9kWu7rZtJgLL5LdKtro2jDPhqhrgX6jFspphWTA,2186
tokamax/_src/ops/experimental/moba/base.py,sha256=syzIzfF7GCejPg05hK16czbpWEdwHX5z-dCc2z1C-ds,9205
tokamax/_src/ops/experimental/moba/base_test.py,sha256=oOjQjUJvpveoYwsEQy3VYOjR7iE0FfzU3_BU8qJcQss,3380
tokamax/_src/ops/experimental/moba/pallas_mosaic_tpu.py,sha256=MEfSUI518eiqavTYkj-KeUKYZ0gu0LGRvt2x3hIqM1I,5230
tokamax/_src/ops/experimental/moba/pallas_mosaic_tpu_kernel.py,sha256=NYwXGnxcpEMMXD_LKfX2dLQ3akEhENDBFGy4Us2hG1Y,7559
tokamax/_src/ops/experimental/moba/pallas_mosaic_tpu_test.py,sha256=ygO5deeyY7sRptRTEGhUqxHuEuOsy6TdKMmzwx0edtQ,3299
tokamax/_src/ops/experimental/nsa/__init__.py,sha256=T43ospuoTI5PggO0UlzPMMd1l2mdZ181nQ2vzACE804,732
tokamax/_src/ops/experimental/nsa/api.py,sha256=rV6gqHHJFOeBiOvW_w1Hcmi4-5UNdc1j2-h4fik9i8M,4232
tokamax/_src/ops/experimental/nsa/arg_specs.py,sha256=EL-WLsL1_aeeRY8EUbVkWNKbMQaRgcyVy1lSP56IkF8,2437
tokamax/_src/ops/experimental/nsa/base.py,sha256=6xRuihXlimQ5r5RFqM0ya5VheAfopPZrbU05Vq7fy7w,12255
tokamax/_src/ops/experimental/nsa/base_test.py,sha256=iJsp2e34Cr1qZ4n2-Y4RXR_vHE_mfDoAHUEFVsV0qzk,5858
tokamax/_src/ops/experimental/nsa/pallas_mosaic_tpu.py,sha256=3ntrNmmnBkmWR1Lqim4gAEzKpDa-cvXs-kjn_kLi_Vg,4815
tokamax/_src/ops/experimental/nsa/pallas_mosaic_tpu_kernel.py,sha256=jx-WtfHOlQt0R4VVt_nuHR925_r8iYJoDR7bebsc9vo,15197
tokamax/_src/ops/experimental/nsa/pallas_mosaic_tpu_test.py,sha256=rloHS-Wr9kHL0ebIR8MEAUElX9ByIZfC27KVmY6ZjQI,4640
tokamax/_src/ops/experimental/raven/__init__.py,sha256=vgQ2dwCu6CCQCk8XK-HrmCZaNbk36V4dcIqcQndy2hI,734
tokamax/_src/ops/experimental/raven/api.py,sha256=bd-pZj4qvDLrF7QS8_eLceCuXRl61dIPnDTifpxiYVI,4360
tokamax/_src/ops/experimental/raven/arg_specs.py,sha256=Q0XeDUD5U0asBOq8Qy7qUHa4i91ZT4VPIo4PqMxNB9Y,2426
tokamax/_src/ops/experimental/raven/base.py,sha256=2omjvtogd-EXbz_PWFcucc3MMxVjbI9Ay2C8Ax8URh4,15712
tokamax/_src/ops/experimental/raven/base_test.py,sha256=f29B8k3y9hN0XdqqUjG6mkrVn9jkBW1cJtxyoZtNTRI,5379
tokamax/_src/ops/experimental/raven/pallas_mosaic_tpu.py,sha256=K8TfDU2tjixPVQk7bU8pzgTGTnkfX8yRoDeL-JBOFOU,6173
tokamax/_src/ops/experimental/raven/pallas_mosaic_tpu_bwd_kernel.py,sha256=11fp3BIX5eNIF1f3RGfbyuHxNcg8QCeAB7xbf_Ex1Ww,22655
tokamax/_src/ops/experimental/raven/pallas_mosaic_tpu_kernel.py,sha256=F6xaEIWDMt1FQ5cKp2AI4qtLaEnSfYmHUaDfhO4QRM8,15353
tokamax/_src/ops/experimental/raven/pallas_mosaic_tpu_test.py,sha256=vUyIRbIJ8h3tJRuMJpoHP58Kw8UyRTRiRTbCAQPksZ4,14267
tokamax/_src/ops/experimental/routing_moe/common/arg.py,sha256=RlKjsVbzzwFNyAyJ8UOjE_u8Prhwyy3CNYTyZRXgb3M,5560
tokamax/_src/ops/experimental/routing_moe/common/dataclass_util.py,sha256=CMMTO225QmjaxLiArbNqm9JHw2XaDhwzXS4gyoH50cU,3358
tokamax/_src/ops/experimental/routing_moe/common/layout_util.py,sha256=C5FPb-344uzKi-BqP1QC_AbyRQjd7TrKXvxMdWgTh4s,7168
tokamax/_src/ops/experimental/routing_moe/common/out_format.py,sha256=7T4gk174pu64OS9vNggCCpaSh00_QbcF6a_B7GrG2p4,6936
tokamax/_src/ops/experimental/routing_moe/common/quant_util.py,sha256=VGBvGc0wLnIgqgK1mPDt9IRWy_YA3spb2OddG4Mi7CI,2788
tokamax/_src/ops/experimental/routing_moe/common/spec.py,sha256=cj1QNxl0O4EvnPC-zVwR4Quf2lni3Izm3Jca_MWnjDg,7493
tokamax/_src/ops/experimental/routing_moe/gmm/compute.py,sha256=fT-8KHUEPZVLVEA9UF0VAnh9wZ4jQINFreRPt6YlWvg,7682
tokamax/_src/ops/experimental/routing_moe/gmm/config.py,sha256=nHs2tL07jbWiL-6nrWusr-HrKSPIu0WghgaUFnzZcsw,12636
tokamax/_src/ops/experimental/routing_moe/gmm/kernel.py,sha256=rVO9e62aVnR9wC9WWkgBSJDDFnqm8L_DducLPYTdG0k,9086
tokamax/_src/ops/experimental/routing_moe/gmm/memory_ref.py,sha256=48KZCSDrBqvOtxTP9JwJlQQwjS4m9mo2cJhqkbk049E,42342
tokamax/_src/ops/experimental/routing_moe/gmm/metadata.py,sha256=kWow3ZRSexSwAPS94UMRMZR4KwwSNprZzwwdH5TTcho,9176
tokamax/_src/ops/experimental/routing_moe/gmm/pipeline.py,sha256=Z71SgdJ3ZUpfQ4XKh9gYfyZjraIcyeVXV5jI-YywPqE,9104
tokamax/_src/ops/experimental/routing_moe/gmm/plan.py,sha256=vZDQX5tflfSl7h1IynTjSgzijcVU2sRks00N0nzqFP0,25370
tokamax/_src/ops/experimental/routing_moe/gmm/spec.py,sha256=3MkOmpnINZ-XXY2elcon_vXjiQ7290OZJ9IsZ08V16M,4821
tokamax/_src/ops/experimental/routing_moe/io_relayout/input_pack.py,sha256=dcakZ4A3jBF712b0HPprGn8OiKX3nHQHAHkafVEO49I,1893
tokamax/_src/ops/experimental/routing_moe/io_relayout/output_reduce.py,sha256=Nhv1_Ghfk3CjR1M_pRNJrLm7c1ZB-6A10hRyuUt1TaY,19924
tokamax/_src/ops/experimental/routing_moe/layer/config.py,sha256=2gyiVONq6c5AG1OZ862IOCCKmMytP9nDwncZEOOZPEk,5459
tokamax/_src/ops/experimental/routing_moe/layer/modular.py,sha256=SsmJ38tkPdJocPvg376gIy6UPvwQQ30KgvA8gkTqUDA,10257
tokamax/_src/ops/experimental/routing_moe/layer/monolithic.py,sha256=kE4-jzBvVhXB0lTPTcDbw1NzQnP2ZnCWWzMK-potACg,4361
tokamax/_src/ops/experimental/routing_moe/layer/plan.py,sha256=a6nQFCHT1vL3MwPSuKhtdiQQKt7qqQEAv-AFHFqs0KQ,2944
tokamax/_src/ops/experimental/routing_moe/test/common/dataclass_util_unit_test.py,sha256=uxAtnKQedsToxHgzA03zcLIuPkEE7b663PQB_hYVTro,4250
tokamax/_src/ops/experimental/routing_moe/test/common/layout_util_unit_test.py,sha256=LI356kR2OGwlZ0igBIgoN55GLVgJiUcEaclQ9NKHc1A,3539
tokamax/_src/ops/experimental/routing_moe/test/common/module_scope_unit_test.py,sha256=77801nSYisxoMSwt86-ng6sDxdbGNMfs74mbWAfffMY,9555
tokamax/_src/ops/experimental/routing_moe/test/common/out_format_unit_test.py,sha256=wGxmwGwTob2iJnv819aY88W2CER7SIDOr-pEe9KtKsg,4983
tokamax/_src/ops/experimental/routing_moe/test/common/quant_util_unit_test.py,sha256=lOiau1nqtRmuJDnCuHxI_0pRWhnVmIo2UkVppxgEMnk,4458
tokamax/_src/ops/experimental/routing_moe/test/gmm/config_unit_test.py,sha256=XBBG-Fat7WUzXN_YpnetikUcY4PJzIxhXVAOjtdA5jo,7078
tokamax/_src/ops/experimental/routing_moe/test/gmm/gmm_prod_test.py,sha256=wF-O4zo3WDMIR2fsGSx1PvrN0uayyr3JnJK310jau-c,10968
tokamax/_src/ops/experimental/routing_moe/test/gmm/gmm_test_lib.py,sha256=eVAVyFZVdLw2w8Owrnv5cT2EaLhF1Q2TAaBkoAlZZwU,8645
tokamax/_src/ops/experimental/routing_moe/test/gmm/gmm_unit_test.py,sha256=CGyfS2LvGTEvyL64KkSKwIXWy-SdpkyoCg-qvw8xuTc,22919
tokamax/_src/ops/experimental/routing_moe/test/gmm/metadata_test_lib.py,sha256=KDYvmx-k60pPrJEnmaeo1ypbKNjlFwzx14SQ-MsFPEc,6709
tokamax/_src/ops/experimental/routing_moe/test/gmm/metadata_unit_test.py,sha256=n2-2Kn9HjHXhvdO64Tnvo7HoT0lYfBpkUpfM5tqDvEA,4017
tokamax/_src/ops/experimental/routing_moe/test/gmm/plan_unit_test.py,sha256=uYa26bEUvZ5GWg89Ir1d2pvl6wwLIhqNfIubuU8xong,10410
tokamax/_src/ops/experimental/routing_moe/test/gmm/spec_unit_test.py,sha256=-WeF2m4xQX8YlJ7oN0Isy504UBGIuwEd5Aw9MgfIIvM,3503
tokamax/_src/ops/experimental/routing_moe/test/helper/helper_cases.py,sha256=5Klw6ABK2jc3OxM8DlE21wCI6YZMCgy8hRHWVbl6WT8,6269
tokamax/_src/ops/experimental/routing_moe/test/helper/helper_inputs.py,sha256=jKwaC46YvtO_0gYR2rNAd7kClG86Mr0_k1KlVj9sHoI,14732
tokamax/_src/ops/experimental/routing_moe/test/helper/helper_inputs_unit_test.py,sha256=s54eOCjb-F4O_0g95oPNXnOfz-xMRoQ6-rtoEQlOJV4,3137
tokamax/_src/ops/experimental/routing_moe/test/helper/helper_reference.py,sha256=oGVYY8wqnMQQxxNnDrev4c9T9YIEZGP2W1eWkoWBIMc,9572
tokamax/_src/ops/experimental/routing_moe/test/helper/helper_shapes.py,sha256=US0zRR0qiycTBYB5r1dhfdO64mgay6QFvHCT4TMjPhI,3314
tokamax/_src/ops/experimental/routing_moe/test/helper/helper_util.py,sha256=VfmBlDJFTWid_l_1vwhTV_roY2QGbsfY-jmPhgcKquw,4965
tokamax/_src/ops/experimental/routing_moe/test/io_relayout/output_reduce_prod_test.py,sha256=nkT8l4T3Iv-D5dLucvXWDiyIAs_QXn3LcxZzLmgVzc4,2476
tokamax/_src/ops/experimental/routing_moe/test/io_relayout/output_reduce_test_lib.py,sha256=ckGbdMJSbiA4RI89ZaWcmYUjTgS49583M8SGvY-gCT4,6800
tokamax/_src/ops/experimental/routing_moe/test/io_relayout/output_reduce_unit_test.py,sha256=MsHGAYIDoZEu9Lf31rP9uiotZacJivaBcaLFSNFxcZQ,7385
tokamax/_src/ops/experimental/routing_moe/test/layer/modular_unit_test.py,sha256=LHzYkPPNEdZub5JHxS_cfcDx6rQRnk-RcNdNuYEzd3E,2203
tokamax/_src/ops/experimental/routing_moe/test/layer/moe_plan_unit_test.py,sha256=a7Lmvpdo4mNPeRTy8OznHnZ0cmojeANs6XBlAgwN-zc,1880
tokamax/_src/ops/experimental/routing_moe/test/layer/moe_prod_test.py,sha256=lbTOUP5oLHFol884Y6O0ybR-W-cOuZe4JWrpB-5C4Ls,4496
tokamax/_src/ops/experimental/routing_moe/test/layer/moe_test_base.py,sha256=jLOrcir0q3546pwdH-HbSBdyvz32fwx92c_XB3f4mGY,8191
tokamax/_src/ops/experimental/routing_moe/test/layer/moe_test_lib.py,sha256=SIjOEFeKC0nunOyDx5upNK_Xhl5C3AMVSS3663mUWUk,13587
tokamax/_src/ops/experimental/routing_moe/test/layer/monolithic_unit_test.py,sha256=JMO-Ajn2um0kzlVb4Y9ZT3JpPqmPciLzPmGH5MkRxjM,1022
tokamax/_src/ops/experimental/rwkv7/__init__.py,sha256=pLuw0Q7NXesbwHnwQYGsFYAw1Bjx5n6ee62j2GvsJ1w,726
tokamax/_src/ops/experimental/rwkv7/api.py,sha256=0I4k0fg37JkYQ1IKuIWVf-D6vcmfmb6tOV8mVk73C50,4103
tokamax/_src/ops/experimental/rwkv7/arg_specs.py,sha256=d1l7N8ZH2WAcxrspbKe6kKhjFyZZlgH5NGyDgIg3VAI,2156
tokamax/_src/ops/experimental/rwkv7/base.py,sha256=l4vYuJ9E7guRCNBisHFaZHeMmoPW6Db-PhsfSz4JXoU,12874
tokamax/_src/ops/experimental/rwkv7/base_test.py,sha256=59_5U-__YtasoYedWpf3cZ8NfMR_DpT2Qhtp8q1lKzs,2939
tokamax/_src/ops/experimental/rwkv7/pallas_mosaic_tpu.py,sha256=hBP414EbOMcgBsW8LrsIHRforztFOVWzpHOgNnnzdl0,6782
tokamax/_src/ops/experimental/rwkv7/pallas_mosaic_tpu_bwd_kernel.py,sha256=Vveute0fNlShyQanF3yPyxTyXdi8yejc4fPdOhwjRrg,14402
tokamax/_src/ops/experimental/rwkv7/pallas_mosaic_tpu_kernel.py,sha256=CQzsE7AIS6VH0z9lKGPy05rirKGa8DCIw4_jwirG5pc,10327
tokamax/_src/ops/experimental/rwkv7/pallas_mosaic_tpu_test.py,sha256=Dza3lJ8ucTLzrqh1kWkVMEn04K9lH5PhTRK2E2Of6wQ,6795
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=BzYFUdsXBrwJ2FMx5EXXTdTYezKU5QR6xbb7812gy2E,3078
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=pzklZLPH40iR5u4TuAwjyqd_8IgMp7pO-RKUM4w6yPo,10399
tokamax/_src/ops/gated_linear_unit/pallas_mosaic_gpu_kernel_sm80.py,sha256=fa4p5Y-wPaFuzhLOVZ88B_7WfF70OYVFnEYLFw5zl7M,7984
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=XKAaAUK8aUkq8fJoLkGgiq8mBJRhXBWeZt0RIlp98VQ,5501
tokamax/_src/ops/gated_linear_unit/triton.py,sha256=O-yPylVQ5NFjV4-VjqQfA2yJ8GxTaN5UQkdbiBrAv2w,10973
tokamax/_src/ops/gated_linear_unit/triton_test.py,sha256=UGqGhFqLpcyxdt9KvzC-XitTxGRPZ2DL2jhEdmhaCrY,10657
tokamax/_src/ops/gmm_v2/__init__.py,sha256=OtXRXu-czwR1bk_ECYH3uJ5bcOomEIkjsyLFNJ4QZQo,655
tokamax/_src/ops/gmm_v2/gmm_v2.py,sha256=jMjeKXT89mYEetHgCbYhYd_b5QheLIZWFXEbFonkg5k,70274
tokamax/_src/ops/gmm_v2/gmm_v2_perf_test.py,sha256=Uy_tbgCj0DwDlwFdkGQsdtQdBaubNbJgjD5Twk_is0I,10825
tokamax/_src/ops/gmm_v2/gmm_v2_test.py,sha256=f3i9ihIQQtpo0bevBLBxrZH97WuQyk5OKOM2-Hw1ObQ,89172
tokamax/_src/ops/gmm_v2/tgmm_v2.py,sha256=-SdWcviGjx4QlCZzNTnX-mPjIG_d-4wIb2G8kYExHOc,35423
tokamax/_src/ops/gmm_v2/util.py,sha256=A1tpYOIhCjHvupDTCDlnffAmjrZ5UNQ8rNZ1MNaX2J8,1899
tokamax/_src/ops/grouped_lowrank_proj/__init__.py,sha256=P0isQ1Ev-qzsu1ahtJ6vzc_8biYHZolgNCffD8IGPHA,735
tokamax/_src/ops/grouped_lowrank_proj/api.py,sha256=uwO-pNVwE6NzRZUKXYFZTIzdbu5I-jxEEUe9fK93dBM,4460
tokamax/_src/ops/grouped_lowrank_proj/api_test.py,sha256=g5Pq_hKnHPGbZ6ly2AvFBox4AKaQIYAplJomrWIv17Y,2989
tokamax/_src/ops/grouped_lowrank_proj/arg_specs.py,sha256=DbcDKq3rf8akULKVspwU1FbXglt-oF66GKAp1bPtMQQ,2888
tokamax/_src/ops/grouped_lowrank_proj/base.py,sha256=T9YmrIgO-fi5LFkcBUWCCyssAJ9Iyt6s_wNr35ggyIE,6815
tokamax/_src/ops/grouped_lowrank_proj/base_test.py,sha256=13q6NFXU0DCYHaHK1_81jWMMc5b8U-h0zXl9cJ3tADo,4103
tokamax/_src/ops/grouped_lowrank_proj/grouped_lowrank_proj_test.py,sha256=kWXF644H8u64LEvvoThLbf8Sh-fsWUKYT7N-6Xga6EE,6668
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu.py,sha256=Sd6-36NLlHrok-txOJ7L0Kj0o5sYnHiIy6wsYVpuAyM,8944
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu_kernel.py,sha256=_rQ2CQGK3sGwJqPzeH7I8Po1QDdbqKsNakd-VZjpxf0,11478
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu_kernel_test.py,sha256=_EUAxjYFcLd_GRp5JFBChPUev_z8hA897QsC-bv9uk4,9353
tokamax/_src/ops/grouped_lowrank_proj/pallas_mosaic_tpu_test.py,sha256=hrpc32IntcSzquYnfl-dO00JcrviO3oRzTuJJiZbpj4,4868
tokamax/_src/ops/grouped_lowrank_proj/reference.py,sha256=gSg5Fbu5kP565wUA6_r_5AWvuHyQtKPzzgN1Eg8xZd0,4105
tokamax/_src/ops/grouped_lowrank_proj/test_base.py,sha256=yCxX6eKDvqL8fv9gPzdOZ-CMOEceTG3mXlyR7hjmiq8,6443
tokamax/_src/ops/kda/__init__.py,sha256=5GYDGBGbUUq9RtvdIzQamyT8GsBUW7jy1uhYopJKn5w,865
tokamax/_src/ops/kda/api.py,sha256=0amfSnfK7Q5S0W6SUu4OpoLPxfcYt51NLt_ePkCX3jQ,9295
tokamax/_src/ops/kda/arg_specs.py,sha256=FqzKT2D84S3cHmetHlYyxVmMUjAzU6fm-5MumusDOT0,7668
tokamax/_src/ops/kda/base.py,sha256=KPaOfvQ6zHv20ZP2r8cliIJdt7anKrx3p5FebEHdX88,9844
tokamax/_src/ops/kda/base_test.py,sha256=E2YLsh9r3M27Tkvsu3XbIsCP-N3IZGWTofOyzrt7Asg,10519
tokamax/_src/ops/kda/common.py,sha256=MWuKOZHfc9yWLNmN-TFhl3ip25jx9IfanfKpaCdsVIk,28392
tokamax/_src/ops/kda/cp_utils.py,sha256=-HfDEcdV4oEt7b1dzKhuKLeyVqicsVjyoop_tdDxTuY,17435
tokamax/_src/ops/kda/pallas_mosaic_tpu.py,sha256=R6hlvA1dD-R6CAxZYR2ejR_cSpaQdwUCwcgEe1_a_Pw,24194
tokamax/_src/ops/kda/pallas_mosaic_tpu_bwd_kernel.py,sha256=AXuWOtgJz_RegRb4ZySlh5vCpJZIdUClkHPS_VD-iNg,93888
tokamax/_src/ops/kda/pallas_mosaic_tpu_fwd_kernel.py,sha256=U67rkIFMv-k4bM5V_q9yRtnr-HvTZmvgAMkk8ZHxjiw,75325
tokamax/_src/ops/kda/pallas_mosaic_tpu_kernel.py,sha256=n-vjY-HkQTrWz5q_5xpO4cwsvWYuKvP6XVSQ_0ItAAQ,1000
tokamax/_src/ops/kda/pallas_mosaic_tpu_kernel_test.py,sha256=Ta6e05GPifHzwf7XWwpdIkUMrOqLHAIHwWjLPEG2dvQ,92830
tokamax/_src/ops/kda/pallas_mosaic_tpu_test.py,sha256=qxGsiSr6C92RPIF9Mt3w_PGtFWzxnEKXU3SoCIjCP_8,9517
tokamax/_src/ops/kda/pallas_mosaic_tpu_types.py,sha256=1UBktLxNgLMLkL5dHObozD7GkxPEB1aN3TubnP6Xdxc,2791
tokamax/_src/ops/kda/reference.py,sha256=Xmb2bxrBimd70cjdDP3bl-GZUDVZV5-Nfw2jsGvt9qc,8376
tokamax/_src/ops/kda/utils.py,sha256=L04pen2jrniCpOqyReksPUE9gpVtV4e4EFtv3e8Ubc0,17330
tokamax/_src/ops/kda/xla_chunked.py,sha256=RqPoNVcgxWf2ws9H_uetKk8M3xWo5pDN0_WAaxYQF-8,9994
tokamax/_src/ops/kda/xla_chunked_bwd_kernel.py,sha256=QkZ-xzSctVGVTRJFHINLr2gOKR07kzVsnKx8Xtk0WMI,2412
tokamax/_src/ops/kda/xla_chunked_fwd_kernel.py,sha256=zggfRQK4J34q3f07i6kZ7yf8ffpRT_KA0_e54RHNqCI,16338
tokamax/_src/ops/kda/xla_chunked_test.py,sha256=-Rx6Qw55VlurceCPedrOzzxhlXQzuWq1alrBQcjVxI4,8042
tokamax/_src/ops/lightning_indexer/__init__.py,sha256=Zx5fOsw9LvX-EK0a3SIACWiHqds0wZ4N-i0B8UxDxyk,689
tokamax/_src/ops/lightning_indexer/api.py,sha256=YZJ75Qa75M52d3LcoQ1sBIMvifrmIizw56KDfp5Z2zA,5020
tokamax/_src/ops/lightning_indexer/api_test.py,sha256=P-vZO9l0ZAFF4wBm9-mBkZ0WeBUypQT75G6tRtCkFcc,11432
tokamax/_src/ops/lightning_indexer/arg_specs.py,sha256=il8b3sNMtLL-3VAPFcJ12Wu-uv_Byd27B8rhXFCVIL8,17962
tokamax/_src/ops/lightning_indexer/base.py,sha256=1HgJd3trmGpaBABRelB94CWHAMd8B9gas--a3Dzbt-0,5144
tokamax/_src/ops/lightning_indexer/base_test.py,sha256=9h6MKONV20YelkBciB4DDvYi-mKFyq9UTa2Tqqqx5aE,10808
tokamax/_src/ops/lightning_indexer/pallas_mosaic_tpu.py,sha256=C6vWWmtk2epzIiKTQG_rBT_RjrHoNz9lb1hpsXlUEs4,13161
tokamax/_src/ops/lightning_indexer/pallas_mosaic_tpu_test.py,sha256=HbfFt9YQ3oCdr__l1qT59EBdrU-nh1cPqEJTgLeKUiY,2505
tokamax/_src/ops/lightning_indexer/reference.py,sha256=bDgNJCV-c1Eo2czCtoLCNnefR6zJOdcb7sAfrUMbBZU,7058
tokamax/_src/ops/lightning_indexer/test_base.py,sha256=LMAE6s-Uhy8RckvDRiw-nLypWY3lTIWcJtdS9-a3quc,7356
tokamax/_src/ops/lightning_indexer/kernel/__init__.py,sha256=ipixnHwBtMoXGa9jiYV5WmQfKpZrfA4tiI3GN9DCzdQ,705
tokamax/_src/ops/lightning_indexer/kernel/bref_override.py,sha256=JvZhj4Z-qLmlIdm4PFFvYHCoasktQG1o0S-o5S5H6xQ,19845
tokamax/_src/ops/lightning_indexer/kernel/config.py,sha256=HG7FnjDfwbgxL_jtIxtUUeONFAIM01lh5kEAQ9_ENeY,5344
tokamax/_src/ops/lightning_indexer/kernel/dcp_sc_compact.py,sha256=9lE_aJriLfHX6n0y611KKO2UVgGyR5CT4H1a6XJBuk4,6194
tokamax/_src/ops/lightning_indexer/kernel/metadata.py,sha256=qsejuaKXvqLL2_WcBDV55fw2oTnL8PKQqIU86zy58E0,19480
tokamax/_src/ops/lightning_indexer/kernel/metadata_test.py,sha256=YBP5EVNCPwLCkyj5fAwAGSkLdxfdTJS3F30APs_7sf0,25054
tokamax/_src/ops/lightning_indexer/kernel/streamindex_topk.py,sha256=UcTKB0BqewkkJpVzzJ2XMij4EdIZpxkBVtNt_zDLUXI,68272
tokamax/_src/ops/lightning_indexer/kernel/streamindex_topk_dcp_test.py,sha256=p3sWM3A6uyk7blqQ4iUtdIvXcxiyjnHtoMea4eZHXU8,13543
tokamax/_src/ops/lightning_indexer/kernel/streamindex_topk_test.py,sha256=0vuYqpaN_flnC-jbQDnF1znMAiLjhW3KVnWxkJb5m34,66436
tokamax/_src/ops/lightning_indexer/kernel/tc_streamindex_topk.py,sha256=ylD2vIu3luLTOEdrshch5_IeP00vyE9kkxqzLa3bv9k,79343
tokamax/_src/ops/lightning_indexer/kernel/tc_streamindex_topk_test.py,sha256=1GhSox1rYtG0dRchpTiXgUFRazHjrmpmUtWkr2aQzBo,4806
tokamax/_src/ops/lightning_indexer/kernel/tc_topk.py,sha256=DcsAoIq9thOyyf18a0cD_0fuJhTuUPwd6AHNCi-MSVg,70908
tokamax/_src/ops/lightning_indexer/kernel/tc_topk_test.py,sha256=osvUoHiZeRuekbif6U7oXHSfnzFwbR4liE8n5Uow8wk,10597
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=cVmCZzggWwJR9GSBE9q0GGSDtH2ChEy0wgl3dlu2QzI,4798
tokamax/_src/ops/linear_softmax_cross_entropy_loss/api_test.py,sha256=StdnlQthH8o3BLdBIpc81D3EW5_Y3tvRYofxWNjUS7g,6201
tokamax/_src/ops/linear_softmax_cross_entropy_loss/arg_specs.py,sha256=wmLi_JJHHMMLJ-m5InX8HyBgyBn5DEWuBuINMKTlPJY,7410
tokamax/_src/ops/linear_softmax_cross_entropy_loss/base.py,sha256=mPH7Roq1_N5d5bbJoc-Fzawinj6rIJ-u-3XepQSSFqo,2842
tokamax/_src/ops/linear_softmax_cross_entropy_loss/base_test.py,sha256=dk9kvrTaS4hFjSlxjWxXmw7rAySnVycKACXYQaUz_RU,2464
tokamax/_src/ops/linear_softmax_cross_entropy_loss/chunked_xla.py,sha256=_1LHtD6JodpNRBM1Cv4EsMQpYcFA_0OO_CmefOtb5gs,12788
tokamax/_src/ops/linear_softmax_cross_entropy_loss/chunked_xla_test.py,sha256=CoN2TugFD5HecgrAU1HGOKQY34WjkbG31dwzkO-LKAU,9319
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu.py,sha256=mvequ2nqE61OcpLxvQoNoCLnQJTJ5q5iNOvgQcGNeZI,10686
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel.py,sha256=yLhysYCKsNmEk6OJNq9p80hvKR44Te110YXeM2BmKCc,58191
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_kernel_test.py,sha256=Cjjo1d7UtfNlLpnbbQDZZyA8BWzFkOM_buxxZH_kfhs,67762
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_test.py,sha256=5Pxh8DR36aZKUKVvJmtqbRMFI8hqg-_9lN4MK-fY1w4,8283
tokamax/_src/ops/linear_softmax_cross_entropy_loss/reference.py,sha256=OAR_er8dlD7WLHfd-FIamL_2e_oDPZwIfSNCu5JhWNQ,2981
tokamax/_src/ops/linear_softmax_cross_entropy_loss/reference_test.py,sha256=mVNHDTuQluJa88CY617OyAbZvZwMoMUdfEFW8KcqfMA,4932
tokamax/_src/ops/linear_softmax_cross_entropy_loss/test_utils.py,sha256=0gjGXVuLWy7dEePckg4zNTbjwmYrnQCNz2KQao3JUpA,1963
tokamax/_src/ops/masked_dense_mla/__init__.py,sha256=E778mfVnSKF_KQi9lXNvRbAVJ-LjoAqirGPyjaCw19Y,688
tokamax/_src/ops/masked_dense_mla/api.py,sha256=wpJEMJc0_hOzZnlrxOhIuZ_wk_x2VZMFy8SPGAbpkbM,4676
tokamax/_src/ops/masked_dense_mla/api_test.py,sha256=uzyJHgLSpGnUDeWxs-E6yWYD_6gRphpSNPB7IWTv7_Y,2153
tokamax/_src/ops/masked_dense_mla/arg_specs.py,sha256=vN4Ik57ZOvAsS5KgF5-9Z4-ommeTHzHyxboelx8quuM,6219
tokamax/_src/ops/masked_dense_mla/base.py,sha256=kO3tYqYRFuW0Hl6b-OmJTzo0VuruJwKPp43F9gC9FhY,4917
tokamax/_src/ops/masked_dense_mla/base_test.py,sha256=D-8M95MWvtQxP-KtXjHDqoQOdgvbBt9RlvL3bSyuBYY,1986
tokamax/_src/ops/masked_dense_mla/csa_mask.py,sha256=WYq0bFhf0g2BkgZCgMVk_mmIad51zY8l8M1X6PAWTtc,5298
tokamax/_src/ops/masked_dense_mla/kv_cache_utils.py,sha256=-VdcGqodouMCBjw4ouF-H9T7SWFH6XSoSVAJmDedjjY,16907
tokamax/_src/ops/masked_dense_mla/kv_cache_utils_test.py,sha256=1_qFH3CFTbZVsX4GMsjVAosmI8UZ3ry07EAmXjzulzU,18833
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu.py,sha256=Y0jEfUFJqBYNBSQcjDy5n-KiQQ70wDtv0gUIPZ9p878,4765
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu_kernel.py,sha256=U_Qq5OuX7M69ez-BtIGBhjCLiaZLMdDiLJCdIM_a8h4,33293
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu_kernel_test.py,sha256=nDtaazaIk_KsXIckkh3RKrNC_aKysT7b-GbH8FiPeDs,15886
tokamax/_src/ops/masked_dense_mla/pallas_mosaic_tpu_test.py,sha256=K8J9y5kXtDFaBtGIS833eCdmrmoT0CHR3xK4MQaWWHk,2067
tokamax/_src/ops/masked_dense_mla/reference.py,sha256=SAmUIfzvnIFXNxltRsjOzobL1CGuWGA0273bBiRnBGQ,6032
tokamax/_src/ops/masked_dense_mla/scatter.py,sha256=t43oRFxij3Vom5_ODm2VZjW01856Ef7SMThX8TKGBlo,2806
tokamax/_src/ops/masked_dense_mla/scatter_test.py,sha256=EQB_fotVENFvvupVto6bNqJKsLdoWny9GoAe6M-ZXuM,14728
tokamax/_src/ops/masked_dense_mla/scatter_utils.py,sha256=vfG3RfnmwLFD8qH11ff3Fnt2-tuPsvXnZ4qYrCG31ys,7454
tokamax/_src/ops/masked_dense_mla/test_base.py,sha256=JEQeNyi6_Lik6rFDf7jJRf4m-k1QyBVzsIH58JhNgok,6687
tokamax/_src/ops/mhc/__init__.py,sha256=xpTpj-P_eNfir6pCWMzDwzPUzd7Iw7l7RJ2586sGcR4,736
tokamax/_src/ops/mhc/api.py,sha256=6WkYYTLrI0n-TtjWBSM8JfQft3QGdDORx5QXX8qSOm4,9713
tokamax/_src/ops/mhc/api_test.py,sha256=QvDKuKPT60e2zRyhmvB7TaeP9z6I-dJrnpqwLh9FBBk,4534
tokamax/_src/ops/mhc/arg_specs.py,sha256=gnu7MdxcqVRVVDLpD__TNIt5VXQ4uagr8nwnJZqChko,4019
tokamax/_src/ops/mhc/base.py,sha256=eTiqzwn6latVnUXLAwvF_oxd4qEQGubnLGfHSZXsT_g,9355
tokamax/_src/ops/mhc/base_test.py,sha256=d6ARSLngBSmJ2l04JO1rPPEkeIA_msYwXPlLdyapwmc,7141
tokamax/_src/ops/mhc/fused_post_pre_kernel.py,sha256=J4_mU1xnga67KKgjjay-qqKaHe6Tq_bt4WRh2e1Q5tQ,11030
tokamax/_src/ops/mhc/mhc_test.py,sha256=QJHcsEQYLw_XvIzXd_6AV5vlrI9h3HjTQV4QcsMMgCQ,19629
tokamax/_src/ops/mhc/pallas_mosaic_tpu.py,sha256=dleyzJHdmuRsIJCrix3Fd1W5nu224XAxwxfYljs6pzc,8170
tokamax/_src/ops/mhc/pallas_mosaic_tpu_kernel_test.py,sha256=cCTHVqYXFTubJtb9PVZkYG3dq3mLGX1D-8CGrvnJH-s,5614
tokamax/_src/ops/mhc/pallas_mosaic_tpu_test.py,sha256=4RDunR0qTmF8XlMSBC9w5eRPHvbGpWCSNb6qR2M4qLs,4051
tokamax/_src/ops/mhc/post_kernel.py,sha256=LCRHGPzu3OtPEzkrAGUjxovS55flqVCL3a6_mC2vtCE,5881
tokamax/_src/ops/mhc/pre_kernel.py,sha256=iORjYQTMJ4xKTt3GoQDIFuh-P03ofysV9nv-8J4zNzM,11905
tokamax/_src/ops/mhc/reference.py,sha256=Zkd72suUXdnGo-sORUU4X66oHF60f7DrIi7Kgvh6nLs,8442
tokamax/_src/ops/mhc/test_base.py,sha256=Wg8V6R6tdj6i-KAkhVxeXwQjVHEBvOTtrwixSS-2qPQ,9077
tokamax/_src/ops/mhc/utils.py,sha256=4MDYnB3j23bHp2G6ouqAa9giJbQXvequx19Nh5BdgaY,5623
tokamax/_src/ops/normalization/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/normalization/api.py,sha256=z04jE0sXt1zq44GJrQPSPq-Wia57k2kWkiyp7OVO76U,4516
tokamax/_src/ops/normalization/api_test.py,sha256=4150si382wUWSdBJTOXrMoLxMXgvPCQAuvOi_rEg3rQ,5167
tokamax/_src/ops/normalization/arg_specs.py,sha256=Qai5dlv7OgAvmQb7RmMOl2dc8qtkpL7Uno0hWPAeP2c,2750
tokamax/_src/ops/normalization/base.py,sha256=1w50Ew4BDNLvb7onehV9Xv7Xq8HL2CEL2yPhni0Cpd8,6241
tokamax/_src/ops/normalization/base_test.py,sha256=XKZTDAMwsol5oEzzdB9Wu-pu2_12eSSadGLyMeKKweg,1027
tokamax/_src/ops/normalization/bench.py,sha256=DxvvboTF2qqmEVs4tSeCutrXfsKX0uZcex51ogzK7ZU,2188
tokamax/_src/ops/normalization/test_base.py,sha256=ODAR0SG-83FHYDIlGxR54ZuJkEQ-1T4hqOH7NDocdqQ,6684
tokamax/_src/ops/normalization/triton.py,sha256=j84k49vOqOfEvW8prKAdiqDbshQmZWV9_ZhQQ_nyD4s,17980
tokamax/_src/ops/normalization/triton_config.py,sha256=Bunc35aBzo3jT1DyeKVRApKfSFAvuDC2bFWylkRhdYk,4157
tokamax/_src/ops/normalization/triton_test.py,sha256=PLv-GSsFQYhsGctQhr6Qns9WBdXuUADIfqCt63z0hAg,5409
tokamax/_src/ops/normalization/triton_vjp_config.py,sha256=R-7FMLYOlvs8aZUw3uVmjtus9Z_v-1EMqKq-CVgxdSE,3043
tokamax/_src/ops/proj_and_save_state/__init__.py,sha256=EyGX1VT2fUTgHUF2DrOYOTHZh_jU5Z4W-32TKxlAJx8,735
tokamax/_src/ops/proj_and_save_state/api.py,sha256=CvAN64b8TyQpa686ZXPTP4R3xfmFGYzUjb-fvi4fKng,4155
tokamax/_src/ops/proj_and_save_state/api_test.py,sha256=s5JH3O87pRqJCToiQNKQ36xln9d6ortebgqbS-mOKj0,2745
tokamax/_src/ops/proj_and_save_state/arg_specs.py,sha256=rSTzFTtUEB7b9hDFlD4Utwi8p4oI2_-B5UjBI-2rE94,5291
tokamax/_src/ops/proj_and_save_state/base.py,sha256=Lo2j4APoPx-oJAMdi3vI9xsdVXwwdbYmQMA7WpGs9Ks,5384
tokamax/_src/ops/proj_and_save_state/base_test.py,sha256=2HBYue-rVBIezedv6wEYmfKdnly4iODdbZ22w6ohS1g,3896
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu.py,sha256=3-tSH3Fkk2uIOAG42FLY-maUdhDqAF_NJUd3E3u_xSM,7686
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu_kernel.py,sha256=wUeo9kUCSXzI-40yquyjfP87-mF6My2JggnotwNwdUY,15642
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu_kernel_test.py,sha256=VZqsnL_frJVoFUQ1_kjpOsJgxPobvz4MrlEkbTx9MH8,13006
tokamax/_src/ops/proj_and_save_state/pallas_mosaic_tpu_test.py,sha256=WoI53DbYkFJGoxLI99zcnaYDLci1007tfZVeazn7KiY,4361
tokamax/_src/ops/proj_and_save_state/reference.py,sha256=7hJvLWWbP89oyAlzmxcG-hdahtFijOlwdh60PJtWxcg,6785
tokamax/_src/ops/proj_and_save_state/test_base.py,sha256=clQPpXCXjxkgDGrk3io-C92n8bSVdBzjw4aqAldGaAc,9612
tokamax/_src/ops/ragged_dot/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/ragged_dot/api.py,sha256=kXJHavZG8hNRfBVK5IHdDJnDo3DLzAqdTGhSiS7Fp-w,11610
tokamax/_src/ops/ragged_dot/api_test.py,sha256=y3VJXdrlKwlDDNlG8r-J1rKP3QpNUQS7hCZ0ul3e-pg,21711
tokamax/_src/ops/ragged_dot/arg_specs.py,sha256=zN8hnWxI4XntjVWqUUdtbiZ2eO1dFCin45GS6sQvJjw,19538
tokamax/_src/ops/ragged_dot/base.py,sha256=iSnrkorYl8NKPVlQKDQxDUXBcEmU-mebOpwGmW48OTM,16430
tokamax/_src/ops/ragged_dot/base_test.py,sha256=v-DKxGduaRvKvVYheRkH12Mf8c12lyTmurS_nvZgpEY,3215
tokamax/_src/ops/ragged_dot/bench.py,sha256=fFv4wWo4UXAmtRa4RwmZSl-sQh5QMoNnVvhts6mtwP8,3845
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu.py,sha256=y05injZSBfcdZrK15ACDFFlQH_8SAWklcgSZFl-Ilqs,21094
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_common.py,sha256=EFf2WC7PVOImeydmDXW9PV4WLyPUdPxDGzeirmC3jxM,8238
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=7SGqC1pr962yso2eJhfLdw6nkZZI36Nu_knI2JZHbvQ,9055
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant.py,sha256=CMBUXQdMDfNbEfc099zCvxeA_vjT86Y_huFKEEQLfhE,16260
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant_post_scale.py,sha256=8PijHSvOBKwYXGLAcxJNtChB8q_tLTCcPcGDShHVO0k,15323
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant_test.py,sha256=t3nbyUWDbS9nuKhjNJn1qTnZreBz93Tle_UBNhd9GRs,25859
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant_x8.py,sha256=HSDHFD3JltbSsz91gKnZComidHRywuLNSGfNgwVD2E4,34026
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm80.py,sha256=SCf6UfZCDU3XeZOTw01M00luSMBU6xvZ2JwiHhKVROA,6596
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90.py,sha256=U33cxx6r8FFUIxdi7znuo7N9oXRydiDOoSis7YeH_Nw,9241
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90_quant.py,sha256=tptf4W53BzOhgRhH2kj3mHGmOyPURnWF0CH4DIgPwIw,13275
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_test.py,sha256=wvNd2SrvJ3idef6bT0kbU43vVO8HjcZZ3wc4wc5lRl8,4795
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu.py,sha256=4Qj5NAh_7QArXYcnOc8QI6m_mE0yqDy3NuwrIt3QazI,17582
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_kernel.py,sha256=uigpElQYBbB-Pa63PHxhhi8Tpv9rv8MGRBAQxwTn43w,34996
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_test.py,sha256=rxDW_F4__xDKHCIVNVgD1Qy66A_J2OLnA88JiKnnUaw,11749
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_v2.py,sha256=NX5gYtgWWT0zPJ81_EhB8bIXSx2xpGijSyIScit4YqQ,19672
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_v2_test.py,sha256=hSfOpWkxRa3SwgvDgHLPrXxaeTywuK1z-RzYk3C3Q2I,23896
tokamax/_src/ops/ragged_dot/test_base.py,sha256=2-1RGAAjg34-jMLCW2n29kT1x29-SULUQ56kCeeRHns,11857
tokamax/_src/ops/ragged_dot/triton.py,sha256=FNY31Qya-3MLUTUpzhbyKPwjEZZIXr7I4OgD423drmQ,21886
tokamax/_src/ops/ragged_dot/triton_test.py,sha256=U9T9DmpMvL4hZLKRfdF2Q6t2gttWm0no0GWqXeGQNgc,3518
tokamax/_src/ops/ragged_gather/__init__.py,sha256=Xh8d6rEAf-Bp-eZ-luzfGMPT-92LDIRQeN7Fncy0XCE,693
tokamax/_src/ops/ragged_gather/api.py,sha256=qRy5aLkFtGAAKK-U77y7_HyRqlfnH4d3Xd9WicH0fuo,2908
tokamax/_src/ops/ragged_gather/api_test.py,sha256=Uxuf73nJhR8X99ZuBIVTpF2JjRRZvmiwOzNS5RHbkdU,2788
tokamax/_src/ops/ragged_gather/arg_specs.py,sha256=f5qEZmpvbF17tsVKBvqeSzOxwXKi3-3b4tA1YpegiTI,2779
tokamax/_src/ops/ragged_gather/base.py,sha256=iRTdsjOPQtGy4fUBaglUY6y6s53Gyeq_WDiVH_svKR8,2881
tokamax/_src/ops/ragged_gather/base_test.py,sha256=dh2SYONzWgDSeGANyMtzDzNnT1biia0A3uXPCtUe8zE,1773
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu.py,sha256=u9V-nMjz8W-aho7O20IR8PFHTAbrttKJjzERretuWyA,2646
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu_kernel.py,sha256=vO0KnKhbc2YOmGVKN8_IxW_TujtqnxJclw5NwyvtHgw,10357
tokamax/_src/ops/ragged_gather/pallas_mosaic_tpu_test.py,sha256=9lPCuXqJBI9_2Tb8PUlFmgg1T5mThIMOxNVbKqx_DFM,1670
tokamax/_src/ops/ragged_gather/test_base.py,sha256=on3Dw3omebh_YE9GO6S28OOOzBCWjk2Ko11OwQWkdM0,3337
tokamax/_src/ops/ragged_gather_reduce/__init__.py,sha256=1yQAi7-n8YX_dQ8BbXTYO42JFSQSFzVPm5gtW712oe8,700
tokamax/_src/ops/ragged_gather_reduce/api.py,sha256=tgZZW5zkYFvZs_jJE8PlU7Nk32Lt1fzAuAa51j2P510,3670
tokamax/_src/ops/ragged_gather_reduce/api_test.py,sha256=S-HPtU2PHPLHzyyE_IFbcrNUL8xhTPQH-5use0EK-ZI,4585
tokamax/_src/ops/ragged_gather_reduce/arg_specs.py,sha256=8VZqkR9kRpAJvvYqVomAcqS8WG7qV6GBGuGfPGVHIqk,4944
tokamax/_src/ops/ragged_gather_reduce/base.py,sha256=hVPleKgA4b5VFbkk7V_wd83SKYCcHch1SQ4ighHmo_8,2719
tokamax/_src/ops/ragged_gather_reduce/base_test.py,sha256=tCG7_0ZQEwRcsYjHOz_fYExWi2zMIm_d8sv28bCjp_c,3674
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu.py,sha256=SVGzNNCXjcHQHD07erJu772B1-eWkYN4oWtvpMSjYcA,6004
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu_kernel.py,sha256=rQi88g_8n3IacNbPMFnyuLg3erPbLaPf1i8dvy9f704,27104
tokamax/_src/ops/ragged_gather_reduce/pallas_mosaic_tpu_test.py,sha256=iuLT41T3gG4m0G7J4wv71MC8-us7StopqLE8-G3bEiQ,9837
tokamax/_src/ops/ragged_gather_reduce/reference.py,sha256=C9MA8og2PQsFO3LlCq37Oxp5b957RQ-WNRlF8GdT9g8,2265
tokamax/_src/ops/ragged_gather_reduce/test_base.py,sha256=97haU4thVhiZiOTMGDyIBrs7h8n5CHF4sAeS7R8BzDI,6295
tokamax/_src/ops/router_topk/__init__.py,sha256=N4NCVxareHoVJ_e4t5ohe_0niltDrq2564ujDjMbASY,706
tokamax/_src/ops/router_topk/api.py,sha256=Lvp-LoAVymupH8w8JAdAimqWyGwg4olvB-qnffEDRgk,3720
tokamax/_src/ops/router_topk/api_test.py,sha256=bmET9--aBMvUtWXXcbv4BwpL8s_R8jBvHELNlGvfyMw,3357
tokamax/_src/ops/router_topk/arg_specs.py,sha256=PCAVrFU-L2C63zC3beO77mCt3QsJM8MqDGO3gnAdMWA,2196
tokamax/_src/ops/router_topk/base.py,sha256=VOs2MHlFtY5-9Nm8nPE2Nwd9FQGPfEFpJ9yz9G7ImcI,4235
tokamax/_src/ops/router_topk/base_test.py,sha256=LGqr70jhXAqm4kSi9gjNmAMtgvY74uIsoIt2aIROLJI,2521
tokamax/_src/ops/router_topk/pallas_mosaic_tpu.py,sha256=B13NR6jqnibdyvylHEOQhdvXOyKsLOeLxPtCAV55Qiw,6241
tokamax/_src/ops/router_topk/pallas_mosaic_tpu_kernel.py,sha256=v3MRJ86caxh4jZdxtbK_GWmJAXlBcIn2kaJZ0xDnKkA,4399
tokamax/_src/ops/router_topk/pallas_mosaic_tpu_test.py,sha256=btrx3-phJdCyWJf6mFARHPdye3AFBKfowmG-6QutkyI,7912
tokamax/_src/ops/router_topk/reference.py,sha256=4_SHnCV3v2E8oXVTHYyC2weljdHoYx9Z1rC8Lykt_aU,2811
tokamax/_src/ops/router_topk/test_base.py,sha256=X4UfmP36FZlHLb8TEgx-KdyYAbd66Rz3g_BTm5CkWwg,8330
tokamax/_src/ops/sparse_mla/__init__.py,sha256=c03KJKlKUVPo8tcxS66T-mpz5V9rhNdEz1b9Ydukp0Y,704
tokamax/_src/ops/sparse_mla/api.py,sha256=L-e9vZDuOLuCEVCNyMb4Sdu6rDTV5aRDU9SXBSbvFsI,4261
tokamax/_src/ops/sparse_mla/api_test.py,sha256=hFQXjJhw--p4hpuHjSueIm9eVuKkWSDMbB4mAfcDREQ,2061
tokamax/_src/ops/sparse_mla/arg_specs.py,sha256=osGWmC52sOSkH4k0BSF_kBpSqk9uG-kUeJg8-jro2lg,5260
tokamax/_src/ops/sparse_mla/base.py,sha256=nyyTwYLUyGv8uCQEYFkrLj4YHlCknCwT1I3bZiBzK6s,4382
tokamax/_src/ops/sparse_mla/base_test.py,sha256=47yhkXB1WahHHV-o7xMAEM24GdPHNRk8BFVrXml1xOo,1975
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu.py,sha256=hUa2uBusRQHAdZa44rvYocox-Lchia30E5_unQ7RnKc,4434
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu_kernel.py,sha256=iQ7XJIxac3HawMV1Hy4CvnFdMbjV-g65TAUApSXb8kc,31495
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu_kernel_test.py,sha256=IMKfQ7de8art7VTtwfGVYSG76YlcxlCLcK2RQQIcVYI,14570
tokamax/_src/ops/sparse_mla/pallas_mosaic_tpu_test.py,sha256=oCyW1BCTP8IEVYZ6_si2YD_DWbyQLdCg2GDIvHRuoqM,1901
tokamax/_src/ops/sparse_mla/reference.py,sha256=qjdWzAP0WZU1tZg7isovq3bUqN41F_-KAFXJmOH-R_c,5361
tokamax/_src/ops/sparse_mla/test_base.py,sha256=LMGbVtmXcI-3NqA6e1gSLtT95Lk6Q0kIRuqe7C69CZg,6197
tokamax/_src/ops/splash_attention/__init__.py,sha256=fmfcX8cskB_eEmnhvsaai-tSLNc3gaGwxL_A9UNTEn8,688
tokamax/_src/ops/splash_attention/api.py,sha256=n-I2BfS_fXahhuShTi3bJbnjabM7RVvmE-f3G_FNaJE,5565
tokamax/_src/ops/splash_attention/api_test.py,sha256=h1ddOtB-Ohhf1mHXxL3n2j1-nrbRs_MmCQ17mbFigWA,16392
tokamax/_src/ops/splash_attention/arg_specs.py,sha256=L-BUJAGwGew_IZ-xNyG7L2T7jvQZbW4zGmplS8iFb-s,4835
tokamax/_src/ops/splash_attention/base.py,sha256=e29aI4bh7xD7wYpn9f1mmc1osO--hUtQK5r5FEQra9U,25268
tokamax/_src/ops/splash_attention/base_test.py,sha256=DsmHFcEmxal24j667tCtomlaydX9u5S4Yy5aIaYyqZQ,19757
tokamax/_src/ops/splash_attention/pallas_mosaic_tpu.py,sha256=YtA1V1CQ6Vy8_t_apMB4lXPUT09D6Ja6DgOs9sk8zdU,13533
tokamax/_src/ops/splash_attention/pallas_mosaic_tpu_test.py,sha256=XQLHGpfVwmWCRAG_76NvX3fnhH4YccPpGNBPnZTP3Uw,15932
tokamax/_src/ops/splash_attention/pallas_mosaic_tpu_vjp.py,sha256=lQ3lezUqNfWPOIkw5L_jDePm6onHaBkDaqUkrgYB-34,17036
tokamax/_src/ops/splash_attention/reference.py,sha256=CUpLZ_KFEi2IPKsC4lJ-o7rfCT9X8Cbwz4M1liSt-cs,10668
tokamax/_src/ops/splash_attention/ring_attention_kernel.py,sha256=o1cTTx8odu5T6x8WO76HUpwOujXpDYCPAE61L-UZmCo,22486
tokamax/_src/ops/splash_attention/ring_attention_kernel_test.py,sha256=vo1Fut9f8AuyTGpzqLmAg94EDBo3ENuKvpqoQS5kENA,5630
tokamax/_src/ops/splash_attention/splash_attention_kernel.py,sha256=jLjlvvhDJfdC7vr-BGGLGXZQYL9G-Em0_4zYh0hN_Y8,116725
tokamax/_src/ops/splash_attention/splash_attention_kernel_sharded_test.py,sha256=_3PpVp44GaR2qV9WvUrSgsPbHwCnbDjloJo-bkaEIng,8089
tokamax/_src/ops/splash_attention/splash_attention_kernel_test.py,sha256=0cRBWAI-Zu_uMArFYRQYvT7xbyN5A1I2_NBxikawJ24,64447
tokamax/_src/ops/splash_attention/splash_attention_mask.py,sha256=8D5BGSx2uZZOhHSblz_BRIYASLN2hnOG0_ltRIpxegY,14950
tokamax/_src/ops/splash_attention/splash_attention_mask_info.py,sha256=kZzYqtn4g_gE69unD9BO6an9JRYAIVyOt9EVxRoL5SM,21185
tokamax/_src/ops/splash_attention/splash_attention_mask_test.py,sha256=h1_D1VJddrysXqNu5aSMjX7xsjekvLZPJr4AQF7AD3Y,54993
tokamax/_src/ops/splash_attention/splash_attention_segments_test.py,sha256=QDsvMwX2SL68mvMdGGZyIXwPhDN__XvMYVlj0ZtFP44,23111
tokamax/_src/ops/splash_attention/splash_attention_test_utils.py,sha256=hT_yGrMmmPekWGoKl8db7CFisWFk0hc7P3bAgHEwWaY,2780
tokamax/_src/ops/topk/__init__.py,sha256=T249jb1MwC6WFuwr1-Evu-bDB4OG_3NmhYqqkvhpbwA,676
tokamax/_src/ops/topk/api.py,sha256=bgwj1-KLaSwpzeo1_dvLuUia46IwdyK9-A0koA72ZNo,3377
tokamax/_src/ops/topk/api_test.py,sha256=YvV0oV40O1zZBupHT43fbdL5DRmNP9QpXueS1xzKjTo,2301
tokamax/_src/ops/topk/arg_specs.py,sha256=PIyp7o032xM5qCHEPlJjFBl9zO8FuN4CfG2L-zR7nuE,2158
tokamax/_src/ops/topk/base.py,sha256=MXrKwOOrCvgrCmhLiFbU6Ja7CNLTGCJlT9F_eVJJRfY,2649
tokamax/_src/ops/topk/base_test.py,sha256=w89OfBe4eGrv6CGM4aubwDqVzKxYIBOsmg0HWr3Z4aQ,1873
tokamax/_src/ops/topk/pallas_mosaic_tpu.py,sha256=hyk_7TaNJDVcG2pnDF7xvVSCfSXB1pUEHRechHGia8M,2682
tokamax/_src/ops/topk/pallas_mosaic_tpu_kernel.py,sha256=eCxkiNQZ5tVfDirwzK8LmbhhBm-A628o8-Lfv-XYmcI,19418
tokamax/_src/ops/topk/pallas_mosaic_tpu_kernel_test.py,sha256=3kiW6fGUh5vSt6ZTg-rIWdIE9QIvEon6_RyMXgG5KU8,14283
tokamax/_src/ops/topk/pallas_mosaic_tpu_test.py,sha256=upuraQjDWdBG26Ll6YTmHJ2ZnXVxq1A_HVgRqfWKVzY,1395
tokamax/_src/ops/topk/reference.py,sha256=2FMQc62K5Lxk_6jqwA4GBEJwIN3Ni6z-i2H1a4dRJz4,2797
tokamax/_src/ops/topk/test_base.py,sha256=YHjFPZDTqV1Ysz7LouPNL1sORcSwQop7WRQpMvWGFVI,5920
tokamax/_src/ops/triangle_multiplication/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/triangle_multiplication/api.py,sha256=UkE12zCsM9IZaL87O7lCUGMb454QMCpQdSZ4Awo3TEg,7151
tokamax/_src/ops/triangle_multiplication/api_test.py,sha256=zQROYkg5dAKqsr7hjNFZFOdssPGbaQM4CuGOETPxaOo,10697
tokamax/_src/ops/triangle_multiplication/arg_specs.py,sha256=BN9SSgfaCd-2Bz_CQMGbUQADuRx2iMn9heeWPWe-WM4,1931
tokamax/_src/ops/triangle_multiplication/base.py,sha256=iFFLKVaehLpR413IQ5JBrYwlv9_43f8FH2vVSwU8D68,9882
tokamax/_src/ops/triangle_multiplication/base_test.py,sha256=yFKsrNMev9EfSoVL8K4HvFyFK86qGi25owy6-3KCRrw,5896
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_gpu.py,sha256=zsg8_dN9qWulmMOmYqCJVDFkbUGlMzf_wMhdSlLgV4E,7258
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_gpu_common.py,sha256=G_s5jlHZMg4PPlXrxY0Fe7ced3lGvDy5I6W9tNvE-SA,19791
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_gpu_kernel_sm100.py,sha256=JJQbxfSsg4jUK-4oX8JYkKK3StGqLjKZ8hhHdhI-GYU,71313
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_gpu_kernel_sm90.py,sha256=D39EYANLAzwNqXKPFjAEXaYBOSBYFRzLOwMOpPe2d6M,53647
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_gpu_test.py,sha256=Zxx1abiCJwLTfRhp8HjQ8RFd2y4DrRoLqGJ_9vqK0is,28966
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_gpu_vjp.py,sha256=Jrq-VaL3R3YH25nup80yJn1GScDWdeJ4bnF-XDlGwcs,9322
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_tpu.py,sha256=f7z9prVZMDZ7cNhbw4TiO_Mumwdf-MiOqckR1c9OVj8,6085
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_tpu_common.py,sha256=3E3g9SRE5UsJenRcy6OufMcBzfF6G4atHWLR1jJinl4,21670
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_tpu_kernel.py,sha256=h_3rEjD2chRVMOisubE8-4a6NwDbTCKnpMK9Fj7vfrw,61047
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_tpu_test.py,sha256=jCaN3Y14e1l1nKTcMpGgbbyExlSnZLp7JGR8EbVxNBk,131785
tokamax/_src/ops/triangle_multiplication/pallas_mosaic_tpu_vjp.py,sha256=DnWS1KnZOudrMoUo0TgtnfA2LGAGWXFsenwCbByLmb4,7963
tokamax/_src/ops/triangle_multiplication/triton.py,sha256=KQxJhadY7y0NmnIEIbUD1h7iGlzraiNHgXsJoSm2ets,30959
tokamax/_src/ops/triangle_multiplication/triton_test.py,sha256=dJyCAApeaMcKjZd2668xKe7My0fQr7KdveQtqlVw_z4,23133
tokamax/_src/ops/triangle_multiplication/triton_vjp.py,sha256=NMuWtZztAWxseIbr5A6Tv2kf1UAj6jxBXOQmhHeOzPM,45076
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=td5TjqGj7WI8kosZ5av9b6b_JlKzJVVHGXLI4Y4kqhg,3763
tokamax/benchmarks/lightning_indexer.py,sha256=sy0bbH9cklZjKPcts88dC0f072zootLOAYhxo2NiarI,3525
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=7KLB0_682UwEhiDblQnT-vNPVd45LaOn5p8NJcjDhQo,10264
tokamax/benchmarks/sparse_mla.py,sha256=E7gwUGVmrzhCavhF9T3MjlClLHBqYaroel_rY3LxHk4,3516
tokamax/benchmarks/triangle_attention.py,sha256=-_UDevuRv3yBZesNugZJqHTYSRAlTfDjPxouprkJV6c,7008
tokamax/benchmarks/triangle_multiplication.py,sha256=XgeivcsmcqOCxzOJysJu-CSmUXQ3OnRkxMqqBxKFzUw,7115
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/dot_product_attention.json,sha256=2CMUbZUkLpcogjQ9aGzBLToNDRGq_AmFajhJ6oLek_w,154660
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/gated_linear_unit.json,sha256=v69-hI5rj8i3nz_tQawcnqexh8TPVXEsEgmMKugORZg,113423
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/jax_nn_dot_product_attention.json,sha256=vJ06XKpbF1XNzU7nEKWj7d62gEO7MZOs1CgAcpV4fEs,132913
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/normalization.json,sha256=EtHU1wN_WZxkytJVhho2bjWnn1tOh3OedGhtnhCLCeo,4816
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/normalization_vjp.json,sha256=EFQXwFNKzYzNI331_3l5vav9IB6N0EQKnj20DHKdB48,2434
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/pallas_mosaic_gpu_gated_linear_unit.json,sha256=MfOt1p1N219Eq4wz7HjRQv8nsa7jVVHxRioLi5tbwi4,98545
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/ragged_dot.json,sha256=4QRFghu37v3PM6TnzysDWl5qNmSbID5Ewu_CYsDKQhs,55927
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/triton_flash_attention.json,sha256=reTzsRz0s5dh5b-JnN8m7S80O6efca6ty6Pwrl9OTnQ,206917
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/triton_normalization.json,sha256=O-C7zYu8x3h7HJ0McOJlZ33AIW5i7r6wAqzXwPqhH_U,5226
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/triton_normalization_vjp.json,sha256=rHQftC_-vmNpBbY7g15iV-gbiHj5nDul80zIOQ_mulg,3264
tokamax/data/autotuning/nvidia_a100-sxm4-40gb/xla_chunked_dot_product_attention.json,sha256=JbwQXwMQAiGqKF83tYHBWiY6eO_KHT7ou_BTUu-9sJA,139064
tokamax/data/autotuning/nvidia_b200/pallas_mosaic_gpu_ragged_dot.json,sha256=U-KOfB8WQ4FVjamhSXWRXOfLvNQGo8jA4hBgaPoQCwU,70111
tokamax/data/autotuning/nvidia_b200/ragged_dot.json,sha256=oH0kUrRXgj54iWOQ--ynFAwQTjbYdXa9GT4GUpl8Tsg,59655
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/batched_rpa.json,sha256=xtrZFYgM4u6oTPt4j3kbS8VCETrCSkrifIFevBG3LWA,82530
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/causal_conv1d_gated_delta_rule.json,sha256=QxdOq6UgdfVDfLvO-6MXJVGC6bOTsgWj5FqngsozSsU,65074
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=9ItKXKWXZDs3TjWxxi8BcUn7WWrSvqhx4Nsip-PTIOw,154799
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=FSDJLvqcDV7eYUDAcLLp4d57qT2mHfjSdFR2MGboGgY,130792
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/csa_gather.json,sha256=5dta11QDImMCn6Rc9vyEdD7FSo2idmJHQVhJ1u2Glk4,1470
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/deepseek_rope.json,sha256=n3-Xcnna51EFtffowXZ_UaqvjgGpaVfUZ2KEhgI0laY,2042
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/dot_product_attention.json,sha256=M6d1b1yNupmVyuYZp9K-OfmB7jUjTMP616LRF5_cMao,171892
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/dot_product_attention_vjp.json,sha256=3evWYyBFhzZ77qX2Kt32iX1l74wUoIRkjz8yx3KyN4U,117888
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/dsv4_compressor.json,sha256=D_uJCSbovCBZUZ0tOBcRghJSB7O6FdWCJde1Nl7L_oo,3001
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/gated_linear_unit.json,sha256=n1bZmFbdB4ACrguLmMibYZf4tuHdTrG5sG5oKAGlrS0,131887
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/grouped_lowrank_proj.json,sha256=MeABaLfkF-hdqeSGG_2RE0KMvpxaF3m_cKr0SQhCDzM,1617
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/jax_nn_dot_product_attention.json,sha256=XpQZNRpSgipzjHPjlv_CZaa4JiYDk_A6DADgDgXVOqk,156690
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/linear_softmax_cross_entropy_loss.json,sha256=ZuLa7Zg25Dd2IeDIRDfHfIFkzz8iQS83QUHolUVkoS8,135749
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/linear_softmax_cross_entropy_loss_vjp.json,sha256=mmcqdJPEu0d__hUxm6oQEhIdVfcP4DCF3LA1enStkKU,105143
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/mhc_fused_post_pre.json,sha256=awdSmV-s1BhXfuHefb4GoZUlZJF9lFMDSQrnNoVbVnQ,1963
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/mhc_post.json,sha256=-f-RbUjf4zBsbSJFlD-_-EbUvL1du_4LBrdaPsFnY_Q,1407
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/mhc_pre.json,sha256=YyhPiQVi28AM6UL_1AtKgxKSujHsk91G61-zt0mdbE8,1670
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/multi_head_latent_attention.json,sha256=55iVUl-4s9Vc5mVsBVXlJ9kVAtjU-RPcnhk7ouSve6Q,2399
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/normalization.json,sha256=wLUycWrDYXQvccIVdr18ErfCtUVqCt5H-OIMqeKxcaw,5272
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/normalization_vjp.json,sha256=Oz7gUFw3wUjRiB7bbnUV-GV2fkwzigxZA_Td5jdmzi0,2416
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_flash_attention.json,sha256=qVSfTTpvk28VdhDYuOg_VAT5uR6hnOQMN57N5N3M4ts,204141
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_flash_attention_vjp.json,sha256=duv2eEzzv_9yTcCyXszV-7YqAveI0Gkt8Eu0LTJh2co,139644
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_gated_linear_unit.json,sha256=WsymRiD9dbYkhccXFKmtCSVtKjpIq67vi5XkjBa9q2Y,129473
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_mosaic_gpu_ragged_dot.json,sha256=fXSy9jleFh13pNmCoDFi5U5ADRFCEzIvuVxDQB2fVjE,121814
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/proj_and_save_state.json,sha256=UorYERTt_Dnb9zFMz33_PGApHapAe9Xq0o6DIFu6Ncs,2210
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_dot.json,sha256=aPJIRpdYdRYRE5qrReA6rUM8qjTUwfynUbQxW_jhByk,83571
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_gather.json,sha256=LIe9Rzf8hd9-IQuyiTmsbx8A2WTSm-7TiaV41nSw3bQ,4224
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/ragged_gather_reduce.json,sha256=OmriL8B8YlAjOpKPY10x24cnAHiIJ0Z1JoyjxbiZQBE,2369
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/router_top_k.json,sha256=LGQAJZ8bt0A0CvdEpYPJYO4QdCAFE6pNLljEtOrX4DY,1489
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/splash_attention.json,sha256=IzV6_mXGpWv11F2GyhRYzuju81BW76Yb2pWw5PRG6ZE,19386
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/splash_attention_vjp.json,sha256=GADcQovhn4WJDHKJAbS0wFp2kf-OWftuT899qp4bCDs,11417
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triangle_multiplication.json,sha256=C_At5fvgpp2N6dgRmSgYO9WhJ4MybF17E4OtcBqffzM,13025
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_flash_attention.json,sha256=MzePzY-nHWSsodlr-pZIz2zBxIzuW6lKK3Hz5o9msrc,206838
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_flash_attention_vjp.json,sha256=G3ywnbAYUKHcTtv10JSmizHUNy20VOFwBQGeGoSIY94,82198
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_normalization.json,sha256=NJ3reixmG-8iXRS1w8-avTspe1Hp312CK_--7T8SZOY,5363
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_normalization_vjp.json,sha256=HZruCDQpJRLo2poVQEk-IC-t031oeTKRT3SIBHsdJzc,3249
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/triton_ragged_dot.json,sha256=UYQ3p57RO-vt-cSuT1LBAjVLqedUbQmW1hY9ejUfdFQ,113763
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/xla_chunked_dot_product_attention.json,sha256=g7AtSpBVIvnI4jo52h7EvDa4p0NEDO102CWEiPaV3Vw,156385
tokamax/data/autotuning/tpu7x/batched_rpa.json,sha256=0rdiRH75egVL1aP9_yr80dGIpJP827n8r0xQZ1sFBVs,89491
tokamax/data/autotuning/tpu7x/causal_conv1d_gated_delta_rule.json,sha256=wBbPlMBtWvvtdLc14FO2mWCwP4bZP2Cyiq7ugO_C3GE,44776
tokamax/data/autotuning/tpu7x/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=U9aryLBk5dAnaciZwvN_LoDJU0n8vm-KofqEn_Wkao0,148634
tokamax/data/autotuning/tpu7x/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=hgwdCO8-SZz2VR8NzNAUsezcRllHOJ3jD-bFsSuq-fM,85931
tokamax/data/autotuning/tpu7x/csa_gather.json,sha256=UQrICpkw-KCfjaqg2NAdbZodwBnm0TLDmF5bs8DUGC8,1376
tokamax/data/autotuning/tpu7x/deepseek_rope.json,sha256=zIQXA-DBlvCxOBN7kp6DLzvf_DT0dV9GPqsksTK68q4,1945
tokamax/data/autotuning/tpu7x/dot_product_attention.json,sha256=sfNirlaZQOT_7x_oaJzssGVUkU6psYor4ENDAVUo9I0,166539
tokamax/data/autotuning/tpu7x/dot_product_attention_vjp.json,sha256=Ro9iueXL72Nd9cWTJX5BAWYJC-S0M2sjk65C9pr2_aw,113771
tokamax/data/autotuning/tpu7x/dsv4_compressor.json,sha256=oETvyD-MW3w8UCUVOf6_T2r-8DhX7A3sPkTiZNs2Vpg,2871
tokamax/data/autotuning/tpu7x/gated_linear_unit.json,sha256=Kud3iB6UMMUKNJ5D0MyuBbCUO9dBhx106zBYR7MMSLw,120192
tokamax/data/autotuning/tpu7x/grouped_lowrank_proj.json,sha256=fldaH0VsXc3mHJ1DKDayuQR6lZJ0ufiNLKTXQP3D0m8,1564
tokamax/data/autotuning/tpu7x/linear_softmax_cross_entropy_loss.json,sha256=uHUXecHgiexovWrpHdF5wdSr2xJhtaHbvbL2vQc13c8,131794
tokamax/data/autotuning/tpu7x/linear_softmax_cross_entropy_loss_vjp.json,sha256=oBjpnxvRnR9PJ7kS64urDywOolWfn1AoCGdQO-RON0o,116534
tokamax/data/autotuning/tpu7x/mhc_fused_post_pre.json,sha256=I6U0TCzTSVJxFfVXOBr1aB6JxxSpTGWLKxH9OvYD0gE,1883
tokamax/data/autotuning/tpu7x/mhc_post.json,sha256=E_pj8Ay1Dezl6S5O0ltkDDUHUThLowHpIK5Ypa8Iq1M,1326
tokamax/data/autotuning/tpu7x/mhc_pre.json,sha256=GGNdQNAEbY3e4qPTswnMzqqY8UXMJy1VgDa6ovNvv7c,1597
tokamax/data/autotuning/tpu7x/multi_head_latent_attention.json,sha256=qKaexY1S9HqjQff1YBdgDjLTawQnpl0jb2zc9wbN3Ec,2339
tokamax/data/autotuning/tpu7x/normalization.json,sha256=JfzLw80F2So-Ir4fwjE94IR1VtCzBJViRVtDwj5H7Vs,4637
tokamax/data/autotuning/tpu7x/normalization_vjp.json,sha256=VUgS55frODqvt3XVwpSZ1J683X_laMbubkK72TZyC_U,2363
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_causal_conv1d_gated_delta_rule.json,sha256=DTZ0uJBaagYFWENvFcxiLKmzbJZus0dfccHCfjELZPE,54539
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_flash_attention.json,sha256=rKpO-GQIV05lftabFJiC65msOVjOcRaLz02zU4l6La0,143875
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_flash_attention_vjp.json,sha256=zsQ-CQUWPAzlKPxFsIZEJoOX8Fo1q6o5tRnLS8pKYLY,94848
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss.json,sha256=sn-6dKaUSGVcp0vh872v0Rq1zwPLtVtpl6O-r_i95bw,163074
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss_vjp.json,sha256=2Upt2m0pWEVcYYiUxm9h-JzE4vbs4QB9n3zg9Rht6UE,91588
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_ragged_dot.json,sha256=PitISK7N459sDJjrdSHR-hpMp25eeoCZjY449oGyIOE,119166
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_splash_attention.json,sha256=eGnuAgvUH5thoGMmQJQCDoMN5cNjLTroh6oxQjFXdUg,50024
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_splash_attention_vjp.json,sha256=_h588ZT4HnBCVnsLcwYcDH19WchwHJm7z7l9Qa8RYrk,26278
tokamax/data/autotuning/tpu7x/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=-N6DJkTiyuPavwTWbYIIfdMkk0yJ0Mfj9CtjYwabf9s,111687
tokamax/data/autotuning/tpu7x/pallas_tpu_batched_rpa.json,sha256=oYV_31Js-NI9GDskQ5-8tZfwakR44gjrjKQtt1LAg8Y,113726
tokamax/data/autotuning/tpu7x/pallas_tpu_csa_gather.json,sha256=6b1O8ojrcQm8nA8EeXyAPvpp2fbTxOfLyJoqBSQeAXQ,1427
tokamax/data/autotuning/tpu7x/pallas_tpu_deepseek_rope.json,sha256=2jcje1cX_720eY8dOTJ-m__8lP7mb0E-rPbTVfRuhL8,1998
tokamax/data/autotuning/tpu7x/pallas_tpu_dsv4_compressor.json,sha256=SeNbzS_u87uXF1YYeR_rOran6WTTl_um9UmtKuaxcf4,2912
tokamax/data/autotuning/tpu7x/pallas_tpu_grouped_lowrank_proj.json,sha256=NPWnObgwlZYaOPAjoxDgLttBzL-ZyZ6tqQOW3gYwc7U,1698
tokamax/data/autotuning/tpu7x/pallas_tpu_mhc_fused_post_pre.json,sha256=b3TCK9uSEgn-2utWGzod6fvm8D66IWYHSxvSjX_6r8c,1951
tokamax/data/autotuning/tpu7x/pallas_tpu_mhc_post.json,sha256=gInuiiDPzhUSPmcc5qCd-I3tdDyVet-PrbKD3en1oI0,1397
tokamax/data/autotuning/tpu7x/pallas_tpu_mhc_pre.json,sha256=1_XKWnhYqplTy5FeM7Pjh8UkPyralzEK-GchARpfWbU,1670
tokamax/data/autotuning/tpu7x/pallas_tpu_proj_and_save_state.json,sha256=C3l7G0slzv7d4d5557nw4kBPBHj1msDzbBD6ajnfLTg,2222
tokamax/data/autotuning/tpu7x/pallas_tpu_ragged_gather_reduce.json,sha256=l9EHPv2yudR1P7pr6ucpmQVj1iD78Zz-5Jzrkmnhvdk,1464
tokamax/data/autotuning/tpu7x/pallas_tpu_router_top_k.json,sha256=-pBnLAXpSX-lesjRtD5x_iDuNGWsKfkgXTUF0DVRkls,1477
tokamax/data/autotuning/tpu7x/proj_and_save_state.json,sha256=3-puiaDEeQG8iZXNsTS8RzgJgLUaKEXAGPlNF0P9aMk,2098
tokamax/data/autotuning/tpu7x/ragged_dot.json,sha256=SLjJNQF10pMy6gnPd0j812gCPHIvcRF_xp79hlZq-lY,111548
tokamax/data/autotuning/tpu7x/ragged_gather.json,sha256=OIGm8BqKL9zBKWTctipngO1RdrpcbYh5bsCcOrrwc-k,4076
tokamax/data/autotuning/tpu7x/ragged_gather_reduce.json,sha256=1hih-HGTrW8t3S8KAtJlw9eUCyLsX4wqNwDWbzhOCFQ,2295
tokamax/data/autotuning/tpu7x/router_top_k.json,sha256=Sim0N-vXZn_0w7mjEOHklWFpTQ_93YrkYje6TFBgJl0,1411
tokamax/data/autotuning/tpu7x/splash_attention.json,sha256=YQZyFsxXJTC3wkhqcavtV6VdlLJBHn5s2rhbNBoGBuo,3577
tokamax/data/autotuning/tpu7x/splash_attention_vjp.json,sha256=1aknf_-iD7KmEs8RHjZhJxpN-c_5P3ZiL6Lxo_yVbns,2137
tokamax/data/autotuning/tpu7x/triangle_multiplication.json,sha256=HgHMT-NCSsHP-lZNFEOganPgV0zJYYVXdIsMtcJE3os,12738
tokamax/data/autotuning/tpu7x/xla_chunked_dot_product_attention.json,sha256=STo-q8wRDwAVex7V5Qs5NnE4YyzTWsWSnJmnylQgpuk,144791
tokamax/data/autotuning/tpu_v5/batched_rpa.json,sha256=9JReye7caSR7vcVjrOaya8ggUlZBkXAzqi8XqYm7HNo,8474
tokamax/data/autotuning/tpu_v5/causal_conv1d_gated_delta_rule.json,sha256=Cl5YkD_-cbgf_ae9lHfHzoe3FiK-Mjzu1zlt9F4FxP0,25731
tokamax/data/autotuning/tpu_v5/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=ep1CUEYZNmhW7clavQZov3RTqwMGEz-GuitoZgnFfIM,117397
tokamax/data/autotuning/tpu_v5/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=B2jrShBOfGv3sxhNbyNdlmG9Iv_rxwveyP6vYHKw_D8,67562
tokamax/data/autotuning/tpu_v5/dot_product_attention.json,sha256=vQXxE-s28f6b-AjhNpMLZaXvTjTEkdIgsVR47vd3mTQ,159810
tokamax/data/autotuning/tpu_v5/dot_product_attention_vjp.json,sha256=9m_5E8o-VwyawTqcS2rsIPZznLPUM7eWbZW6mMDQ8tQ,45283
tokamax/data/autotuning/tpu_v5/gated_linear_unit.json,sha256=cQqkCrr8IH8UZ6IEK2Q8II-T86PLWvvHm_5n5iKNssE,121636
tokamax/data/autotuning/tpu_v5/linear_softmax_cross_entropy_loss.json,sha256=KExjGMfhPPKs5lEkP8J8VrVhrjVTYO50P8PpU9ePOF0,106645
tokamax/data/autotuning/tpu_v5/linear_softmax_cross_entropy_loss_vjp.json,sha256=YbnmZ5c65O02k4Q7avdigK1vuFoqR5dYZADnTXZYM3Q,62129
tokamax/data/autotuning/tpu_v5/multi_head_latent_attention.json,sha256=lBG8Ce463QEZcjw9LO1YYLNXpiOmX8y7xOSJsxS1_lM,2341
tokamax/data/autotuning/tpu_v5/normalization.json,sha256=ni2Lq0LbYE7GJ0Bc-F_2e6gA7XZ7fc23JeaSZOSKv5Q,3710
tokamax/data/autotuning/tpu_v5/normalization_vjp.json,sha256=Oq0LwdPVzuNLw2Z3q_wuqBM8a77BU0-vOGdxEZ_CqwA,2337
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_flash_attention.json,sha256=BmbHDDfHLVIH14I6Ds32lwSSDB1ACESDQuJzOtlVb2I,139426
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss.json,sha256=v53qXqkDyufDDic6PFpXxBaTHDAAC9hzYK0vcqcHaEk,128518
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss_vjp.json,sha256=ryfq9L6ij-xl5BNyyc6OTJIO5seOSY9Hc-wZvsN5U_c,69601
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_ragged_dot.json,sha256=UZdU3jTovk57Jr2RDaj5Aq0VRAy61vrqgI4aJGKcEmQ,111695
tokamax/data/autotuning/tpu_v5/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=SfwSn7hB-EACaAithbM3sRUG5GmH-mM9X6SaS3RRskI,93127
tokamax/data/autotuning/tpu_v5/ragged_dot.json,sha256=jxKvKpae6On_WXCPjKfzoteWBOFBUFIiwJe0k0LoY4g,123311
tokamax/data/autotuning/tpu_v5/ragged_gather.json,sha256=i_YCc-E4fYsf76l33XjCmUd2A2WI4EvOjJUk8ELYFyw,3255
tokamax/data/autotuning/tpu_v5/ragged_gather_reduce.json,sha256=CnTd0oUVPAXrT7AL1wgWj7Ts1EUzBhRBDRFPuRTA5Po,1835
tokamax/data/autotuning/tpu_v5/splash_attention.json,sha256=kh6i6BoLdw3B3-KAaswELtQuWLKIhIhF_QouwIuzn0I,2984
tokamax/data/autotuning/tpu_v5/splash_attention_vjp.json,sha256=iJRczLbwoLTbwghACu9NOO1pv4Jv16ci03xySPimVMU,2144
tokamax/data/autotuning/tpu_v5/triangle_multiplication.json,sha256=PKqmhIDUBquQppbhWMsDNpQr8ZjGjqo2c-l-JQpUExY,9523
tokamax/data/autotuning/tpu_v5/xla_chunked_dot_product_attention.json,sha256=7rZ9BIi6tm92TOvu2B48F6nTgGmmP2Co7tTOKKtKbaU,138008
tokamax/data/autotuning/tpu_v5_lite/pallas_mosaic_tpu_ragged_dot.json,sha256=ZSGr6YJxQvOaNW5XsZi2bwIdjmTGS8x3AGBjx4VcheI,36354
tokamax/data/autotuning/tpu_v5_lite/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=AB6uzRLDKy5dL9d68CcdP30wtpcITpR7EgH42ff8XN8,498
tokamax/data/autotuning/tpu_v5_lite/ragged_dot.json,sha256=d_0WfKfk2ajwsYIDgQ3CG_JoBfnihE5rPLxkUqr-2K0,155947
tokamax/data/autotuning/tpu_v6_lite/batched_rpa.json,sha256=Fyvlml0tCqSDGXNzWZ3JdgEdCBzbaaml5qZQpRv787c,10033
tokamax/data/autotuning/tpu_v6_lite/causal_conv1d_gated_delta_rule.json,sha256=SkZ7DUdlRxueg1fuPLTmHSYaBfTEKcGls9KicSJnFzM,30642
tokamax/data/autotuning/tpu_v6_lite/chunked_xla_linear_softmax_cross_entropy_loss.json,sha256=OICV1uapy64VHkdNfHK8ZVIp79vVD3PPiDk-IntJEac,149039
tokamax/data/autotuning/tpu_v6_lite/chunked_xla_linear_softmax_cross_entropy_loss_vjp.json,sha256=0EcRPrc7PjlXe6spAXL5sfiynxSbTiJhLdqB1RCZHaM,85550
tokamax/data/autotuning/tpu_v6_lite/dot_product_attention.json,sha256=dUmfCJhYwEZzvMD4bGA_j0DBwkVOyZUpqKzPwyK4LzM,165732
tokamax/data/autotuning/tpu_v6_lite/dot_product_attention_vjp.json,sha256=SZsrFShM2x_0XeJnPVUWg-Oqjo-7QYvDYHJcWe-lrlU,55230
tokamax/data/autotuning/tpu_v6_lite/gated_linear_unit.json,sha256=Udv01hgx4KDg6EXc60iZRsQg9Mhq1A9tBWeDLHIxOXw,104641
tokamax/data/autotuning/tpu_v6_lite/linear_softmax_cross_entropy_loss.json,sha256=-v_UwiS6b0GElvpLRRsW4G0tt7ZVyi42833UrLCmqCw,118127
tokamax/data/autotuning/tpu_v6_lite/linear_softmax_cross_entropy_loss_vjp.json,sha256=VPU_UZIVEC8GXTz8jkvsN-r6cQhqKXOSWCoMZMa5_HE,68047
tokamax/data/autotuning/tpu_v6_lite/multi_head_latent_attention.json,sha256=PRljbjfJun6dEutlqgNRzMz7zxH2rMl-gx7PgdQ9p1g,2341
tokamax/data/autotuning/tpu_v6_lite/normalization.json,sha256=4rsbGy8fsVqb1I1usBAXJVky6yg5Sy6ZJLtPrLB2qpk,4630
tokamax/data/autotuning/tpu_v6_lite/normalization_vjp.json,sha256=1DspvOteCu_WMRrUdwro3UcU6FE5yQ0sTOhfjdAMdvE,2339
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_flash_attention.json,sha256=UoGWi9kFdR3d-wkPTGdk-MrxGr40SBRVh-hf2GuC4v4,145136
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss.json,sha256=28_iJsfQ7Ccp-aejC3AP-YkKT93xtPvhuAqJakHv0e4,162109
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_linear_softmax_cross_entropy_loss_vjp.json,sha256=9qOQzRBki1BJMcVqjAYm1h6M7zM-RS1KyNfspcne6z4,78788
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_ragged_dot.json,sha256=-JW6cEQvU-IiLRO-Or88REDoE9aMxL3BF7u9nhC37aY,107843
tokamax/data/autotuning/tpu_v6_lite/pallas_mosaic_tpu_v2_ragged_dot.json,sha256=uUlwc_wgwJCixTGkaqxVtsxpntit-yf3OMGASymwHy8,78437
tokamax/data/autotuning/tpu_v6_lite/ragged_dot.json,sha256=bjDivUalkG_m8GLxvbXMafQsFca8H42HFI3CGR2sp6Q,117981
tokamax/data/autotuning/tpu_v6_lite/ragged_gather.json,sha256=A9gieprmFSVsmq3Vug90JzJii68_rgzo1_BCWHIpg5Y,4064
tokamax/data/autotuning/tpu_v6_lite/ragged_gather_reduce.json,sha256=SHDIaOI18J5Al4xoDnwsmtN8xETeyv9W4xJB5i3aspw,1836
tokamax/data/autotuning/tpu_v6_lite/splash_attention.json,sha256=oxn6ugVxClFQyaTnsQRy6Ati1o8-e48jyPWOHixZK5o,2734
tokamax/data/autotuning/tpu_v6_lite/splash_attention_vjp.json,sha256=2FcVMN-zLWre_vHKuhNS_8r6ohiOv45R8nR3eiXmYrA,1565
tokamax/data/autotuning/tpu_v6_lite/triangle_multiplication.json,sha256=8tHJ7WKtI2u1iSNNye5mPmyPS4ZpF2UAQhjWjKPihpc,12716
tokamax/data/autotuning/tpu_v6_lite/xla_chunked_dot_product_attention.json,sha256=uM04ykym-mjElx6BrSVt-S9-VskkAT-TBjC6v1kGs_4,154226
tokamax/experimental/torch_tpu/ops/__init__.py,sha256=6uetLP5WppG3kRa7gp0f3tDU0ubvVsfbo72xul6mHns,4374
tokamax/experimental/torch_tpu/ops/torch_op.py,sha256=_Tcc_RRrbSPmbkgUzZ675Gzparl_gMbSMtplf1D7Lno,23458
tokamax/experimental/torch_tpu/ops/torch_op_test.py,sha256=ezhCbm8262goo0-jSqkX_VoD55YNGZ38v4vSTyTonPk,40310
tokamax/experimental/torch_tpu/ops/torch_test_utils.py,sha256=coV8PeFYHNWYlYJgIUmdRfAbpyDsNfa9LFDwKg_d3ag,1505
tokamax/experimental/torch_tpu/ops/torch_utils.py,sha256=bnny9UhrMH3f40ihKUTV5F9SkZ9Dc3yjEfzvkYCAL1k,10936
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base.py,sha256=43ssgh0CQuSBE82mSzF0L5BZ-NbR5Sr36VEXP1_De-U,7895
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_base_test.py,sha256=JUyPnAbFcOcAxLAjgq3X28Gx7eIITlD79f7cyKIEj2k,3939
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu.py,sha256=Cn6dzGnDx0bVEG8uvYlaYNCh2HEnHOGRS4fWdLYbDnM,9958
tokamax/experimental/torch_tpu/ops/batched_rpa/torch_pallas_mosaic_tpu_test.py,sha256=Lqmkbh-CFv1X2yVDUyPU6o4d7jXbBYr39a7lOCld2n0,10187
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_base.py,sha256=tZxBu2gzDyT0MXiBNbDnOyinLokKbHTPiONnPIcq7uc,6133
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_base_test.py,sha256=GAq8fPOvg0N9J_3cIBD1l-GSHfdmyHPmFivHZbSw3AI,12697
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_pallas_mosaic_tpu.py,sha256=fmFmHyBMjLrHVu-Rdx7cKLg3_vo5wuhf6fQQPhik4A8,9692
tokamax/experimental/torch_tpu/ops/causal_conv1d_gated_delta_rule/torch_pallas_mosaic_tpu_test.py,sha256=RPMf3rzr-NlvqtIJaNATJAA5kM0NcwIaQVF24umGkmw,17068
tokamax/experimental/torch_tpu/ops/csa_gather/torch_base.py,sha256=61yhigr1fIDNTyu3deTCP6LVBpGUWvk6QL7_Rphv7vM,3384
tokamax/experimental/torch_tpu/ops/csa_gather/torch_base_test.py,sha256=8zuCeRyCIW7uuj0jqETyjsYLOtA4WW6dWbt7q-63T10,5360
tokamax/experimental/torch_tpu/ops/csa_gather/torch_pallas_mosaic_tpu.py,sha256=tMI28CgKhpMS9rSzdg5nvnY8IhR2CP_aSanF7PUN74s,3752
tokamax/experimental/torch_tpu/ops/csa_gather/torch_pallas_mosaic_tpu_test.py,sha256=OkcKnqOEZQU60hgMC55NJODNNazAyOXlTdLzK8DAEpI,8704
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_base.py,sha256=9ud3J5YzqucTlkXzwXZyklj8ksYvCZl-TSyTGljEqhI,5649
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_base_test.py,sha256=G9-FjKJJW617o1IjsRT9oH4o28MBtJhq88eqzxIIsq8,10349
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_pallas_mosaic_tpu.py,sha256=5-0Lwin2oo1lyaoN3LOyQ17LgOyQG2527seaD3qFNJo,4802
tokamax/experimental/torch_tpu/ops/deepseek_rope/torch_pallas_mosaic_tpu_test.py,sha256=yMsDymQxtboN8m6W1fgIY_D8EhpdgxUIOjLz7Lvuq2Q,13789
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_base.py,sha256=YrBMRJt3RrAiwUWHauE76V75I-lyjvhGQwNjmyLj1S0,5658
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_base_test.py,sha256=dos2_mcVZl9J03u_7YXIbTQZkX-k62fk06BdY9IANlY,12550
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_pallas_mosaic_tpu.py,sha256=qJ1Nj1KzQ7trCbaPO15BPUGlV-x7rqTC0XoXSEZkj88,5641
tokamax/experimental/torch_tpu/ops/dsv4_compressor/torch_pallas_mosaic_tpu_test.py,sha256=mTTRCRmRx2Ooep-_mzoWSIoUfxWohDy1rJgegLtlqbA,17613
tokamax/experimental/torch_tpu/ops/fused_moe/torch_base.py,sha256=xsfPKsY76B4gzIWC7NpOiryvg401nwijPL5Vm1ts5CA,3133
tokamax/experimental/torch_tpu/ops/fused_moe/torch_base_test.py,sha256=X6jtXYJbAFOKcGSaJJ860KJNjx1keGrDi-x22xBejIU,5109
tokamax/experimental/torch_tpu/ops/fused_moe/torch_pallas_mosaic_tpu.py,sha256=vXiXxDo_FNYe-yQWkc_cIL1SWPlyVRT-RU5HR0W-4cY,4975
tokamax/experimental/torch_tpu/ops/fused_moe/torch_pallas_mosaic_tpu_test.py,sha256=ZLMfahf3BSsoy5moaaOfEh_4TgDfY2Qbe7WPQabjFas,8407
tokamax/experimental/torch_tpu/ops/gdn2/torch_base.py,sha256=3i1sMHSl6_sOu_XijRLrwdFgcidN-Tn9j75J15LFU20,10687
tokamax/experimental/torch_tpu/ops/gdn2/torch_pallas_mosaic_tpu.py,sha256=cukxhvp8SMCY-WiYA1G4eG9b7F2Pk4mzKgEyzV6k5ns,7109
tokamax/experimental/torch_tpu/ops/gdn2/torch_pallas_mosaic_tpu_test.py,sha256=Anvz4rmv-vK6lQuJT-r0vQyGoGbdqPYxdS5yDOa3WJE,6249
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_base.py,sha256=OVV78HQT5ur3XPDm3GnTpqDY-ZWGTGlmLeo6HGMJqHs,3577
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_base_test.py,sha256=pgUDN3PhWcnXSLES3KK9hCEt4s-zuq-MTD85vXjkoAo,8774
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_pallas_mosaic_tpu.py,sha256=G6DjYrLLslYtE8LeTk8f9nLoOdY9pVKNKERVCsl5i78,4539
tokamax/experimental/torch_tpu/ops/grouped_lowrank_proj/torch_pallas_mosaic_tpu_test.py,sha256=fkfeWlmDMLOPEi6wlvMX2_lNxLlOrD-gg04qJKVXmog,14453
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_base.py,sha256=-Pe7uLhfemJKvhnc_hl10Ty1565-NgyBarOVkewOJV0,5502
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_base_test.py,sha256=u-JNNZop6DA37ZgvCWB6UNjS5nzgW-nJobSX4aRc6W8,11188
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_pallas_mosaic_tpu.py,sha256=Dc1hFsqp95Ss7KBjhL-tVuYDsbhH34D0ty5V-X13ZGk,7125
tokamax/experimental/torch_tpu/ops/lightning_indexer/torch_pallas_mosaic_tpu_test.py,sha256=nFj-2FFGA4ic_ug8oHP8wG-tM81xSNfaLWfP8Yiamxg,15859
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_base.py,sha256=t6VzHFTStBnbV8w78Tc9Y_TzS12nTwR6I_cjTelZBII,6977
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_base_test.py,sha256=qm56BmrTwXS5JslkYdIjfX17eagyxGyONg8qTXGQE-E,6015
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_pallas_mosaic_tpu.py,sha256=FTa-qRRAsF2OHBhR2uQzqei1uo5qxzlpMooNr7UPl0s,9530
tokamax/experimental/torch_tpu/ops/linear_softmax_cross_entropy/torch_pallas_mosaic_tpu_test.py,sha256=N-5RhKWnyOGmbWVKGrNTToNN-y9CuTuAUo3lFtL5hZw,15267
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_base.py,sha256=4o-ElTWzwKIVbp4mOnuL-u1jdkr92qrug5wMMlFl00E,4967
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_base_test.py,sha256=ygOK0fZNom2-jqdbMG3JeLKOlOWm5BQ6M1_iGqOnvPg,12160
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_pallas_mosaic_tpu.py,sha256=3UzUnOQnqKNZERMrd08ouTSQaTuxCV8isdcMOt2oZVo,6083
tokamax/experimental/torch_tpu/ops/masked_dense_mla/torch_pallas_mosaic_tpu_test.py,sha256=OF6O3IgP6P40BwZn23r3GOVhPcgfbhnF3IwxhISkJBk,17863
tokamax/experimental/torch_tpu/ops/mhc/torch_base.py,sha256=B5sr_ITae5ox6jUx2QHIAZnVN6hAk6mPg5tM3W1rhVE,9874
tokamax/experimental/torch_tpu/ops/mhc/torch_base_test.py,sha256=4YTJeJf-QhgeFDkJIyAjWCNXzqm2ZF9Y0soty1w2KpM,15916
tokamax/experimental/torch_tpu/ops/mhc/torch_pallas_mosaic_tpu.py,sha256=sXLpfaRRTPMTS0xFPO8OAXlJZKiBO4vW_ZX0ab5tVAo,12121
tokamax/experimental/torch_tpu/ops/mhc/torch_pallas_mosaic_tpu_test.py,sha256=KceQw8ji0mUjaW_JEN0A4aVMOSoMpwHNMhkR8vTDIWw,25717
tokamax/experimental/torch_tpu/ops/mla_swa/torch_base.py,sha256=_oThkEp3lNpqJEbMZoKILt0tjvLJIIr9B-2FbuQOTgA,5383
tokamax/experimental/torch_tpu/ops/mla_swa/torch_base_test.py,sha256=00phUy8FRiJe-MIlmljKSniczeUBQWL8lQwBokaUtvA,12768
tokamax/experimental/torch_tpu/ops/mla_swa/torch_pallas_mosaic_tpu.py,sha256=s7L9EzDY95q9DSoCuyvFf9Wet9hL0X9njXqs6VbT4eA,6453
tokamax/experimental/torch_tpu/ops/mla_swa/torch_pallas_mosaic_tpu_test.py,sha256=YfoS8vwGXpzvl5zAykMg2u1A9L5fe7FQ7g77JodPZEw,21107
tokamax/experimental/torch_tpu/ops/moba/torch_base.py,sha256=_Y4u9SbPDtriqoheoQm3Kov3Qrns5-bbjTAsrTtePQM,8373
tokamax/experimental/torch_tpu/ops/moba/torch_pallas_mosaic_tpu.py,sha256=l7qv3GmfIeTZD37om4A58aJMWkqcJM3NnLc3CE7Ak9Q,5310
tokamax/experimental/torch_tpu/ops/moba/torch_pallas_mosaic_tpu_test.py,sha256=kXzQb2J4z3-J1Ac8Yh7hKY9NHT_3zBjsotmPFTD3asg,4430
tokamax/experimental/torch_tpu/ops/nsa/torch_base.py,sha256=XQnYZGklQ33QZjDJiOdY_BrrUU2iUgg6NxVwkeFSJxM,10944
tokamax/experimental/torch_tpu/ops/nsa/torch_pallas_mosaic_tpu.py,sha256=vtYVS0w6qvDE88o33G5mNfy6hvXL-dhWOzL1lmp5IFQ,7134
tokamax/experimental/torch_tpu/ops/nsa/torch_pallas_mosaic_tpu_test.py,sha256=mkHO80Yh7_IZ8HJWAMtNG6JbAysYe_Ou1zvmCAod394,9748
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_base.py,sha256=1oUUUT5QreMQGtF1Ndnv1sd8nOGHD8aMFFZw2sCOXNo,3645
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_base_test.py,sha256=QOJbh1LO976wHYsBtay6s8Srroy20_VafhO3XCMmZgo,9325
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_pallas_mosaic_tpu.py,sha256=j9u1pdv1bbKczI7u2e2ue1ViUIRTEPuWFxsz8Lux6qQ,4419
tokamax/experimental/torch_tpu/ops/proj_and_save_state/torch_pallas_mosaic_tpu_test.py,sha256=OtcvJXdk3t60pWHlOeMUtgx2c6mS_QcMjtXx2uRpFFg,14395
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_base.py,sha256=-mTSX3Ci_H50idrjo7Es-cmnlBaavwPvzpOtqTIQpvw,20667
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_base_test.py,sha256=T3_83MbazqGmDwTiZjh-8i2kHftQAyFavlHxNTufYSM,8559
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_pallas_mosaic_tpu.py,sha256=bM0advuu6hU7N6iJP6mVO4OlXrWA_8N6whHgFRhKMiY,8733
tokamax/experimental/torch_tpu/ops/ragged_dot/torch_pallas_mosaic_tpu_test.py,sha256=0XgiSThP_JesBVL5UOuD_Z9JG8i4cbHxPgx50_4rFKk,12290
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_base.py,sha256=G8HivJMqc2kxSs5xkrL6GRUmqKyCE0V9wWudzyBoATI,2912
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_base_test.py,sha256=QCe11ftSKcypkVRfTTKz5I7odh6Yq-wZxDznVLib_qo,4439
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_pallas_mosaic_tpu.py,sha256=w0iRZzIenyHcQFK6RHRMe5pKnazySUNDULUbcF8eSPo,4451
tokamax/experimental/torch_tpu/ops/ragged_gather/torch_pallas_mosaic_tpu_test.py,sha256=iXVGPHVNGJnvZWTB0PmKH3WQs3edLiHryIXkRpfY98k,6940
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_base.py,sha256=ZONWwoBhFcwF-Jpajo1MnicCsyPoqcQ2emwjgCj-QeA,3294
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_base_test.py,sha256=V5dwXjxM9RYtfBVmdUb-Gl8voEFiSqXMhWd3Ldzc7gc,7007
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_pallas_mosaic_tpu.py,sha256=dTRC_tKiFgpmH0wUS6TzQM8OCoAiW8YS6ErUey5lyTQ,4006
tokamax/experimental/torch_tpu/ops/ragged_gather_reduce/torch_pallas_mosaic_tpu_test.py,sha256=hQDCExkqdicFilnCJlPbDOEoBfenkzNdeZM-8U1iFcM,13841
tokamax/experimental/torch_tpu/ops/raven/torch_base.py,sha256=0uTGMSrzk7iWMEC97I2BVOn-RdIAASWGipQ4-ZxeMXA,8702
tokamax/experimental/torch_tpu/ops/raven/torch_pallas_mosaic_tpu.py,sha256=VOmmDmnsE1Lw6-xnUNxkKABbJHbVLO_MBaxFnoFXU-g,4378
tokamax/experimental/torch_tpu/ops/raven/torch_pallas_mosaic_tpu_test.py,sha256=-OT6YChMOwaXC3Wyo_weoxSoi9pdWGm_8HST7oAUF_A,7032
tokamax/experimental/torch_tpu/ops/router_topk/torch_base.py,sha256=CIeYTtHxqlloZjXlX4wmQRJUQYRQJ9nVmxnWEQtOnsk,2937
tokamax/experimental/torch_tpu/ops/router_topk/torch_base_test.py,sha256=WqPCPF0rlbQPDEE7xr46mNl2z1OiaFKfG0k98hX957Y,4477
tokamax/experimental/torch_tpu/ops/router_topk/torch_pallas_mosaic_tpu.py,sha256=jES1D-WNZudAJt_X_avSDB10gTRqzam7Br7ANvCjGck,3505
tokamax/experimental/torch_tpu/ops/router_topk/torch_pallas_mosaic_tpu_test.py,sha256=8JkrWyhQv161CPJ1fzWmWNBOk-eJchcjPMpa2UGptIs,6529
tokamax/experimental/torch_tpu/ops/rwkv7/torch_base.py,sha256=oM-3kWeKp2e52qIKqHXNcgMgTYFuU0C00n4sMf2b3rY,11146
tokamax/experimental/torch_tpu/ops/rwkv7/torch_pallas_mosaic_tpu.py,sha256=xF_-DosiWEnFFzsMfn5psoj9xecWvcVquVKerndiEiw,8383
tokamax/experimental/torch_tpu/ops/rwkv7/torch_pallas_mosaic_tpu_test.py,sha256=t0Wp2VwdCAIh0O7WXHj93ygHlSngMI1b_nTVLOU8HvA,8040
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_base.py,sha256=i2CyhcdxAsdOGI3om-RYFPBZGjj2Cz80_Usa4A1nAlI,4336
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_base_test.py,sha256=fQqQPBbP4xwJqiIfxnnDUqejgQETJZQntuKBuqtq-pg,11978
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_pallas_mosaic_tpu.py,sha256=uA_w6_gD7L_myVuQBKOdIMko-umYbUi8H10hc2e4-5g,5020
tokamax/experimental/torch_tpu/ops/sparse_mla/torch_pallas_mosaic_tpu_test.py,sha256=NfWnZWdxv7nDAekpZqEp-U0jNT_I4lxz50bop7BHSjc,16695
tokamax/experimental/torch_tpu/ops/splash_attention/torch_base.py,sha256=rHr45LHcrW3ACEi6li7cgqtNfEY9MRPl_ZE5WJgxa7Y,21589
tokamax/experimental/torch_tpu/ops/splash_attention/torch_base_test.py,sha256=OAE8LcFouX6ywPizlu-ahdmDEIZz8YlM76uF9KFFylM,8811
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu.py,sha256=D6K7sZ1J9j1hEqrcVMCCPQVXmGXw38G7reQCmGJB5vc,9206
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu_test.py,sha256=_2N7M2LucITW5njjgEuHCKlki1g7NyPVJMiO38p0A7w,11458
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu_vjp.py,sha256=1Ytl0OgLrvna8PNqkPfqfMJC3lm0954ScsdXjNkmcc0,6994
tokamax/experimental/torch_tpu/ops/splash_attention/torch_pallas_mosaic_tpu_vjp_test.py,sha256=o7YdXpIy3WqqTOAUdHVHLfOYrzO8L_xTcfb-wfJBXoM,10007
tokamax/experimental/torch_tpu/ops/topk/torch_base.py,sha256=ppiKz9RR0G1ElMlYuwj_konTTfdaXWWIOeKyzwFLp1M,3552
tokamax/experimental/torch_tpu/ops/topk/torch_base_test.py,sha256=oJGGxob_U24ZMBzrg2b7-jqX_R5ZHGkHN7gUjB25uSs,4828
tokamax/experimental/torch_tpu/ops/topk/torch_pallas_mosaic_tpu.py,sha256=KgM4V6UQ3MznsLRaiA4ybZg7C5fFtaGZ6LufjJs8Z9I,4064
tokamax/experimental/torch_tpu/ops/topk/torch_pallas_mosaic_tpu_test.py,sha256=ea1Z7dtfSdsTTq4ywWPYd0xXxalgyqvXHNWDNn_JFH8,6892
tokamax/experimental/utils/tuning/tpu/ragged_dot_benchmarking.py,sha256=ve5LQUw-OdJWW0V7xkeJn-ryLAmgM7LvcxOClLhevu0,13063
tokamax/experimental/utils/tuning/tpu/splash_attention_benchmarking.py,sha256=1lYpYpckg1iXzZS_4jD3IiuOpTAGUnLh4J4rU09vFBY,10260
tokamax/ops/__init__.py,sha256=zvLyWSnUPP--RVoqNO8zoPMchj3FiBhobbBQUp24lOQ,4533
tokamax/ops/gpu.py,sha256=JvJrVD4vMsZxuM2lGBviCR1IJsQY7GGEDc-uyLx0e_I,2756
tokamax/ops/tpu.py,sha256=JP7j1eiwP5ofJexAv7OgGitUl2jWHnqX8kJyjQP-0cs,6506
tokamax/ops/experimental/__init__.py,sha256=9LZXOEYqVU-QE64NBag8UXS_aSpdJ-HON3W7rniNK60,1324
tokamax/ops/experimental/tpu.py,sha256=6gYJz-wWyZi1272znxKSpDjFmnheKEVhgxNy9bbc1Os,1947
tokamax-0.0.18.dev20261107.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
tokamax-0.0.18.dev20261107.dist-info/METADATA,sha256=wWsLPQXVw-T-zp5RemvhMg1J8U4kQOyvoEmshaAyNu0,17182
tokamax-0.0.18.dev20261107.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
tokamax-0.0.18.dev20261107.dist-info/top_level.txt,sha256=xHgO7qOuYjrLH3047bLQLnwrjI76i9-xlChxNQIZn-Q,8
tokamax-0.0.18.dev20261107.dist-info/RECORD,,
