attn_gym/__init__.py,sha256=8KWh4x6pdlmLEheSVD99rU3MQHKbXFFLIEtNzsVKO_U,500
attn_gym/utils.py,sha256=wlNKPzjm-wVZqErtsnkUWMGJv-LHYwqDvuypg2E-778,13716
attn_gym/_backends/__init__.py,sha256=fQUAAYv7aslxFM1Tuk3J-pgvhRJWa7MZ48INUK9g_Hg,69
attn_gym/_backends/cute/__init__.py,sha256=tSFLdw5XqX-IV1DMHWst4OJzEDVkgf0tu4lbQYgEAIY,686
attn_gym/_backends/cute/_compile_driver.py,sha256=xJ-A69gaVjGGPKLi5KQenlVJqv_OBub3P_FpsebmZ5g,3230
attn_gym/_backends/cute/_key.py,sha256=8PchXmf0wiZNJ0K_WKCiWVsaMWOoIbEylUiAN3J8X0U,9977
attn_gym/_backends/cute/cache.py,sha256=JmQi-bLewi5scXYbrda6PuwN-8JtQDj8z1o59bqL_NU,16190
attn_gym/_backends/cute/compile.py,sha256=wpHkVewmKo0UvupOyoeRHbguxfp_yTs06jmx9IyuRso,6571
attn_gym/_backends/cute/device.py,sha256=lJlhF--CcF1DUNn0KO6SXuWvHNPhN9B7HNOHmLWinyI,1772
attn_gym/_backends/cute/target.py,sha256=hLJLIp5dAoSgCVlEYjGNB6G0QIxDGxG-qdXUkZ1h49g,2744
attn_gym/_backends/cute/tune.py,sha256=78HChfugu4gJ5u_7zCnN5xZlQJspvv-xmqlIZVqLU8M,13468
attn_gym/_backends/cute/utils.py,sha256=b-VUvItfLuWnuWEd1S0x1b_1kqsroZQq4bzr691_60c,6428
attn_gym/_backends/triton/__init__.py,sha256=qy5k-WkLx7BbWIc04Jio6cugJerz1CbLPCEZX7mLLEU,58
attn_gym/_backends/triton/utils.py,sha256=ZuMD3SKCGPPUFRuFFdp_ews19RC63yHRCCO1JPJKuOk,3474
attn_gym/linear/__init__.py,sha256=GXou-hpScTJKEJgoaMWfKGSrWKMAcqmJa4xAVvVAWOE,1831
attn_gym/linear/types.py,sha256=pRAa3EOmuranAW0j2sQoxNV80ajH85GDUb1Hy9Nbiug,818
attn_gym/linear/gdn/__init__.py,sha256=vHiu6uRen4_lEzmYcj_WT7k3Y2KcvgOfyzNIG3oGOjU,184
attn_gym/linear/gdn/api.py,sha256=fosrDXVN5UkVmIJE6_d3IZxqSXEfQCb1U0KfzBtwtjc,5363
attn_gym/linear/gdn/validation.py,sha256=08y5ddllvhiWB_QGGplgNuqgIeY5KG6nx8f6fMv4yRM,2518
attn_gym/linear/gdn/impl/__init__.py,sha256=lkQBAQQkke1zrzYcLmvW9SgMgLbTVG5kXRf5o1GB_Ks,48
attn_gym/linear/gdn/impl/reference.py,sha256=YxTez3r-guDi5iaNutvDe0eXotvtnojHdhn4R3wG9ng,5819
attn_gym/linear/kda/__init__.py,sha256=vz7_J7UR49de6g9ccelWm-HlzTxQzgCzTWCPpk62ZfU,2222
attn_gym/linear/kda/api.py,sha256=SBmwdnbYEPbcqE6rOyiNrr49vwo1pypVGZqzgwAY8TM,22688
attn_gym/linear/kda/chunk_schedule.py,sha256=amf2IV1hjW4x8vcQV7t5IiuANF0ueTAy44sx7utO6X0,5630
attn_gym/linear/kda/chunk_scheduler.py,sha256=Z1eCUWhwLbEiUOvDrw2rhyvZl4y5b_fSvrjupPWu7mc,17330
attn_gym/linear/kda/constants.py,sha256=EmYTmRP_QiYONzsqL1TIDe6nmutkR2CKHx_RUz35Wuw,927
attn_gym/linear/kda/gate.py,sha256=ok3hlO8IPUL5FmNMMXHb4qdvZtrXr9GTeuOdJOA1dLU,5473
attn_gym/linear/kda/masking.py,sha256=aI48EfqbnWwBkQpev2wqasG7cOBMkwX2bEoHjtyHQxM,6155
attn_gym/linear/kda/naive.py,sha256=g3LcH06a6j2QgLGeLnEwLgZaMyQs250vaCMe6BEUUcI,13724
attn_gym/linear/kda/ops.py,sha256=6eG5jalhHMbMnBJ37wqCTUDIbjokMsIw2_G6k3rbMnE,30746
attn_gym/linear/kda/utils.py,sha256=AjAS8cu0mIZfXLveu8NCVc90ROshjHqQuYKImzcujvA,9389
attn_gym/linear/kda/validation.py,sha256=RxxyZBlEt1Vv43XUbg6n-Zio9Zr2A6TdBp_sNkFfbig,3210
attn_gym/linear/kda/bwd/__init__.py,sha256=ILrJFMVawrvTK1q6PFRaOZBcPc09h9TIinZeIouzYEQ,213
attn_gym/linear/kda/bwd/cute/__init__.py,sha256=ILrJFMVawrvTK1q6PFRaOZBcPc09h9TIinZeIouzYEQ,213
attn_gym/linear/kda/bwd/cute/chunk_delta_h_bwd.py,sha256=n9o8FQWOMRxL_3K2PUpqDB3pyrMapGrdW4mX64b9XxA,92765
attn_gym/linear/kda/bwd/cute/chunk_kda_bwd.py,sha256=swBIriowfv6O-BCqpkBUOE0CZ_qYFq46f2Ta632vMpU,4304
attn_gym/linear/kda/bwd/cute/chunk_kda_bwd_intra.py,sha256=7ErU4vmxTwceqjwqH9GvTiDryEhKyAlOvvW7cm4HAEU,75926
attn_gym/linear/kda/bwd/cute/chunk_kda_bwd_wy_dqkg_fused.py,sha256=Dw_hOAGBfEhCmGB5w8drHhzDL6AjX-DHb9cfb-NEqHo,154715
attn_gym/linear/kda/bwd/cute/gate_bwd.py,sha256=Eb2FgKSSgJ9tAfZVNgy8_bGZYicwD6p57o_TgJ3vHM0,17101
attn_gym/linear/kda/bwd/triton/__init__.py,sha256=ILrJFMVawrvTK1q6PFRaOZBcPc09h9TIinZeIouzYEQ,213
attn_gym/linear/kda/bwd/triton/chunk_kda_bwd_daqk.py,sha256=YBPbznk9Y7_hoCVaWZfoO2Fmmz3aV7GucYezQkFyc8Q,13497
attn_gym/linear/kda/bwd/triton/l2norm_bwd.py,sha256=bcDuITTHkCWSyxyMXrVCecdKVDD6f_UHjq-8ll7WxlA,2899
attn_gym/linear/kda/fwd/__init__.py,sha256=ILrJFMVawrvTK1q6PFRaOZBcPc09h9TIinZeIouzYEQ,213
attn_gym/linear/kda/fwd/cute/__init__.py,sha256=LsnPhFUhcLqE0iSGoBzd-30ExWAinZZitTg_TUL_8Ks,290
attn_gym/linear/kda/fwd/cute/chunk_kda_fwd.py,sha256=NaU7P5ZnTuSwkHcV_vy7kYxyM57hh4sXPR7vclhmlQE,17383
attn_gym/linear/kda/fwd/cute/chunk_kda_fwd_inter_solve.py,sha256=JLozIbrYZWOkVLGIguRyss-mneFokG1R7On1NhOJVAQ,21273
attn_gym/linear/kda/fwd/cute/chunk_kda_fwd_intra.py,sha256=d0OodWZx6nSBg-CMvF6KHHQrLMj8YtjdEVvFfm5HYgo,4582
attn_gym/linear/kda/fwd/cute/chunk_kda_fwd_intra_engine.py,sha256=Wt_evPN0FovVnuYFuezAbEVNyrX_d8wpCaUUC4hCTHc,41177
attn_gym/linear/kda/fwd/cute/chunk_kda_k3b_offdiag_cutedsl.py,sha256=bDfEsMxgV2jxGeFmcF7za8Mk9qbwqdlmBdFfLwBet14,21106
attn_gym/linear/kda/fwd/cute/chunk_kda_k4b_inverse_cutedsl.py,sha256=m-erAKKZ1IzBOokuk-cAX2BEJNTxqxBEXEzDz6EkBQ0,32849
attn_gym/linear/kda/fwd/cute/chunk_schedule.py,sha256=bjmi0GZfB-OljtzI_9om_dWCXHW8ZdH24lTRQEOzWEU,304
attn_gym/linear/kda/fwd/cute/chunk_scheduler_cute.py,sha256=b1-RIqC_5uSkTbf7Y-A9vL-ptCmLlXjzWLAzrNB9ed4,7250
attn_gym/linear/kda/fwd/cute/gate_fwd.py,sha256=EDiXGPGgGSQiBT1DrngzxZccdfBcxFqwgWVJrmOzLpA,8874
attn_gym/linear/kda/fwd/cute/recompute_w_u_fwd.py,sha256=tyDKemTb6Rod_X5dD-3st5t6Io3OgeRbTNYQiLC3d2w,72289
attn_gym/linear/kda/fwd/triton/__init__.py,sha256=ILrJFMVawrvTK1q6PFRaOZBcPc09h9TIinZeIouzYEQ,213
attn_gym/linear/kda/fwd/triton/chunk_delta_h.py,sha256=iEAixkn1Jeqmgx1B9ZFG9w_9eO1hp53mq_yzT2RZn88,23809
attn_gym/linear/kda/fwd/triton/chunk_gla_fwd_o.py,sha256=25mBZChbGbSl1eqkf1dycNeK15NdnKF4B4fUUhyF4XE,19938
attn_gym/linear/kda/fwd/triton/chunk_kda_fwd_intra_sub_chunk_forloop.py,sha256=LVyMqWLI6wOxmBKCVMRvkKqs227PzkWDpfAQccabdQc,10887
attn_gym/linear/kda/fwd/triton/l2norm_fwd.py,sha256=QjPK2wVkTIZ0z63Tf08HyYxoY9BbJ6qutbMscjP9iaU,8228
attn_gym/linear/kda/fwd/triton/masking.py,sha256=yEhFgPOAJUUoca_SmI7jI93Y1Mq5CWcNkBxpPQAuQYM,1254
attn_gym/linear/kda/fwd/triton/plain_gate.py,sha256=qMT0V6tjlqYoxccmEV51Xa0pHnM4iYU1FCVsKGcH9jo,5144
attn_gym/linear/kda/fwd/triton/recompute_w_u.py,sha256=Fba2AT603XyEjY2yh9Zsd2gGtSosPCAsxcae2gzeb1U,17759
attn_gym/linear/kda/fwd/triton/recurrent.py,sha256=ApbaOGXnxTluh1WjCgF9IizPIUYWMtSdHZ3PdJVkrEA,14899
attn_gym/linear/kda/impl/__init__.py,sha256=kDMF1XH0X8OjXKa_zrdrusBFttn-hoOZzMtm7_qeq7Q,257
attn_gym/linear/kda/impl/fused.py,sha256=tL8_d98xJe5oFpqQrjiIDSKDSPU7xBsgCXt5mw8oZLA,8637
attn_gym/linear/kda/impl/reference.py,sha256=JZWnwupbp2PplLz4K416ssr3ECeyOV-YcLE4fiU-A8A,3873
attn_gym/linear/kda/short_conv/__init__.py,sha256=I-bIED-mUKvS84AoL2iL-HtHi1PNv_incPwQquPKMX0,676
attn_gym/linear/kda/short_conv/activations.py,sha256=XH8vk1zP96-HBVMEBqEVuDi_9FZlWoLuz1nFAmF53J4,6251
attn_gym/linear/kda/short_conv/cute.py,sha256=91qBBNVZw9Y5wOEqfMaMKOmuAZCTUaB_sglLnso66ZI,146089
attn_gym/masks/__init__.py,sha256=QdVVtMMu5DY5PxI8Mxn6-1NJ8MLG_wd00cTujXYvz1g,1633
attn_gym/masks/batchify.py,sha256=h6fl6Q7MdBxKb8WZrl1iwFb1iWaAcOUtXT7xDeiHKT8,2210
attn_gym/masks/block_diffusion.py,sha256=UZX62jZRBWrGP-hPQH639wpZ8Cz6bU0e3_f9eeIwCEs,2721
attn_gym/masks/causal.py,sha256=Zjes98okQ0oe6rHoGvzrUA5B2McqXqUQfR3IoMB8GX4,5165
attn_gym/masks/dilated_sliding_window.py,sha256=PBxW8ERi04D39cBYPEzn8VuZ_-kbNTlL5D79XmhBCVg,1827
attn_gym/masks/document_mask.py,sha256=sc11bMLgbU2JWfzCFL1MC1pBj8zQ3lxNdg_jClqutaI,4336
attn_gym/masks/flamingo.py,sha256=RmjixPyPi8pEzcrWO8y-8Ha4B84nQKmWDlGShrJ5Okg,2843
attn_gym/masks/global_sliding_window.py,sha256=JYfY0OleSDkhHvXPi6e84bqYuZ3FKREZFdcF-rs78h4,2396
attn_gym/masks/jetspec.py,sha256=Zcy1rUYKfKGX6d7Tp3j8hLQ2Mbf0Wt7BOjg4ifJtq2s,5897
attn_gym/masks/natten.py,sha256=BbnzqT2Im4Svf5nWmgj5WOI5_ZGLkWmUM0AKeAhSOsQ,8036
attn_gym/masks/prefix_lm.py,sha256=AgFTr3F9n97IMJykTyGU0cEEm1fPA-7sGu4pj43abUA,1638
attn_gym/masks/shared_prefix.py,sha256=8Q7ZmVV4YqMPXaIge5oeQ5CtSuL18eIZtubvoZXXlg4,2951
attn_gym/masks/sliding_window.py,sha256=IXvYMBGUhOwXkL7ZFm-qXhbNIavf93jY4_U87fjwffo,1578
attn_gym/masks/sta.py,sha256=S6TMLHS_wsZGsWBk_5x1aFseTWFm_Gt72KabQMv5hkc,9836
attn_gym/masks/svg.py,sha256=suOQLWfsaA1TILZKhF8S6N5u-MhKt8GnWMdo8s88J4g,5036
attn_gym/masks/vsa.py,sha256=2WtbXAtQutjXRDIDKS-hSDkJaQFCiuRgm-7HOojUzMs,35707
attn_gym/mods/__init__.py,sha256=BZ7Kkb7Dn8Xzm8U7PFVvrHQdhXr45kiAUsXdL5uHBjg,495
attn_gym/mods/activation.py,sha256=CUZU5T7QaZj7IjhMNqVGfDgiTLfhLtfHcBhXcwHQCko,3751
attn_gym/mods/alibi.py,sha256=YQ650wmOFiApcR68P-InIY3_6Hqb0KaeS-gH0FqL06E,1669
attn_gym/mods/graphormer.py,sha256=MCtffRQmY3JboILBhVLfnc1sdWmTnpu_44cZL47ltso,9058
attn_gym/mods/latent_attention.py,sha256=fIT2UnxU50KvYYcNhQLZtCLGp_TxAcdmY3LcY5jd0OA,2547
attn_gym/mods/sandwich.py,sha256=_yyrvU5hDJ4pj9G-2QM4rteOLkdMYIPZjCCe0IZbpo4,2978
attn_gym/mods/softcapping.py,sha256=rquN7jFjC0nmErzLI9gN_5vVEAiYFtsrUYbB_TY09qs,2970
attn_gym/sparse/__init__.py,sha256=nAOCvh2VjfG8dwZRwzFSxlV0DZBalGHZ_Oc_o1z_CJc,121
attn_gym/sparse/selected_attention/__init__.py,sha256=DgcIBGJRChI_PbtO6izUKDm9Fk-olSItgwYDiTi2m04,106
attn_gym/sparse/selected_attention/api.py,sha256=kW6-waNMTq-WbHKa3ng4_WDn8VYPQyZi1GSc9ze4xls,10068
attn_gym/sparse/selected_attention/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
attn_gym/sparse/selected_attention/impl/cute.py,sha256=_6w__BVgPoVtyjnRbPoDqXMqqnlUMciMcBJylzPm4KU,6091
attn_gym/sparse/selected_attention/impl/reference.py,sha256=F4CV36K4CqY0MVRcDx6G-49mYnR4ViQ39qrHiNFCbo0,7693
attn_gym/sparse/selected_attention/impl/triton/__init__.py,sha256=ICKeclOprg7_ZqShhzj1EsHjyND3QKO_z8npR1TBrKk,4635
attn_gym/sparse/selected_attention/impl/triton/backward.py,sha256=3Bpz3eI_dkFooCj68ElErRCU29DyMfDgM0KJnpYrORI,27440
attn_gym/sparse/selected_attention/impl/triton/forward.py,sha256=O1HqjualnJDznIdxYqmoYfnRtfxRVeQ0FHMeqhySjGA,18969
attn_gym/sparse/selected_attention/impl/triton/primitives.py,sha256=7m7VYTWJdGUcQ5uAEK-EgSyU9nTX3CuwJPA62LFlCaE,3110
attn_gym/sparse/selected_attention/impl/triton/shared_backward.py,sha256=1i_Cx4rC-6HnOPtfd7QBHUJ-_h_BRqkgO-QtBRi1QI4,15943
attn_gym/testing/__init__.py,sha256=Jzc-mT2lWdN6vleoMYHM-xXwq5LR8-saTwLsghrvH8M,152
attn_gym/testing/kda.py,sha256=kpHzQ33GO9bLJtg9EtM_WJVW83FBwjlXGmqMqXa8p6I,11833
attn_gym-0.0.5.dist-info/METADATA,sha256=e8tb5-veq9aEeEb8Ram_RNaEQYLv_WB1QM0ahNaGvx4,6087
attn_gym-0.0.5.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
attn_gym-0.0.5.dist-info/licenses/LICENSE,sha256=zxEbkIPIQLB5X1Fmmc209oHgBh2TxbdupicAi0GVK2I,1501
attn_gym-0.0.5.dist-info/RECORD,,
