tokamax/__init__.py,sha256=4mXPc4mfpGz9xNoJpbAK_SMSIIpUhaGKtWjfaAX07T8,2202
tokamax/autotuning.py,sha256=ozAUPHbs1_-t28etjF6pm1gKrBoXS65kziFLkQjXE6k,903
tokamax/benchmarking.py,sha256=sJxtRSAuHgW4JbGMyHw07vvUwRR_z8NRVYGVHCZrehw,1097
tokamax/config.py,sha256=FwB97rXW0P1VkMwhS-Mpti1LeBbdpC4l3AFcLAZXgKk,868
tokamax/tokamax_test.py,sha256=3egE2ygYoKF0Cb8PZLSMUgE5eo6Ek7TKDb2ctcATwQM,4269
tokamax/_src/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ad.py,sha256=gN-DnLTOWracL0cfa2dzq0Zc1Wqg7Z3-vY9ZzR926FY,4378
tokamax/_src/ad_test.py,sha256=5e-2DkE4i7qE-GBpoajwvUFuQAoBRrOWeids0AaYl2c,3701
tokamax/_src/batching.py,sha256=vWVlRYxXekVkm38nh0RkFpZY2WKNRV7CbA6dYvQU428,8049
tokamax/_src/batching_test.py,sha256=M3WGauph8yoMEI7agmRi2W-BWkA_osnXntftArDXRRI,4493
tokamax/_src/benchmarking.py,sha256=bNa8gADHOeWlaxZegP48KHarUINKRWAaGa69r_XhxNQ,17944
tokamax/_src/benchmarking_test.py,sha256=qehEcMYwZV_jCpaYqpBPp-6LhoMiLy13-w8_f-GSKLM,8292
tokamax/_src/config.py,sha256=SfrNOPUNn7INSZEighOqOGkGGoNpm8XltXha9go8vUk,2798
tokamax/_src/config_test.py,sha256=HuekTGgya8RPo2UYfrQfq8dM7FLonua7U6pUa825Vtk,1561
tokamax/_src/conftest.py,sha256=PC6zlS73bx_BB5pFvbOTB-ZASHZojUFrUReMVtozP0o,1225
tokamax/_src/hlo_utils.py,sha256=I8WX9f_yaSV6zvIWHaHv56-3gpfxJLIQPtcyw4DDX2U,8387
tokamax/_src/hlo_utils_test.py,sha256=6AHAIU3WH40EqUMU_C4cLFTN2JgEFZrjvumxJ6dvy5M,12615
tokamax/_src/jaxtyping.py,sha256=_W4COs9cj3GdNE0B_J2QNhn86tt2JPTUaVzp8g_rtP4,1389
tokamax/_src/jaxtyping_test.py,sha256=vDDdV7JyVds-SpyReKiqZr1-WbAT-3qRnXloYLVRYGE,1558
tokamax/_src/mosaic_gpu.py,sha256=98WSJRzAP-iSrYTrfw-R8d1GVKJup4aUSwnriXIcuko,1147
tokamax/_src/mosaic_tpu.py,sha256=Spz9iyiPUYZcDp0se0ShzWIrw4UrCHz43oTFH8CwUOU,10798
tokamax/_src/numerics.py,sha256=6teOewj8l9XLMagKPWFOAilN4VbIT953k2C76ZGnOaI,7908
tokamax/_src/numerics_test.py,sha256=CCbFN9QJfR5an5dpzzqbz2D0TsSbOoRfT_8L-qRtqeY,7154
tokamax/_src/precision.py,sha256=XRT401YjovNtpyeWwhiPVvSGHTLf9TvPHWuQpnoiWwY,7339
tokamax/_src/precision_test.py,sha256=Cpba1dPi8mKK22gfb4fCyAd1NE62MAd11Du3GUx5DRc,3584
tokamax/_src/pydantic.py,sha256=LChfpmvBnTrAAjYnAPPhNVH109-lU-Mqru1XLXaC-7I,10169
tokamax/_src/pydantic_test.py,sha256=VxpavdVO8ITmzbU15eKwXcntPYvjMXc81-T2-Pszcds,7737
tokamax/_src/quantization.py,sha256=1CPWZgDWX5UVTjVii_xqNW5Vd1AQ6b3_nKlYQl-1gRo,5741
tokamax/_src/quantization_test.py,sha256=cFEJrvZjat4TEjz2kERRxdA4J4C3fRVqtVbwbmn83TA,3133
tokamax/_src/shape.py,sha256=772lVlGbXfDgKiUrTSP_jlYKSPn0HSOsVIVwCExK0bU,2935
tokamax/_src/shape_test.py,sha256=8C6awpma4oJzPlcIpqwVvf5SSfKmATfghDA6hZkr5Os,1560
tokamax/_src/test_utils.py,sha256=xh9tc0LnKp6SrT1VXY34HU-FmZzwh0CmbTh5_eRkt10,3205
tokamax/_src/test_utils_test.py,sha256=O1457xfEX1w4v5X8tW8s7SLCWNWZ9l_lxz5tausF_48,2097
tokamax/_src/triton.py,sha256=mp-Hzqn07R55vuNe3JdfWikfE7VGiRGtnNGTeguXXMo,1428
tokamax/_src/triton_test.py,sha256=0Y-4nft57BwZvII3JZJTu6BjdCu893zj7D9hrcVAmhM,1045
tokamax/_src/utils.py,sha256=Hv8u5AztBwt8GK9YSn0PaDIlHoBZdXCKUxKg8zyZqqE,1941
tokamax/_src/autotuning/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/autotuning/api.py,sha256=W3XeCu7FtPdo18uwiWPews9CdSXOaJJvnyw1EZ650YM,11984
tokamax/_src/autotuning/api_test.py,sha256=hfepk2SSAjyi3bOEu6-lYyTitmgZR4Dypkk2NZRqjro,14545
tokamax/_src/autotuning/arg_spec.py,sha256=-dL63Jot8KLDQYXW4EylBAsmCe0MyFlpsWEXnEI5paQ,1767
tokamax/_src/autotuning/autotuner.py,sha256=0SSJjfZigfiMUlwaFUK9rP8muObAD9qOoclvqx4i3rs,6910
tokamax/_src/autotuning/cache.py,sha256=hz-TtRuZVilWTK6MqKTaUSLOoGmABRisgbB-tP87Go8,3472
tokamax/_src/autotuning/cache_test.py,sha256=W_PQWWC0YfX0OgOw_o9i9YOg03HUYGU29DML1UZi9wQ,1367
tokamax/_src/ops/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/op.py,sha256=I8660kXkd5ad-DWAt2Cz0plL9S3sKtVjfKCiU_1UKH0,23370
tokamax/_src/ops/op_test.py,sha256=gxVSLnzc_4y8Ic8EJWPsSU4MVlHg2PDyCBZlQ1zXR_U,7578
tokamax/_src/ops/attention/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/attention/api.py,sha256=WyO_Py9gS0AOtjRs6RXYyyRhzXhALfIdHn1i_7u9nxM,6128
tokamax/_src/ops/attention/api_test.py,sha256=cM5jT6qegkjU0rvcAH9Ksar9HlLUj4Oj6R4lusaYRV4,11873
tokamax/_src/ops/attention/arg_specs.py,sha256=F8_JuMoJ9LAgyhXAujGdR_bewHKWGLw-yUL1UW52njA,2902
tokamax/_src/ops/attention/base.py,sha256=_UGU9K7m1fgMXdMrS1ocmE830k0DSZODeq3Vl12OKVE,29196
tokamax/_src/ops/attention/base_test.py,sha256=L9gAlnSReP0oyNL-lpQ7i9yrXcyfMoPV2NilwdCYjCo,5286
tokamax/_src/ops/attention/bench.py,sha256=Ib7jPVvykBvrv9bh1x840w2aLSJ-f0Umc9h3Nz6ltCQ,2654
tokamax/_src/ops/attention/jax_nn.py,sha256=9n19wgwJnnYIb3QC9jczNANCuZ9aCUIpJXT1xC0_lM8,5746
tokamax/_src/ops/attention/jax_nn_test.py,sha256=LXUF98jnagpQPRDg8cIRllYTLN78W3LGG4EdeT_Xb00,3889
tokamax/_src/ops/attention/pallas_mosaic_gpu_flash_attention.py,sha256=YckXah0g5Q3GXlvD_9pkEcUyf1KAeJFleV35L147aIU,25149
tokamax/_src/ops/attention/pallas_mosaic_gpu_flash_attention_test.py,sha256=P3uCk2dTSYz2lpogxLM_NtxRNW_tiyQFKie0Jj7VPSg,5800
tokamax/_src/ops/attention/pallas_mosaic_gpu_flash_attention_vjp.py,sha256=4lL94pcQnn0HJSgKo6MFOHxBHLLtrfEM5Gs3EOw8MD0,30832
tokamax/_src/ops/attention/pallas_triton_flash_attention.py,sha256=MpNtyPA6wjqiMCYeyGICRyxHTsLqwF6Czfhne-UEoNA,23835
tokamax/_src/ops/attention/pallas_triton_flash_attention_test.py,sha256=8KX5gCXd2pxRc5ETwsu8Dhpv9L9dqnTky0CnrJ9xMCU,5784
tokamax/_src/ops/attention/pallas_triton_flash_attention_vjp.py,sha256=nGvCTjErWhn196aS7VHyhDsHeTaf8NoMx2BREsSoATQ,18115
tokamax/_src/ops/attention/test_base.py,sha256=QvWBuTGw5NuhHQxKcUBClmJpitI6HbE_z5tLEfDXq9w,37750
tokamax/_src/ops/attention/xla_chunked.py,sha256=RZHEpQxj0K5lDy1MpZLlkZ138Bjh2jkYzx6G_WRm4gM,12369
tokamax/_src/ops/attention/xla_chunked_test.py,sha256=NPpoiuhOB2qQHpQqMq848AOne-p3WyhI4re5JyZPTHY,8773
tokamax/_src/ops/experimental/tpu/splash_attention/ring_attention_kernel.py,sha256=xQxREN1LCAaY4MB8nLjhfGWo92G43JrnvialGKGtkgw,20739
tokamax/_src/ops/experimental/tpu/splash_attention/ring_attention_kernel_test.py,sha256=-gRa5H_WmZDrXMBCtbB8HItG8k845loDhIoVLB0xeiA,9261
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_kernel.py,sha256=-uSuuk6vFDTYTzIojtOTg76QfSRlKbymFP2Nm23Vpus,69096
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_kernel_sharded_test.py,sha256=1mMtGhyw-AVlmMF6qbnqAN7Jyr5FSTbRQcb-EuKu7iI,7171
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_kernel_test.py,sha256=WdqjXUxb3BBSSNjoRcZvROuOie90TZ_ASM4_7sd6rvk,22139
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_mask.py,sha256=oAYV0-2SHK6d4heY191G7s3QzJzLw96gF9iIUXfAwtk,14948
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_mask_info.py,sha256=dtilmpY7JiGT1vvCLi9AQx9HWI0oWtvgJTgI0NqPRlA,22468
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_mask_test.py,sha256=0_3XDuNBtSxVnGMzAl8c1mh74IXawhwsWmxxsJbPob0,48658
tokamax/_src/ops/experimental/tpu/splash_attention/splash_attention_test_utils.py,sha256=2eyOdAbTzCKt20qnCvSTCk1da0mxCBn8Rh9jCOQh8-M,2279
tokamax/_src/ops/flex_attention/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/flex_attention/api.py,sha256=Td5kwvuksLSyCRlPpFCAjkpZkryRmywjpoRc3_1p9bE,1131
tokamax/_src/ops/flex_attention/arg_specs.py,sha256=DECEEmyM-WzMoD7z8nd48JBDKUK9zyhY2InIT76AVaU,820
tokamax/_src/ops/flex_attention/base.py,sha256=YBCBaDpAmIYuNSAbv0WpJevJfykmAvGH65kJril-a5s,12199
tokamax/_src/ops/flex_attention/base_test.py,sha256=l4y8geGYw4AIJ_HnNOfXIyHXw2LKVc1xYEBlwNhIws0,1475
tokamax/_src/ops/flex_attention/pallas_triton.py,sha256=euGr-bcaX8thCvFy34uAnolPGB0MD2rlkf_gzPDUlIk,18512
tokamax/_src/ops/flex_attention/pallas_triton_test.py,sha256=Nau27kj4rnaZunL7gOA5x06FlbNr6kBrICz-hNnbtZc,1896
tokamax/_src/ops/flex_attention/test_base.py,sha256=cw8HWDq1pFWlZ0qYoe1VgTPHyHileXKtBKEE1ZONGeM,2959
tokamax/_src/ops/flex_attention/wrapper.py,sha256=nbt78mZTH2SInxwx0nw7HkfsW6bxC8g-mRu7LUvYzkA,3481
tokamax/_src/ops/flex_attention/wrapper_test_base.py,sha256=yMK2Hm8M2QyCSkf2x2iFRlePbb5WQxR3bZhV8M9fWG0,1335
tokamax/_src/ops/gated_linear_unit/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/gated_linear_unit/api.py,sha256=t_8aLiBApvV4_-88vETgLFudmyLPp3Cb3m9Fp2eguTQ,4369
tokamax/_src/ops/gated_linear_unit/api_test.py,sha256=7kqJA_pS1Mt7TNJI3HHaNWu9_SMODiKmvF9s_6DILg0,3727
tokamax/_src/ops/gated_linear_unit/arg_specs.py,sha256=DECEEmyM-WzMoD7z8nd48JBDKUK9zyhY2InIT76AVaU,820
tokamax/_src/ops/gated_linear_unit/base.py,sha256=xo_fxsku-Cwvj13xm7k6n7yWgZTqy5pBUnEooCZuzxc,6542
tokamax/_src/ops/gated_linear_unit/base_test.py,sha256=zNoOZbucSKmx2up3WFSuuiNG3Nw8wBzs3ekHfm21vUw,2701
tokamax/_src/ops/gated_linear_unit/pallas_triton.py,sha256=pMZRFIyx71UT0JXyZcoqrayUvNGUyzPbjHEGIklj7GQ,8452
tokamax/_src/ops/gated_linear_unit/pallas_triton_test.py,sha256=0CYyz24N5KnqqpGl4zdvWmtEjU9GAA4fP2m8SqSylwk,1692
tokamax/_src/ops/gated_linear_unit/test_base.py,sha256=3q504rraPtugBrWMfBE9v_m72XCxa10KGkFQfvuITbE,3733
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=Dkl9rqRK7vVtPeLCqNPO26xxl-PZ16ZUKOQgTjnTq-o,4286
tokamax/_src/ops/linear_softmax_cross_entropy_loss/api_test.py,sha256=o0_epX0qM6YkwNAkkGKYnAL1xdlYGtAAbmaRPioBOzw,4443
tokamax/_src/ops/linear_softmax_cross_entropy_loss/base.py,sha256=IXs5poYvF2UgW9VUVrr194Qc2MkjXb0324_Rko-bpGM,2810
tokamax/_src/ops/linear_softmax_cross_entropy_loss/base_test.py,sha256=PlmNj8iET7fNEmxpAgTUtadowNwiqCG-8vkZT0N6DtQ,2494
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu.py,sha256=bFuo1Lmq8zOXqloSGbo03VFbRvU_O7iBnlPjQBSPczc,23435
tokamax/_src/ops/linear_softmax_cross_entropy_loss/pallas_mosaic_tpu_test.py,sha256=RuolgJFhM7-TBCLkpRlEzLWiZRvVNKmr6yAU_CNJqLg,11388
tokamax/_src/ops/linear_softmax_cross_entropy_loss/reference.py,sha256=I_z_vvq7lFr7llMzaIXGUffmVKmdp5YZUmHBwuewyXE,2505
tokamax/_src/ops/linear_softmax_cross_entropy_loss/reference_test.py,sha256=mMTXhvM0A7tx8Def7jPzoVqOz4NF5mQalA-GC_wkav8,3703
tokamax/_src/ops/linear_softmax_cross_entropy_loss/test_utils.py,sha256=8kffMQWUuqoUvRTRO6x1mZ0d07LrJtriJmp_ehnB-mA,1679
tokamax/_src/ops/normalization/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/normalization/api.py,sha256=JLjrk-4xOiep3uYHT93bq2bN6o4eMUMUKn7AkRY_1Qo,4603
tokamax/_src/ops/normalization/api_test.py,sha256=zgqj2IGx9X82sFGxVTIzWmDNTbhpR1O6uMlYXNzimwg,5359
tokamax/_src/ops/normalization/arg_specs.py,sha256=trQYL9acYBYVBO16mdozgBeHbWSUSlDuKljK6UZ4nAs,2739
tokamax/_src/ops/normalization/base.py,sha256=9oMy73lUaVwMkkwXF2nQGr1HjbXXaBFG6qSgAd8XrrI,6320
tokamax/_src/ops/normalization/base_test.py,sha256=XKZTDAMwsol5oEzzdB9Wu-pu2_12eSSadGLyMeKKweg,1027
tokamax/_src/ops/normalization/bench.py,sha256=cME1KGCi7yxlepZd2YH6zGZOz5wVvfbi9bePk5Kr3Jw,2193
tokamax/_src/ops/normalization/pallas_triton.py,sha256=n8gsrAFsYMM0xiZeetZVZfnP5umrsROyWbWIXa9n7Vg,8024
tokamax/_src/ops/normalization/pallas_triton_config.py,sha256=YZ-i-pT3k-0mIJcVO06Bgr3657j2yD5jHxYlOetmsI8,4130
tokamax/_src/ops/normalization/pallas_triton_test.py,sha256=-s9elX36iQbgaFIlVK5Ii1qWSYzHWeX6SesWDb6sMBY,5154
tokamax/_src/ops/normalization/pallas_triton_vjp.py,sha256=N71A0wRMMJHy8emsTesB7LwtS6EYXnGrrYINdTN4TqU,7314
tokamax/_src/ops/normalization/pallas_triton_vjp_config.py,sha256=4EUNT-bFybyWiWz8GLa1iZwWw55eOaREmh345Plnlps,3118
tokamax/_src/ops/normalization/test_base.py,sha256=shzt2qWy3HsdgQqdC1vSEcOoF5-xQs6nsjQO4-qlTb8,6683
tokamax/_src/ops/ragged_dot/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/ragged_dot/api.py,sha256=wQ1WRZNeEGS95l-dCrcNCUC4tdmoLBfVAn1H_9ULLIQ,7714
tokamax/_src/ops/ragged_dot/api_test.py,sha256=cELFMkosRpjjQWpeJHBOiI6yTsj7ZuKhcJkcGjPT1dc,6819
tokamax/_src/ops/ragged_dot/arg_specs.py,sha256=PrpRofNRI0blrYx8EHJ4nrqyWu3eqcXkMPQpmso9RX4,4978
tokamax/_src/ops/ragged_dot/base.py,sha256=Co0yitGKo99USXJjQl4JArer7UyXqEX30KGkSr7pXDA,9613
tokamax/_src/ops/ragged_dot/base_test.py,sha256=Q8VIIfo7boCCMZ7-4BNNnn09VEV77w5DPEpy2Kmq73w,1525
tokamax/_src/ops/ragged_dot/bench.py,sha256=4Xw0bg2ch3aO1N7wfTF6vJQdr8SLnCS9ApsRvfwAHCI,4109
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu.py,sha256=Om7AeXoTNYmz5GeaBPFeLmyWjuvTpQmxEs61wkuJzmc,10355
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_common.py,sha256=OpCu4PTAvMACl82PUKhNGI1PDt1eSQJtWMXIgXAxa1s,14156
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100.py,sha256=Luu_w2WMVR91gKfdmOHVTIpQGLCn6LDTlle7GrR05w4,10693
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm100_quant.py,sha256=Pk0mJw_FcFEEASmtJuXSk_RQAbYP9S55LToxJ--2oSY,14497
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90.py,sha256=gGWgmWh_1aCa1hEJXs9BUHEwYdSXCyrGdpotzyoliXs,5345
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90_quant.py,sha256=04xbj2CCPnyHnkyEatE6atDuCbfYlrnDEr2XkLuNrXE,5023
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90_quant_ws.py,sha256=raoskUDbxtRlJXenPFfmMO_XgqndQkBMsmi4IJeR8V4,5769
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_kernel_sm90_quant_ws_async_store.py,sha256=CbhMdCCLrPx5n7es4J8UZhtvwbsuXoGaccSSd3HTrdY,13897
tokamax/_src/ops/ragged_dot/pallas_mosaic_gpu_test.py,sha256=rWyHgD4QkkW5q3XFVVbMldmczpT5EgdHepC56xr3Saw,2848
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu.py,sha256=NG2EKV61jmphnu_mbgAw4Nnm2LsijbNcvfD2xtVdLcs,14372
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_kernel.py,sha256=fPjM3uuz1Ci2UfRWLLIi-slNrdZqjpUBGra0V7DGAFc,31333
tokamax/_src/ops/ragged_dot/pallas_mosaic_tpu_test.py,sha256=iTm25Z8mrdwwQUWiplKo-u9tOPWt0wpqfo5qrNdH_8g,5018
tokamax/_src/ops/ragged_dot/pallas_triton.py,sha256=LFGcSejIl5q2WEzbMT679FLavAgkZ5cocoZEwEu1KtU,14690
tokamax/_src/ops/ragged_dot/pallas_triton_test.py,sha256=OGU2VqAY07PyPd9JZb7DnjwseiKI4vpHQl_hKA5Dlp8,2555
tokamax/_src/ops/ragged_dot/test_base.py,sha256=e0YeJ7VrONtrH4OieZMiFbRZFtsv4PdRnJkjAwXfbg8,8385
tokamax/_src/ops/triangle_multiplication/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/ops/triangle_multiplication/api.py,sha256=55rXRwL-y1TrfsznoDx84TRt3UAa7Kw_idxCiMZCNpM,4360
tokamax/_src/ops/triangle_multiplication/api_test.py,sha256=h5s0pMZyElSv49Wx_TPEi6VsQXCtN2j1nMxMIAFhxSw,2894
tokamax/_src/ops/triangle_multiplication/arg_specs.py,sha256=BN9SSgfaCd-2Bz_CQMGbUQADuRx2iMn9heeWPWe-WM4,1931
tokamax/_src/ops/triangle_multiplication/base.py,sha256=fq_nwnV1aueOvuslRAydaLu-6JWOcfPvm_DcIILyEAQ,6446
tokamax/_src/ops/triangle_multiplication/base_test.py,sha256=LLjXY0wTXuUMsradIVGN1VLCMI3zhj145saHr4XeZbU,3452
tokamax/_src/pallas/__init__.py,sha256=sojZirG99P27D5bJ9BjPUNR3NbS1SYi4vZMSjOBmGHM,696
tokamax/_src/pallas/block.py,sha256=P2LjA-rm5uRCpYY4dUTdKVCkDAvfYh5QhmTYwh5IOTo,11529
tokamax/_src/pallas/block_test.py,sha256=s1lNHBl_guAh2BwMjjWiqeKR0Nytl8jn9SF0dkFdN5c,5519
tokamax/_src/pallas/grid.py,sha256=d_T5YafLTkixhDr8otTNh5frmIcpuHosYrWTbCnqAKQ,3641
tokamax/benchmarks/attention.py,sha256=mAlMp6dooK0BO9kCyQzD2SDLRWcJMdQiNjUnjTUWLOY,3372
tokamax/data/autotuning/nvidia_h100_80gb_hbm3/pallas_triton_normalization.json,sha256=kvKhWPajdF0uD9oXplGDV_Tl1zAsahCclU5dOcQWWk0,979
tokamax-0.0.8.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
tokamax-0.0.8.dist-info/METADATA,sha256=qHtDCyySnD_-s4pA5MsIWJSjlMSFxueyQ6rrD0qYvS4,21078
tokamax-0.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
tokamax-0.0.8.dist-info/top_level.txt,sha256=xHgO7qOuYjrLH3047bLQLnwrjI76i9-xlChxNQIZn-Q,8
tokamax-0.0.8.dist-info/RECORD,,
