ejkernel/__init__.py,sha256=1d53a56ad4efc9447f82d05f54c54e2189bebe26add0846d9ac6126b9251cd40,1931
ejkernel/benchmarks.py,sha256=703e7889f9dcc3a4306a4026e203530220e44bdc7299b073d929870da3e537ab,29901
ejkernel/callib/__init__.py,sha256=b2fdeecaba896640534e11468b7a4d09afd70d64c27624a8028fb68fdbbfb697,1570
ejkernel/callib/_ejit.py,sha256=55a6a495f747911cc3e0c0fb69a92aa03a890596f2fca91f39051813265bcb86,18718
ejkernel/callib/_pallas_call.py,sha256=b821d3337cb57af92f25557b6e467dac01e454fa4a63b1fe5bcd0d296d1eff46,7459
ejkernel/callib/_triton_call.py,sha256=deb433fc1f6b39ccb502669b125359c35b5dff654b39be56c0679950f0c97bc4,29126
ejkernel/callib/_utils.py,sha256=396724d03f567ebe5933529d731d65cdbab18c048f81befb86485a8526121256,6915
ejkernel/kernels/__init__.py,sha256=e191da5a771e667f109772d97e953b92d972b22f89a5d31b4df360ae1c4f6f5b,2065
ejkernel/kernels/_cuda/__init__.py,sha256=2bdd8310fb59a3d51154bf1737eb3712be0321d78a6cb3441b5c221279dfb52e,627
ejkernel/kernels/_pallas/__init__.py,sha256=11b6e6af98b006accfe78c746647fb561aa740e6c60c71327b102a0b245b58c3,676
ejkernel/kernels/_pallas/gpu/__init__.py,sha256=041193c2dec33bc75c9ef57ad1bfef71f9536cd3616c1d9077cae1a6a9ef4a35,830
ejkernel/kernels/_pallas/gpu/ragged_decode_attention/__init__.py,sha256=cfa51e85f0b19709d89a609f0614000b18d4034bebd9406d8c47fed3b58e1fd6,715
ejkernel/kernels/_pallas/gpu/ragged_decode_attention/_interface.py,sha256=de36f8129bdcae859ee58fa3b1df40b2444ab81d808e0c51084a7bde80d18e5d,3742
ejkernel/kernels/_pallas/gpu/ragged_decode_attention/_pallas_impl_fwd.py,sha256=37775621c7822adf6dd5a8af172a9e985daad193238e14748718bbd90691e625,17255
ejkernel/kernels/_pallas/gpu/scaled_dot_product_attention/__init__.py,sha256=904d6543c185ece358c3f5c6d5816cf0cb0aa533b602437d00f9fc0f4acbc14d,1023
ejkernel/kernels/_pallas/gpu/scaled_dot_product_attention/_interface.py,sha256=353583d59303c05e2f8062eaa17038f5e2da76043831e1b0faa321dd8899068a,6325
ejkernel/kernels/_pallas/tpu/__init__.py,sha256=6a4f0a08450b7050613273406ee7f402539b4e43c2ab5f400032973a64a23673,1369
ejkernel/kernels/_pallas/tpu/blocksparse_attention/__init__.py,sha256=c39e8452d2fec05ec97d9510e731ec395bec973f99577addc6c7d3d102c7ff4a,1742
ejkernel/kernels/_pallas/tpu/blocksparse_attention/_info.py,sha256=c354fd3de0d5350f6158a87e6d75ac2173cc4f5d0bfed0facf89652411ab9cfa,36575
ejkernel/kernels/_pallas/tpu/blocksparse_attention/_kernel.py,sha256=f62f226368739dd58a274591cd9b2f25bef5c51975761e412c586a7c514ffba6,82858
ejkernel/kernels/_pallas/tpu/blocksparse_attention/_masks.py,sha256=c64a372c919a070c03bc329ccba87e4779fae626a68d179b6e2c973cb3a3c486,17194
ejkernel/kernels/_pallas/tpu/flash_attention/__init__.py,sha256=4670c157974cbab379026cddc7414feaba82de08f53468172a1c7d0b010922e8,911
ejkernel/kernels/_pallas/tpu/flash_attention/_interface.py,sha256=842fb7c1889891cd66a329b1979f24a1e59b46f572fcc3411b14b6d0b48f52c2,7322
ejkernel/kernels/_pallas/tpu/flash_attention/_pallas_impl_bwd.py,sha256=61567defdba9bc103ccdaeae260ec821e92db73255ca3f3875db28d9b4eb16d1,22478
ejkernel/kernels/_pallas/tpu/flash_attention/_pallas_impl_fwd.py,sha256=2f8c1fb9e30d6952d110d08ca91129f98a19122e65435ee2e5bc52e0dd9a84ef,15082
ejkernel/kernels/_pallas/tpu/flash_attention/_utils.py,sha256=df318496246e3987791e102b598577e73b465d84f47969fefc6348c11d542e61,10540
ejkernel/kernels/_pallas/tpu/grouped_matmul/__init__.py,sha256=db277fcf812baaa35e3195fe53c0e735dab8a499a3ebd2e825df3b3f820bbfd0,697
ejkernel/kernels/_pallas/tpu/grouped_matmul/_interface.py,sha256=ccc21c71ed8dea43299adee31ad17f993ee41644ad51c190dbcafb518fd767d8,10612
ejkernel/kernels/_pallas/tpu/grouped_matmul/_pallas_impl.py,sha256=ca3ffd347be80cfa42343bbcbf940137c1ed65c9720bcfe5bc6ae90e3d13fa6e,40041
ejkernel/kernels/_pallas/tpu/grouped_matmul/_utils.py,sha256=a5dc8b83c44fda1e1991dc8794cd3d77c1e9644cbc118597a1f246328b436c7d,6368
ejkernel/kernels/_pallas/tpu/grouped_matmulv2/__init__.py,sha256=2bc4a2413ab8f97573328479d97f293ed80b2ad4732dc66158014b0966b20b32,701
ejkernel/kernels/_pallas/tpu/grouped_matmulv2/_interface.py,sha256=8a4eb9c867d0dfa365b4ad9e2aeb6e72d16f635379ecb897c82121ccf7a8c338,10556
ejkernel/kernels/_pallas/tpu/grouped_matmulv2/_pallas_impl.py,sha256=e9122661bc669a5f93718c870bd98da7c527ac545fa4b75d383c228df3e965ee,20319
ejkernel/kernels/_pallas/tpu/page_attention/__init__.py,sha256=9e2b012eb9648da97e2f245e9213f67e577316d9afc3475b50b41565e3f2f894,697
ejkernel/kernels/_pallas/tpu/page_attention/_interface.py,sha256=062e09273f3bb84202550851a507550983156ef44676795bf4fe2c771e54dde8,10504
ejkernel/kernels/_pallas/tpu/page_attention/_pallas_impl_fwd.py,sha256=dd6e253e7afc361c9728ccf239369aa38f917853d533687f340d19d0ba6a8ba3,9467
ejkernel/kernels/_pallas/tpu/ragged_decode_attention/__init__.py,sha256=bd3e12482b6d6a02c7a471d00c5031fd6bcfdb744a0a616294c04e6e57610cb1,714
ejkernel/kernels/_pallas/tpu/ragged_decode_attention/_interface.py,sha256=91d067ca60225ceb765db534e987a8b0c4240e1c659eefb45f1b6f6f1c38ec79,3305
ejkernel/kernels/_pallas/tpu/ragged_decode_attention/_pallas_impl_fwd.py,sha256=22dfd56b46fa1edce2e50d4251ab6e898b780a27f955561f73045f1eae3503e4,9247
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v2/__init__.py,sha256=d96b3c0d7a95100c6ab4f8ff038d7dad06f000dc13ecdcebe97d6395c8a61e89,754
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v2/_interface.py,sha256=dfe8d6d47e7cb1463b7ee625910c87c5383d88bf005c4159c4e2ab0a3798435b,10319
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v2/_pallas_impl_fwd.py,sha256=3f3b25c4f7c9de788d8489a3ec8e57dc9d2d8227a9783b7d2f892e145e59a150,23712
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v2/_utils.py,sha256=592c5278304abff27c6cbd71ccaac015a267a6b35fb0d8aa8ea565bc4ee5884a,25463
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v3/__init__.py,sha256=f06fdfc8b61fe4747e381c1ed951e16684bf87e7d7434d612730b63122466f89,826
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v3/_interface.py,sha256=86e3a6d2ba59617e3517202bc2cb55a51f97c07ad6d1238e969c229f0f8e7989,4893
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v3/_pallas_impl_fwd.py,sha256=9ad8b3c889f72cc03d8b835f622b4cbfbc63a932aeda4e538a216c7d6e01ed25,46675
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v3/_pallas_impl_fwd_h64.py,sha256=33e09099755d6b35d40bac979d37322117291c6e9f181f032f9490faf6bcc248,44525
ejkernel/kernels/_pallas/tpu/ragged_page_attention_v3/_utils.py,sha256=d32b155b0240c9f2990b161c566ef682bdff00f1c5b5ae855b80a17f88fb4b51,153723
ejkernel/kernels/_pallas/tpu/ring_attention/__init__.py,sha256=a4d5ec4fe0811b43b2ec6b2897ed41b615e36928b3c1d0429c839d439c698d8b,935
ejkernel/kernels/_pallas/tpu/ring_attention/_interface.py,sha256=c6fff0c32e93a3350a3173fc4204b0a18e0efebaee3039683e0ae9bd4243515d,8151
ejkernel/kernels/_pallas/tpu/ring_attention/_pallas_impl_bwd.py,sha256=cbe755d485b1ffc6bfd11223271c054c4e4ac4bd3d93bda5344bbb29d2275440,34709
ejkernel/kernels/_pallas/tpu/ring_attention/_pallas_impl_fwd.py,sha256=9c7bcc64a04617d21fee0864f4014fa9abc178d835f942aac3e0f5af1d37204b,25204
ejkernel/kernels/_pallas/tpu/ring_attention/_utils.py,sha256=32f1cca7b29e155b8552515675a833d4db310d240522789b937bc21af4a71f4c,6508
ejkernel/kernels/_registry.py,sha256=df93fed7b7473340344080479757763244550ca3e24774f73b7a5ffd4a22c7b4,13507
ejkernel/kernels/_triton/__init__.py,sha256=074c5794b430bb8344f689f1fa202a9f5197b676fe6d262a8957253c2487d7cc,2276
ejkernel/kernels/_triton/blocksparse_attention/__init__.py,sha256=2262fdf74c7ed197e830a9541358dbf743c19f72848cabfd25ac1a03771daf7e,1153
ejkernel/kernels/_triton/blocksparse_attention/_interface.py,sha256=997434c8fcea614bc3301b21b430c29e2713acc77a25b8585f328896c8b89a9b,18161
ejkernel/kernels/_triton/blocksparse_attention/_mask.py,sha256=898d7f019d04763c5c27c11f4d3d06bdd339619fd68d72c668f7d5acaef2b955,21633
ejkernel/kernels/_triton/blocksparse_attention/_triton_impl_bwd.py,sha256=92602ee466f044d880228613da0a0e0f5ba5d90f76910ab5c887241ce20ab063,42951
ejkernel/kernels/_triton/blocksparse_attention/_triton_impl_fwd.py,sha256=2cbd30a81d619a0017b9adeb919f2923753669188e0e170a554c638ad8c6b75e,24698
ejkernel/kernels/_triton/blocksparse_attention/_utilities.py,sha256=d958656d7085127925588caa5fb4916a5055ff1f903dbf49c1a464d7db879a6b,15317
ejkernel/kernels/_triton/flash_attention/__init__.py,sha256=5d152a64fafdb71e9c6c33820250daf040aef52a75094d5fbd6cb1cd3f2dd596,917
ejkernel/kernels/_triton/flash_attention/_interface.py,sha256=988a84c3fac8fc1123d5e260b6ccd1e5043642198b542a7347b355f851a622ac,16109
ejkernel/kernels/_triton/flash_attention/_triton_impl_bwd.py,sha256=c2d97b7db7d0557f9bbe4595a3b3cd9bc71dde4a4291491e2e09a8d2efad199a,46148
ejkernel/kernels/_triton/flash_attention/_triton_impl_fwd.py,sha256=a859e8056fbe648281ee11008ddd7f3379ce314aa3feab3b542f48f2d5da0298,26343
ejkernel/kernels/_triton/flash_attention/_utilities.py,sha256=33fe270800d9bdce61ef9cd14a96fc41bfa4dbdbaa8a97829f98aa9a730428af,12888
ejkernel/kernels/_triton/flash_mla/__init__.py,sha256=04e4ac07c744e64c672594430112a3663d6d3d0b7b73d2b6fc76524ecf65615b,1285
ejkernel/kernels/_triton/flash_mla/_interface.py,sha256=7c60d7a1d10e63a59db227a8b976fcd6b66b72df5ce95885cc8ca705d631ea69,4261
ejkernel/kernels/_triton/flash_mla/_triton_impl_bwd.py,sha256=2bdd8310fb59a3d51154bf1737eb3712be0321d78a6cb3441b5c221279dfb52e,627
ejkernel/kernels/_triton/flash_mla/_triton_impl_fwd.py,sha256=81bb0b037307d7f4fd9e9e860ba4a0114bb15884de58d150998e36db4d277e0b,5987
ejkernel/kernels/_triton/flash_mla/_utilities.py,sha256=2bdd8310fb59a3d51154bf1737eb3712be0321d78a6cb3441b5c221279dfb52e,627
ejkernel/kernels/_triton/gla/__init__.py,sha256=b43a63a0a265da090bfc78cd4757b722eef33e1c44a557e095b9640dd5e5ff08,695
ejkernel/kernels/_triton/gla/_interface.py,sha256=18b3609f474fb48ae07890de77adf80db8c027e5c0bcb84154ea3a559d7ec37e,6145
ejkernel/kernels/_triton/lightning_attn/__init__.py,sha256=b9a44c4c0ef978b8837cfd30e196996893e1c14a7bc0a894a472612fa203aa53,697
ejkernel/kernels/_triton/lightning_attn/_interface.py,sha256=0af313c03ddc995f8041c52e160aeb0ce3d40a5cd6e01169639563bbc204dab9,6408
ejkernel/kernels/_triton/mean_pooling/__init__.py,sha256=c9ba8d8207c6a4381a0a2a822eafffcb1d6d0030bc9b825d61fb837a512d6eae,898
ejkernel/kernels/_triton/mean_pooling/_interface.py,sha256=31c855f5a015080f18442792da04033201fbcaecefbf7ccc76c5fc6ad314e642,6083
ejkernel/kernels/_triton/mean_pooling/_triton_impl_bwd.py,sha256=139688c13aa62357da6a111487eec88063d104e64f9625688ff420078621f1e9,4794
ejkernel/kernels/_triton/mean_pooling/_triton_impl_fwd.py,sha256=b6fb4dffeb54ae5f15db3ffa03a088a774a09362b6374af6bac11bab707ff6f3,4738
ejkernel/kernels/_triton/native_sparse_attention/__init__.py,sha256=d657385c27dcd11f9cf51a9e05d8a7db4ef0c3aa271a438196a492bd695e9f60,1036
ejkernel/kernels/_triton/native_sparse_attention/_compression.py,sha256=ae2a33f5983cd01c8a71d4c3ac02f3e7f7156332dbf5acfcf77e2597f2801740,24864
ejkernel/kernels/_triton/native_sparse_attention/_interface.py,sha256=249c8336faf2293a07a160bbe9d6a552994f3a98218b85fe0681ca4f184f5bec,16025
ejkernel/kernels/_triton/native_sparse_attention/_triton_impl_bwd.py,sha256=67b7fce78f68ba747cde6fd9e1b70cb7a1bd7613b5c2d4e8c5f69f32c9c98297,12843
ejkernel/kernels/_triton/native_sparse_attention/_triton_impl_fwd.py,sha256=e754269cbb85d89470b96bc7a744af99332d6513bc5cbe83c8c75c368ed361dc,15144
ejkernel/kernels/_triton/native_sparse_attention/_utilities.py,sha256=d37ab27a0378a2969cf7a7a383c386e2d94808a73e0a8530fc28bd0d1e62bc75,2613
ejkernel/kernels/_triton/page_attention/__init__.py,sha256=9e2b012eb9648da97e2f245e9213f67e577316d9afc3475b50b41565e3f2f894,697
ejkernel/kernels/_triton/page_attention/_interface.py,sha256=b3031272a68d4bc5015c822b841841bb86bfcb26cdcbcba90313589499381e54,14646
ejkernel/kernels/_triton/page_attention/_triton_impl_fwd.py,sha256=a852b6611646ee74f72ba99ed74c83410056a2332f1f46b6409d953a315dc8cd,8422
ejkernel/kernels/_triton/ragged_decode_attention/__init__.py,sha256=cfa51e85f0b19709d89a609f0614000b18d4034bebd9406d8c47fed3b58e1fd6,715
ejkernel/kernels/_triton/ragged_decode_attention/_interface.py,sha256=148d615c1f4b87438f661476dde916b56f700a62165890d44a4c025b83eaa74e,2731
ejkernel/kernels/_triton/ragged_decode_attention/_triton_impl_fwd.py,sha256=dfb992839cfc7c2711541b38be94f8bfdc822b58138b7dcae24fdb5aca1ce5a2,9542
ejkernel/kernels/_triton/ragged_page_attention_v2/__init__.py,sha256=f7505c49535dcc00235f027220b4ecca365eda5d864b7a5d638fa8d4c7bd4241,717
ejkernel/kernels/_triton/ragged_page_attention_v2/_interface.py,sha256=ec7888b6c5ac51c37f1b449bb7ddeb6ddf248b1c7d12a73142eb1d6f585ad152,10144
ejkernel/kernels/_triton/ragged_page_attention_v2/_triton_impl_fwd.py,sha256=5561a74cc823cc110bb45e9a769b8bb46f78d2a68cacaf026417e0e2945f5d2f,19729
ejkernel/kernels/_triton/ragged_page_attention_v3/_triton_impl_fwd.py,sha256=e30c6f64658a0e72354a0c8cfa50e466a46b38221ade5ddf13cc80434eac927f,6372
ejkernel/kernels/_triton/recurrent/__init__.py,sha256=6152e73b5769f07c29e1e31b951a2d79ec3bc57f40d3a0a76d3ba22f8c550948,880
ejkernel/kernels/_triton/recurrent/_interface.py,sha256=0c1ea241eac91caffb2d23dba74d83ca81e8ea051aff9430a17e3e335f07618b,11630
ejkernel/kernels/_triton/recurrent/_triton_impl_bwd.py,sha256=881ad8faf2f7665f8316d2663c8984e7154b39d9ec5270f38c6fda718f95db09,14180
ejkernel/kernels/_triton/recurrent/_triton_impl_fwd.py,sha256=1f8849977c415522f682f6da7addbf93cd94752826204c9c4bf9b7586bc54a21,7956
ejkernel/kernels/_xla/__init__.py,sha256=cf75e5c337d5883019ad0956751e62e53fce0370f59078626d9d8a411b7c2a92,1674
ejkernel/kernels/_xla/attention/__init__.py,sha256=9c95ef43c4ba47e04ade05b050d5cbe37bf2ee04a35f6e072b6322355f8a7b7b,687
ejkernel/kernels/_xla/attention/_interface.py,sha256=deab94684c4bffc46e87cdeed3f33d5a6b6bee8a227462e097690727288154b9,9485
ejkernel/kernels/_xla/flash_attention/__init__.py,sha256=4a0771e987035ce61511f9dbf86a111fc9ae5aad82fce44a98884818dad744cb,758
ejkernel/kernels/_xla/flash_attention/_interface.py,sha256=ee091c38f92a19fe3f16eea242f276650c09a66d0c6f278bf9e6a0111a5c93d0,11778
ejkernel/kernels/_xla/flash_attention/_xla_impl_bwd.py,sha256=017e1aa602beac85c86171bf6da1c4ac06decd95d553dee89ba159a313eb1075,2430
ejkernel/kernels/_xla/flash_attention/_xla_impl_fwd.py,sha256=5a343faabc67b28f71bc51a64f7ff12f4856fa47da73146e2cebdcdef4828d53,17420
ejkernel/kernels/_xla/gla/__init__.py,sha256=aa3a1c70f3d00a83517a1461f15c50cf86f911092e4bcc0f3dfb43f7c9cd59e3,694
ejkernel/kernels/_xla/gla/_interface.py,sha256=ad850ca1f0831a54989d8fc96fd24b57a1e8b47430a77529ea84933da7bf0545,5429
ejkernel/kernels/_xla/grouped_matmul/__init__.py,sha256=db277fcf812baaa35e3195fe53c0e735dab8a499a3ebd2e825df3b3f820bbfd0,697
ejkernel/kernels/_xla/grouped_matmul/_interface.py,sha256=8da9285918cc85d0440e1bd2ee59cafd085b95870eb6d13048211cdeb1369102,6433
ejkernel/kernels/_xla/lightning_attn/__init__.py,sha256=b06790c84413a9c3acb9fe10e08048afc4ff788fda92e6d8af9fd0c1466b8b43,696
ejkernel/kernels/_xla/lightning_attn/_interface.py,sha256=ee39cbb6bd5e8775b8fc6876ece970299f70b637f493784321ae0ef61dcf27ee,5525
ejkernel/kernels/_xla/mean_pooling/__init__.py,sha256=14eb5328822ce3c162509d466fceb8a2feb9f0efffacf12c3d706b2b658c278e,692
ejkernel/kernels/_xla/mean_pooling/_interface.py,sha256=544d1aacd3369d0254f63583296ba26f31c7c03e8e72751e1e77acf6c85d2f5d,5578
ejkernel/kernels/_xla/native_sparse_attention/__init__.py,sha256=37da0071885c5f326199846c79c2d03d7c9b97f632d5e7fe2c28ebc83b388d5f,1068
ejkernel/kernels/_xla/native_sparse_attention/_interface.py,sha256=cc74e9d7b7b49438b75d18118596e568fff77a12f52324515a0a83544b6cbde1,18355
ejkernel/kernels/_xla/native_sparse_attention/_xla_impl_bwd.py,sha256=87ff5761db651d47e17ff78a963c799a47c86dcc1f6f052865743818dfd17f45,4805
ejkernel/kernels/_xla/native_sparse_attention/_xla_impl_fwd.py,sha256=10229595ea78c128fb5da92caa87877e1315c33e767df75c27cc6f171b406abb,3092
ejkernel/kernels/_xla/page_attention/__init__.py,sha256=d8c81d5ba5a275ea6a8d8ad722744c3c9e69d8dafed5a90327b829df6bab7e11,696
ejkernel/kernels/_xla/page_attention/_interface.py,sha256=5738b03ae11bf37448390812476a730a0436693ca9cf594fb62b9e82af722007,5872
ejkernel/kernels/_xla/page_attention/_xla_impl_fwd.py,sha256=a40d1c3e5ee5ea458bf80818ccdc8a6e39cb4707c70df1b399c1aa25e6accb7e,3595
ejkernel/kernels/_xla/ragged_decode_attention/__init__.py,sha256=bd3e12482b6d6a02c7a471d00c5031fd6bcfdb744a0a616294c04e6e57610cb1,714
ejkernel/kernels/_xla/ragged_decode_attention/_interface.py,sha256=8ef05b6b58e8a7e5636b037c4d9796ce0e269001e1557cfabecafbfa55916334,5320
ejkernel/kernels/_xla/ragged_decode_attention/_xla_impl_fwd.py,sha256=8b20930e83481bf2859b08cf0ead86588b324b4fa4149362a6bd95a957f84fa0,16606
ejkernel/kernels/_xla/ragged_page_attention_v2/__init__.py,sha256=b6add257496f63bdc1f47f96f9989ce32e4e21bc68b6ffa26dbd04f6b1d9a496,716
ejkernel/kernels/_xla/ragged_page_attention_v2/_interface.py,sha256=d3526a7bbaa3f514fca2a48c3e416cc1d78b27940006f8b88e7877533e413e01,5169
ejkernel/kernels/_xla/ragged_page_attention_v2/_xla_impl_fwd.py,sha256=bc59acb7aa03dfa1d3f97001871ad03ae568eeebfdf2e6011634b13a51f9d629,9141
ejkernel/kernels/_xla/ragged_page_attention_v3/__init__.py,sha256=a4e29b8658cc8760ff90bd35de7a4f5cbcfe941e9955fa3df7b472d525754275,717
ejkernel/kernels/_xla/ragged_page_attention_v3/_interface.py,sha256=504638df892eaab5810573b52f0e1ecdaf793fdfffa833dd53dc5159fe3410e7,4060
ejkernel/kernels/_xla/ragged_page_attention_v3/_kernel.py,sha256=35ce051f5edef134b3b8e0b69b543ed1b1782f9ec9e8a29af2e10041f47f44a3,14178
ejkernel/kernels/_xla/recurrent/__init__.py,sha256=78e034ca31c11a294851349555987abdcbb567adb924217654f883223b8e4106,890
ejkernel/kernels/_xla/recurrent/_interface.py,sha256=972cd68f6601bf586a719fca9e749f62dd16d33c837a170a875e055469f09aac,8556
ejkernel/kernels/_xla/recurrent/_xla_impl_bwd.py,sha256=dd1058309b9f8580f245ef0f15331d7768b59defeed8d6a8cd41e2fcd4842f1e,5577
ejkernel/kernels/_xla/recurrent/_xla_impl_fwd.py,sha256=e984f733d6362b1e1e93cef31aaab0e1bcc307339b29a369009bcef04ab613f4,8356
ejkernel/kernels/_xla/ring_attention/__init__.py,sha256=e6a2b16bea3f444cb2c2ee666b9f4621132dd4ee695933f5266c99653acbfe32,909
ejkernel/kernels/_xla/ring_attention/_interface.py,sha256=de11477d506c66ab6376e2c4f7c7c123573783335cd3962528516349cd00698f,8241
ejkernel/kernels/_xla/ring_attention/_utils.py,sha256=285be6530f920f1cb8de64ce6b3ee9a14f945a42d1ef589695ecb97ee9852148,6374
ejkernel/kernels/_xla/ring_attention/_xla_impl_bwd.py,sha256=b76c2fbaecdff3a5bb080a4c52777495c9d8fa603ac955f74f67c7738a159c5f,13350
ejkernel/kernels/_xla/ring_attention/_xla_impl_fwd.py,sha256=18830fc170e2db64bff8b4d36f8b463bdb835728f7b5d203330e6bf53b0d6f18,16852
ejkernel/kernels/_xla/scaled_dot_product_attention/__init__.py,sha256=6126e4cde30bf82c5baf0c8c00ef632f3f46225a798863ded18d2887bf1eb8f8,925
ejkernel/kernels/_xla/scaled_dot_product_attention/_interface.py,sha256=f51183421baad0472c8b38fd234b1b09fbf1e20acb9e6c86e3345d938d1368dc,4772
ejkernel/loggings.py,sha256=12faa74e66adb967f9bd04aa4a79e1e62216824432578eadeb174d7accb1aced,15582
ejkernel/modules/__init__.py,sha256=4fa008326b657c60a4baedd674ee44b7190559e64645d331329e38ef4d007c63,4251
ejkernel/modules/base.py,sha256=9323906a8170df45a0ba8f45fdc3acc4f66de371d9289db94b1c4b1c33d7024d,7585
ejkernel/modules/operations/__init__.py,sha256=34b974c0b8767f73be4d708e047562de2fa3ac763e1fc4cfed0bda87864fc383,5467
ejkernel/modules/operations/attention.py,sha256=c2fa09f78d8b994d3008aa10345ee9852aa6421b0f652aaba0c98d0e44bdf1c3,10419
ejkernel/modules/operations/blocksparse_attention.py,sha256=0354b909378fbd988a9d0f70c5e9ac4a12304836414e7be5bafee654571680ae,36058
ejkernel/modules/operations/configs.py,sha256=f05c933f737cdf640015e1be7c54fa11b1e98367e159d2021a87b230b75a892f,11650
ejkernel/modules/operations/flash_attention.py,sha256=8553693575c364f3e10c41da39320810f8c3dc95f7717c9a95917954f24aacdb,33062
ejkernel/modules/operations/gated_linear_attention.py,sha256=9f7bb69d2cab083a70315c68107ae54b02d87bed358e5979060767e491d0d9f6,11201
ejkernel/modules/operations/grouped_matmul.py,sha256=48a13468bf978e56064ba38dbd532c964e4427b261c4bf07fac0c7ede9878bf0,15845
ejkernel/modules/operations/lightning_attention.py,sha256=cfbd73d77e18597a7d67b5b5d0543730e6ee3398ee529c88f88f719e8ab31c1a,11463
ejkernel/modules/operations/multi_head_latent_attention.py,sha256=249d01ae0b17b87dae726fe327eb633ed2aed3cccad66056d72928116bf257c6,10615
ejkernel/modules/operations/native_sparse_attention.py,sha256=ea60eda760c8619b546f471e6a0b04c291a61947eb7d594ef5f7a979178b0fa8,14103
ejkernel/modules/operations/page_attention.py,sha256=6f376cfc89ec54f131a635c25a0792df3ed350a414fab7eb77ec012e11962f04,17841
ejkernel/modules/operations/pooling.py,sha256=0caeb680d4c39dd2396cf775a44fe7fd4575519853e894e28f9332038f74d379,8385
ejkernel/modules/operations/ragged_decode_attention.py,sha256=6f578806131dd63af7ba8326f20a5df032b386a91daa511a6ad2a6591a3d86ce,22962
ejkernel/modules/operations/ragged_page_attention_v2.py,sha256=434d0eb4dc24fd84fcb156ccda37ce99332890a59e26d00b2c7ae9a3869905f0,27072
ejkernel/modules/operations/ragged_page_attention_v3.py,sha256=e6c53678594c83f36ee26935a51a753b34d85805a246c2a5b99f7a6811e2e488,46561
ejkernel/modules/operations/recurrent.py,sha256=a2c59921a2b07e3d8d46852f60f1b05351c39d96aee8ac4c1faba1c0ce694afe,12487
ejkernel/modules/operations/ring_attention.py,sha256=5a2c4faaf7e665cc18a3dc2e37591dec870c23d29f32818b61e943b6f2665b98,24567
ejkernel/modules/operations/scaled_dot_product_attention.py,sha256=a9f0cb29a3425ec6b4a5a5d74b035fa170f4882dc9c5ab20937713d00c9630c6,16487
ejkernel/ops/__init__.py,sha256=e94ae82dbb9ed98e246d501969aa53df26aa4e561873e71fc586658028371659,3788
ejkernel/ops/config/__init__.py,sha256=685d5a63ee2e7835a705c0c20f234fca24e2642330ce4a4f4f1bcd63a541d176,1609
ejkernel/ops/config/cache.py,sha256=9aa6c803ea85a4c6b6735a8390c382f434e7db55208ec08841f4f203559b8b46,6301
ejkernel/ops/config/persistent.py,sha256=98bcb706409774e010e0b4327b5d04c72515acdc2ad55ce9aa8a8f44ce5e9e9d,7023
ejkernel/ops/config/selection.py,sha256=4888694a9b67fce844c8416eb29f191f42c3967a4488938a2d25c3c6117a6d7c,22920
ejkernel/ops/core/__init__.py,sha256=340f958e66fa99791a6695ed2c0d8984dd43290ea13520a9f02a2534a06349e4,1388
ejkernel/ops/core/kernel.py,sha256=f90757724b2a764149f5b1783da5170433401ef50fdb23649007366b897f7fdc,25617
ejkernel/ops/core/types.py,sha256=e9b99227f416e29b73dbd857abc8457e3cf3d60dff2127ad2601948352b83285,1722
ejkernel/ops/execution/__init__.py,sha256=607c6056fddea483fdb0303682ca1376a6fe9d7364fc7320d0c57833328fa7f0,2168
ejkernel/ops/execution/batch.py,sha256=541e138bc9f7fe50dae5057c5b1d1b082cfc005324df09be3765f34f2d14cc7a,7338
ejkernel/ops/execution/executor.py,sha256=8a2e8ddd5f5ba3f64205f09825fe51658fdbfe10c66efb313654f803e4a1e96e,20132
ejkernel/ops/execution/offline.py,sha256=3b7f769a2c4e4e422018f2097b07a5ee796ab9b0d1317a0a1200bbd102360cd8,3116
ejkernel/ops/execution/profiler.py,sha256=eeca2c718b552910c812c66db069180cd78ed2e2d4c12b59d79cbcc7f8e4dc06,20100
ejkernel/ops/execution/tuning.py,sha256=3ad8dd5950253e2ebfe588ea2e5b9d2fa067d369c3f964a71631b8daf062dae5,47963
ejkernel/ops/registry.py,sha256=869ef9d1f00bc23c6ad56f2e569a69aec498c6e091858abc613493e210a2b07b,3534
ejkernel/ops/utils/__init__.py,sha256=ef6f93637ab03d38e077aa88019d3c715773fb1f327d2dac63c6aac43c087bd4,2277
ejkernel/ops/utils/datacarrier.py,sha256=5e5932ee6879a8af697dcbb4e7f41adb9492fdc742d966f5cd953ccc7d1deddf,4867
ejkernel/ops/utils/fingerprint.py,sha256=2fa5fc44f67b560f6acc8ffafcb66f899aa42bc3fcaf73f5b20fb68490f330ac,12135
ejkernel/ops/utils/meta.py,sha256=0f5bcb12fc277063840235dc732360bd19d365133ee436c4c4b1dc5d183d68ad,5387
ejkernel/ops/utils/serialize.py,sha256=9a1c8b975bcef5b9060c29d0aa20b774bd6a3da83fbc0129c87fbd98234c4764,3100
ejkernel/types/__init__.py,sha256=60e06f33ffb74a72ca86abf7371fe0e1aa98f839a4c3ab5e4ef5a1bfe6793b8c,746
ejkernel/types/mask.py,sha256=db56f13f057f09fc187ee78c943a7330e6e51fa864a12fe805a8dc6d5563b071,92352
ejkernel/utils.py,sha256=efc3ffdf9ccf4fc2e1b051b395d6f954c26bb152f1c77d3898f2ce887dee9e60,26924
ejkernel/xla_utils/__init__.py,sha256=16facdf953d4bd0e9ac844e189000bf6c9fe35d7cc1766a943795f883a985ffc,1388
ejkernel/xla_utils/cumsum.py,sha256=9ce9d5ae89d28f4ac62c2ad8defd2f972d9b29ab2b4076dafd8b2edee0ee252f,10350
ejkernel/xla_utils/shardings.py,sha256=54577cd2d7c69673fd611216db5bbd06e8edcd180eda417dd4533efea2ebaf32,3600
ejkernel/xla_utils/utils.py,sha256=1a75221c7e0a0c0de6d542edab8bf6ff350f582a35d3fe5e09da45d05087e41b,5944
ejkernel-0.0.21.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
ejkernel-0.0.21.dist-info/METADATA,sha256=0b0f2276ddcd1d131d4ea21f310256dfbba18e7b685e0f9db65b4140e7c6edc1,23590
ejkernel-0.0.21.dist-info/RECORD,,
