attn_gym/__init__.py,sha256=8KWh4x6pdlmLEheSVD99rU3MQHKbXFFLIEtNzsVKO_U,500
attn_gym/types.py,sha256=MXBa9vCUGjq5jAfAXrzhZTsW2DOHrJ6Ql1eqO1bDTfQ,786
attn_gym/utils.py,sha256=zBQ05EzqcYzFIkc-JYWGvIMXIDxOI73Ih5TjEsRjVBM,15597
attn_gym/_backends/__init__.py,sha256=fQUAAYv7aslxFM1Tuk3J-pgvhRJWa7MZ48INUK9g_Hg,69
attn_gym/_backends/profiler.py,sha256=2wgdNkjzhaGK2rgAsLCPKYM4g68-7JF_qZd_D_sAlbE,647
attn_gym/_backends/cute/__init__.py,sha256=yMq6kR1DEj4dc76eEPmWvd3csL41Ea4k26e7zbvct3Q,860
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/ragged.py,sha256=mUW4xdlY1b1SLk8xNftB-G_mifu1oN_ry733x6dnRvY,559
attn_gym/_backends/cute/target.py,sha256=b8d23NLPOz5x9CngNRJmCHPVK7PPPJmfPp0iJVsusms,3240
attn_gym/_backends/cute/tune.py,sha256=78HChfugu4gJ5u_7zCnN5xZlQJspvv-xmqlIZVqLU8M,13468
attn_gym/_backends/cute/utils.py,sha256=mS41-3ex2BzCZ9EACbigXJIbCbTF34FvB7szL3FZydo,8921
attn_gym/_backends/triton/__init__.py,sha256=qy5k-WkLx7BbWIc04Jio6cugJerz1CbLPCEZX7mLLEU,58
attn_gym/_backends/triton/utils.py,sha256=WL0PePuAVuBlb6ysKocy9983pMXYkWVxhroDv7cGEqk,5195
attn_gym/linear/__init__.py,sha256=NvaNA4m5cjJ0n0_pSMEJ-gf54dMaG_Y66NyVqQmZv1I,2390
attn_gym/linear/_lazy.py,sha256=np6KERw_eUx3Syo5ylYuQ9KsY23S4zj9YHhWmI1raeA,1319
attn_gym/linear/context_parallel.py,sha256=6gVWlFGI7lFyVD_81foY3SBwSxtRL7RZkL2J77rePb4,41519
attn_gym/linear/types.py,sha256=DBvsp3GpscS0RokZJPC4PfY_E4GhYd_GPP_hgWsLhc0,2372
attn_gym/linear/_delta_rule/__init__.py,sha256=Q46BZe31PZJA118m7YHdh4wRO5gFxppuEKW71xyUixM,63
attn_gym/linear/_delta_rule/chunk_ops.py,sha256=ld9u5p6yltoweLlby99mX3gFQjDYE3lUPf7w0usDSx4,2502
attn_gym/linear/_delta_rule/chunk_schedule.py,sha256=O5Kxu7Bx1NUckwZ2meW2n8U_DF2HO7vUUgjNGDQfPTI,6324
attn_gym/linear/_delta_rule/constants.py,sha256=afPblFbn-JfTrX-8ITqPGLmh0MS1GnOpIgEy0eZchhw,168
attn_gym/linear/_delta_rule/decode.py,sha256=cXv9xiLu8WfiHuXDxF-f02_HhChE6wgHzaGUf7xFS3M,11217
attn_gym/linear/_delta_rule/gate.py,sha256=pAsaBFjo_YVPom0nc4lChXVXgDJ-8E88CpEnaeI02wM,12292
attn_gym/linear/_delta_rule/paged_state.py,sha256=4sbT34GRrQ_9uc9BJSF7bHxvtg6N4FBjXJAqIcb_Xa4,4886
attn_gym/linear/_delta_rule/recurrent.py,sha256=80V4fsuhZRDh3gyb5-wp91dElM8BNXujFxknFjsX5UM,12263
attn_gym/linear/_delta_rule/reference.py,sha256=UhkSQHLRKaF4I_Ys-D1C_QhKq67r3HCWZmlog2pzKDI,2190
attn_gym/linear/_delta_rule/span.py,sha256=hMiAKL_u8Kq6x1IxYfbiu5-k9sKDlnx5jtP3LGv3qSU,2572
attn_gym/linear/_delta_rule/validation.py,sha256=yXmZo-qTUfAaG1GhGAAvzkMe95q1_fkzI4B4mvXnMHk,8248
attn_gym/linear/_delta_rule/cudnn/__init__.py,sha256=J5YPHUMzeBCf_Au3t46KUM28LHWyiPz7VlXzpS-w0eU,238
attn_gym/linear/_delta_rule/cudnn/backward.py,sha256=UZ5mtwJxR-Z-3AK3gP_2PpPdXVLCmnsVHfJvAi2Eyt4,7241
attn_gym/linear/_delta_rule/cudnn/forward.py,sha256=hPSp1buAen1N8yw7jutEp2wOrfvss5i4SsmUVRtUk5Y,8917
attn_gym/linear/_delta_rule/cudnn/gdn_backward.py,sha256=QcyH_BnBRHGMW8I7vq0OmVCFVgg5dM5tENgL-yLeYiY,5451
attn_gym/linear/_delta_rule/cudnn/gdn_forward.py,sha256=IxU2aIE3Puqtv8JMf0IYePmED85B3CKbLDhxdTrtshA,6845
attn_gym/linear/_delta_rule/cudnn/schedule.py,sha256=FWUfv7af8xyk5EVJ9P1gdA1AjjoizgeLIiE7__mPN_g,2933
attn_gym/linear/_delta_rule/cudnn/state_summary.py,sha256=MmL7rszh1gCVE3E9IfO2TTuLeCK8xuNX3dFpnrn02cM,12225
attn_gym/linear/_delta_rule/cudnn/kernels/LICENSE.Apache-2.0,sha256=cJ5cgC1m_KtD_EMPMM0vTpK6jZiBshr_abakNiDd__Q,11437
attn_gym/linear/_delta_rule/cudnn/kernels/LICENSE.MIT,sha256=ah5ZklgKre7qECOyWU1BWodFekrxIWLE635eEWj9kq8,1116
attn_gym/linear/_delta_rule/cudnn/kernels/NOTICE.md,sha256=FS89MsSh-ZUQN1tTq46Lxnge1icNpBy8-lK9VXctmlw,3085
attn_gym/linear/_delta_rule/cudnn/kernels/README.md,sha256=sHfjFadgfYe6ofJell1mUo_HntV-dmE1C04uCxxQjMM,2810
attn_gym/linear/_delta_rule/cudnn/kernels/__init__.py,sha256=1PeK5MeQvWxkBsMPF0tnS4_iWU5Ip-LWGkB0hfLwZ48,68
attn_gym/linear/_delta_rule/cudnn/kernels/gdn_bprop_config.py,sha256=nIoTTVPJC0jhTiGcv4u0p3q6wGH02quX5PDSmm1pL5U,2396
attn_gym/linear/_delta_rule/cudnn/kernels/gdn_bprop_f16.py,sha256=RnlzBF9oH065gRoiCW9dPs0D7utIV57WolqZYdEBlrc,229386
attn_gym/linear/_delta_rule/cudnn/kernels/gdn_prefill_config.py,sha256=rppkQs7qIZL81riEgxhjMsSaulw7txhwvTvN5blRjx0,2610
attn_gym/linear/_delta_rule/cudnn/kernels/gdn_prefill_f16.py,sha256=GUK3IC6yqPuVkOmSp-rag-F2Y53T6BLa1WW66ZsrMaM,167382
attn_gym/linear/_delta_rule/cudnn/kernels/gdn_recompute_config.py,sha256=Zz8fjJFrNuZz74SUpWt-TIsagbfBaxjmdgyKWmgnJOA,2465
attn_gym/linear/_delta_rule/cudnn/kernels/gdn_recompute_f16.py,sha256=rmZYnOzCGPhugW4Hk5wVYVq7Hztaw12mS3ybDQrzYXA,137731
attn_gym/linear/_delta_rule/cudnn/kernels/kda_bprop_config.py,sha256=DrTRJ89Ronx43LV8KJW58rmdIJ2zV_UkKOsiUzvSlRk,2659
attn_gym/linear/_delta_rule/cudnn/kernels/kda_bprop_f16.py,sha256=8N9kwwUlPJJ-hNJGtLwpjkpaqLo52y3Dc2iqY7YP0J4,208279
attn_gym/linear/_delta_rule/cudnn/kernels/kda_plain_gate_bwd.py,sha256=gIKL0TgrjEDCnr2tMpsL33aKSewI7z49QMnjvbsuSt4,9512
attn_gym/linear/_delta_rule/cudnn/kernels/kda_prefill_config.py,sha256=VqJ3dmFlvEsHcuGGvSi3PhA6y4YKiqc-7esFkiXE9XM,2462
attn_gym/linear/_delta_rule/cudnn/kernels/kda_prefill_f16.py,sha256=W6H6EJ5eLFfFxv8z3sJjKjEasTNz_IMMS8x_Q19DS00,138587
attn_gym/linear/_delta_rule/cudnn/kernels/kda_recompute_config.py,sha256=1JKHOQkRPfVkVmmSmE897dEnqc9qDIVxzZu1OK4LoKM,2406
attn_gym/linear/_delta_rule/cudnn/kernels/kda_recompute_f16.py,sha256=ixRLxqaXXzJNc83k6s_zA7xHxnNcDEeEnFrQDNeqEHo,130046
attn_gym/linear/_delta_rule/cudnn/kernels/common/__init__.py,sha256=PPYa7Hz_Iu31GTQRWLk2bmGNAguEZtKFbk-AePY2rGA,318
attn_gym/linear/_delta_rule/cudnn/kernels/common/elementwise.py,sha256=eSi8_NudBRpkemh34Fov7v8d9SAFjDtks8IC5Q3IX24,753
attn_gym/linear/_delta_rule/cudnn/kernels/common/host.py,sha256=dj1oWH69KODm9jTEHfDzJSLgNn45tkmVn3Ig4Yn_x1w,1479
attn_gym/linear/_delta_rule/cudnn/kernels/common/paged_state.py,sha256=iTfA8HnZnON3REVdrBf9vljWMuh80pnCVHQ7DvineGs,1214
attn_gym/linear/_delta_rule/cudnn/kernels/common/split_k.py,sha256=P8YWUB4BjzhJS7GXhr5ZSeFmS4Tjwf5lGPwFK7iD0iw,52085
attn_gym/linear/_delta_rule/cudnn/kernels/common/thd.py,sha256=-FWeFhIAtaT9C0mlvCIl9rWNSABAmhP3ER8rOkZbLjY,6659
attn_gym/linear/_delta_rule/cudnn/kernels/common/tvm_ffi.py,sha256=i6d0YdAzSdNnwcmSnbt_DYMEo724j9mQVj62GZqWgOg,3049
attn_gym/linear/_delta_rule/cudnn/kernels/tile_dsl/__init__.py,sha256=kdjVnTnQ31oPypxn-nDwqUlZh2jvEN9kq0MH-hY99CU,173
attn_gym/linear/_delta_rule/cudnn/kernels/tile_dsl/barrier.py,sha256=iKjzLOXEL3pcdCqqqyLU30rPMoi-GxW1_WBtJjq2wxU,3948
attn_gym/linear/_delta_rule/cudnn/kernels/tile_dsl/handles.py,sha256=2302VaIGJnQ7gXSd8v1fFr8AO9uvItOxoUyemOo1J1g,4649
attn_gym/linear/_delta_rule/cudnn/kernels/tile_dsl/mma.py,sha256=Tk8vsxm9CAFkOPPOGg72nanUZzmaaoxxtAPcFUdL4e4,5785
attn_gym/linear/_delta_rule/cudnn/kernels/tile_dsl/pointwise.py,sha256=L7HmtUTjHn86o-HUHb7ijI3YUBQ06QI3gsAajRtVSJ0,4647
attn_gym/linear/_delta_rule/cudnn/kernels/tile_dsl/swizzle.py,sha256=55yOlotFsy4oYaHBzl_yzUHGG7PgSsLzCPbrsTQu6xw,2064
attn_gym/linear/_delta_rule/cudnn/kernels/tile_dsl/tma.py,sha256=pn1BQI2HBhbWjWuFcZyXF6zsiPd7lTqE6jZMo_3fHpU,2776
attn_gym/linear/_delta_rule/cute/__init__.py,sha256=UBuqO4OhQrDx97vAYU-uEcBSuvvFmAg_FgnOohkLdYM,262
attn_gym/linear/_delta_rule/cute/affine_summary_fwd.py,sha256=cFo7Q8u9rsiqqD-rDKqq0J6DptxUldMB1hA7W3NDDXI,54652
attn_gym/linear/_delta_rule/cute/affine_summary_rev.py,sha256=CVnqQnedx9W00P-ZOp7nFY8ChHeqNIqhCIJIrGQM0Kg,63351
attn_gym/linear/_delta_rule/triton/__init__.py,sha256=sRTHJDhncUu-uKc3Ni-4DUxVgJDeMjzN2HMycMoOVec,68
attn_gym/linear/_delta_rule/triton/affine_summary_fwd.py,sha256=MJCLmAIgPDuwYtbbP2wcXxOWj9wAex0VGbnSI41G85Q,4806
attn_gym/linear/_delta_rule/triton/affine_summary_rev.py,sha256=GL3-FUS6X7MTrCLMhgW6pIdyin9we1dqljaOTP9SF9s,5854
attn_gym/linear/_delta_rule/triton/chunk_scheduler.py,sha256=d113EY2vifazLseFqRo84P4V8w5ul_8nDksbkNwHEgY,19504
attn_gym/linear/_delta_rule/triton/fold_slots.py,sha256=q90sLXvVkD3KE14kDvaRTyPfMk8cYKCEBKvK25ravaM,3872
attn_gym/linear/_delta_rule/triton/paged_state.py,sha256=djvU9wXcAmb2v1UnVIB_sr1uf1DmouTsiJCFEih-62E,879
attn_gym/linear/_delta_rule/triton/plain_gate.py,sha256=hY3cB9TmKPrVoBQ_Dm4FGqijc5vM4LFlZpfVzQAlZx8,7141
attn_gym/linear/_delta_rule/triton/softplus_gate.py,sha256=9hkEDluF5l7q2T70htIBTCFAVlvfLb_JYGaqT8-PFJA,11134
attn_gym/linear/_delta_rule/triton/work_items.py,sha256=hcGw6vI5PyqJdMuNq7_-jfej1ylOwCYuJFnh8Gm4qIA,11284
attn_gym/linear/gdn/__init__.py,sha256=qaqKfo9Cy2LHVxxfkDZ__sxKgcXwAJn0PQfBpzfA_TI,874
attn_gym/linear/gdn/api.py,sha256=ZNtIhJytgpgyM00GWYI9wWs2mTSaHlX1x4z6cOH1W1A,16484
attn_gym/linear/gdn/context_parallel.py,sha256=3_DQ8rTKPn5xGOzIzXmoI6DnTfXV0Fi_ZVnK6E9KkNs,1328
attn_gym/linear/gdn/ops.py,sha256=-dNHgq-1X1ixEhKZM00hTmg0tx4D53Z10-dEJsAuWTw,24446
attn_gym/linear/gdn/stages.py,sha256=BW5iyvIJ-YbpapgVSvZxG4NDSwzJ-gh9f8XbP9f-vFA,9415
attn_gym/linear/gdn/validation.py,sha256=GosgQOWhZF5jqLaNw568tQlZjpFEJQRNdJ3pQXFgocw,2215
attn_gym/linear/gdn/bwd/__init__.py,sha256=kD3HXweJaejwxOuQdqQqeU-ISUOOWs4oRalOklA_N04,40
attn_gym/linear/gdn/bwd/triton/__init__.py,sha256=Ji4qLL6HVTbzaWIKIgdkUExQxo0OCcb1F6NbvHt4YrU,47
attn_gym/linear/gdn/bwd/triton/chunk_gdn_bwd_delta_h.py,sha256=Y5lEimIEgOtzkYuzPzqU6KjUHmHNpgsVM2ZppU17d9s,15963
attn_gym/linear/gdn/bwd/triton/chunk_gdn_bwd_intra.py,sha256=XnlH4pRll1GPR4NUOQHSq122kWoOJNClf1QoiclYoTo,8386
attn_gym/linear/gdn/bwd/triton/chunk_gdn_bwd_recompute.py,sha256=OhPV6RszWjBAxwwq8jFNx1mz3Jt2znA3T2zvu01hDWk,4649
attn_gym/linear/gdn/bwd/triton/chunk_gdn_bwd_wy.py,sha256=D1BCt03qYiBhD2d5KEf6K_cJiE_QhLks1__Wy5HpOK4,19285
attn_gym/linear/gdn/fwd/__init__.py,sha256=-moHgDrtgoLU3eG7ehtlpnu2hR1ssuoZWfKgeKqmUVc,39
attn_gym/linear/gdn/fwd/triton/__init__.py,sha256=PH_L_7WtGth4mOTRyTVrBH4F3W_6Gkt19ymAVxkSXD4,46
attn_gym/linear/gdn/fwd/triton/chunk_gdn_fwd_intra.py,sha256=L_homMcg0SbYbVIgajklqwtywg0adSz_Oj0cWiy-MYk,24986
attn_gym/linear/gdn/fwd/triton/chunk_gdn_fwd_output.py,sha256=AnNbeS7p5Vp_4FuSlBO6qobZCd5chXi1P25fEG1cJzw,7709
attn_gym/linear/gdn/fwd/triton/chunk_gdn_fwd_recurrence.py,sha256=dEptKS7gPmjmPV9HzWz6iLp5nnyvfaix0mHkvj40chM,5668
attn_gym/linear/gdn/impl/__init__.py,sha256=lkQBAQQkke1zrzYcLmvW9SgMgLbTVG5kXRf5o1GB_Ks,48
attn_gym/linear/gdn/impl/chunk.py,sha256=mPVh_MqBVCwzsuy7oriqQPgJVkv7_IYryPYr6e3o-FI,21330
attn_gym/linear/gdn/impl/cudnn.py,sha256=gQ4yaM-LTDkaiqdppFMOO5ZxJSYbd8msL2JGiz9bUHQ,8117
attn_gym/linear/gdn/impl/cudnn_ops.py,sha256=Nyxh9VLNK2aQmIWTemIrd5jAQkZLwk18nQ_dbL02-Wc,15440
attn_gym/linear/gdn/impl/fused.py,sha256=cNknvCM4_kE9OoMJF-fm1Btl9U36ss8LgMPSbPl2Wx8,4028
attn_gym/linear/gdn/impl/reference.py,sha256=YENZRSWlFZL9ZkQqaFbWgTjcrsxGlvPxBElVL_y9_bA,6352
attn_gym/linear/kda/__init__.py,sha256=Q7VHsdta4-Gs8OKrDXJph0SIJ-wwAADCHjUyDcIeAIQ,2399
attn_gym/linear/kda/api.py,sha256=fua1CfFpbQDVuDGbmcX4mg-MNOpTvLZgTPRPXRaYXwI,22042
attn_gym/linear/kda/constants.py,sha256=4wyCJD3f5EHP7a9zStZzaQcijs9hBkX3V8UjZROAa7A,1250
attn_gym/linear/kda/context_parallel.py,sha256=iUiR3mbXRa-R2EnyJvemPHqZ2doDXCqcGAp_Kh-MnP4,2410
attn_gym/linear/kda/gate.py,sha256=dSAc86jSBzZOMrZj2xZ1qzDIaaCAGN6LS88wFzPUZ0M,1858
attn_gym/linear/kda/masking.py,sha256=aI48EfqbnWwBkQpev2wqasG7cOBMkwX2bEoHjtyHQxM,6155
attn_gym/linear/kda/naive.py,sha256=wqf6lTyxvV49CibdwczTGXBB3qkoQEeGLO1grzCwCi0,13752
attn_gym/linear/kda/ops.py,sha256=LYpZed5vVOkjxEKU_h_JhaonIK6u9jiv-AwZX3SNRzs,24844
attn_gym/linear/kda/stages.py,sha256=P16BhJZ-ITPk2SU-kaBXOd-zJiWUxBL7TISFRtHitrE,20811
attn_gym/linear/kda/utils.py,sha256=QznXO7UVfBefGrfWBbWmT_faKAVlUZKIF6upSy6K2Rw,8592
attn_gym/linear/kda/validation.py,sha256=8bd7UiZr_zmM2lkHSg_Ts737wgNVgaU3UPpa4XfV15U,4099
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=INre5z3TLf8-317-efCikwnBlrDoBh2Mr5JzjFTP0UU,96026
attn_gym/linear/kda/bwd/cute/chunk_kda_bwd.py,sha256=1Nwdc5bYBpCDwJXyi9o01OgR69uUY9-POtMpBmyQvVc,8698
attn_gym/linear/kda/bwd/cute/chunk_kda_bwd_intra.py,sha256=UXCsnM3ST2vTO1h_Uzpnx6ze-QsjGTfYinfHc-735hM,79968
attn_gym/linear/kda/bwd/cute/chunk_kda_bwd_wy_dqkg_fused.py,sha256=dwMs6uzgWNhRBiXiEo3thvLJcVRheIOMGP3v-Jz_piI,155206
attn_gym/linear/kda/bwd/cute/gate_bwd.py,sha256=G6kExD1kyFTy-Vxe18pepVso9ZVcRNoamdiW6YMx1Do,18539
attn_gym/linear/kda/bwd/triton/__init__.py,sha256=ILrJFMVawrvTK1q6PFRaOZBcPc09h9TIinZeIouzYEQ,213
attn_gym/linear/kda/bwd/triton/chunk_kda_bwd_daqk.py,sha256=d3pd2yDNZbUymPssUQ9RoNZZT4LbsZHiAZr7sn_dMDM,13512
attn_gym/linear/kda/bwd/triton/chunk_kda_bwd_delta_h_triton.py,sha256=JfLoOOAW8t6O7EhW2J3Yt-pbRDT_eI-iEELejAjUM24,13270
attn_gym/linear/kda/bwd/triton/chunk_kda_bwd_wy_triton.py,sha256=XB4hGivM-gUouHGiHbo7A-azrd97BZwsFdXADkmbyDU,22767
attn_gym/linear/kda/bwd/triton/l2norm_bwd.py,sha256=lAmrIO27vJgeKIIccGeKUcQe5XkGr55HdSCavQrKgA4,2882
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=qOg42SyNwpskq3ZKnf4jNah3AsSCHZPvBLr6bnugNik,19239
attn_gym/linear/kda/fwd/cute/chunk_kda_fwd_inter_solve.py,sha256=3hGbDDzFbIca5BeDsThBlKvjGl3vo2SYjkZ3hoBV8r8,21315
attn_gym/linear/kda/fwd/cute/chunk_kda_fwd_intra.py,sha256=QnHRB9WBD5gsTxL-JU2r-Fv8n-psT_l4W-Mna5YG2Wg,5279
attn_gym/linear/kda/fwd/cute/chunk_kda_fwd_intra_engine.py,sha256=1P6vOEJZUfLWFw3dAvdtKrADK36Q8ffG5RrXkJYHVuo,41562
attn_gym/linear/kda/fwd/cute/chunk_kda_k3b_offdiag_cutedsl.py,sha256=JI-2UsAtFiNXDp0-TrdIguHKKKpdLD9Wyg5DebEkaQs,21114
attn_gym/linear/kda/fwd/cute/chunk_kda_k4b_inverse_cutedsl.py,sha256=3iWergOIoQapnwpaVeB9_MRMbzN37bjxXFs7Tn3crCE,32857
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=mC_iiLh547DQdeR7FoL-dDfvQYb1n54VhL1ivpXEWJ8,7079
attn_gym/linear/kda/fwd/cute/gate_fwd.py,sha256=e_LDN93zYdxLiFb8_OX5nfvmGLv6gkX_rqPvCAT4HFc,11403
attn_gym/linear/kda/fwd/triton/__init__.py,sha256=ILrJFMVawrvTK1q6PFRaOZBcPc09h9TIinZeIouzYEQ,213
attn_gym/linear/kda/fwd/triton/chunk_delta_h.py,sha256=jw3VGmLobfbza2M5eqWboRKT1BDuBXI1guadhCexBh8,26252
attn_gym/linear/kda/fwd/triton/chunk_gla_fwd_o.py,sha256=4PVVhBPbuJodFEa-irqNfi2AqQlk-gvcscEnCuGIKTE,20079
attn_gym/linear/kda/fwd/triton/chunk_kda_fwd_intra_sub_chunk_forloop.py,sha256=C4W8wUYWU8o_9RGYtYP12-AEUDSszOlquckAYYP4RCQ,10928
attn_gym/linear/kda/fwd/triton/chunk_kda_fwd_k3_triton.py,sha256=qAhAHY-9xrUwd7cPmtdTwi_4ZtzyyTcnqUzLEvNL8OU,16593
attn_gym/linear/kda/fwd/triton/chunk_kda_fwd_k4_triton.py,sha256=gILn6F9UvluLm1YmeeGOb4NcqImXewUIlyDMJUO7eXs,11714
attn_gym/linear/kda/fwd/triton/l2norm_fwd.py,sha256=ILshhX0BZ6mWN10bOaqsOg3fXFv1aw_pcSoVXvhgArw,9133
attn_gym/linear/kda/fwd/triton/masking.py,sha256=yEhFgPOAJUUoca_SmI7jI93Y1Mq5CWcNkBxpPQAuQYM,1254
attn_gym/linear/kda/fwd/triton/recompute_w_u.py,sha256=s6C8rYoge70g6uKCDkiT_4svNEtohxrLPRXRj3g8l5I,17913
attn_gym/linear/kda/fwd/triton/recurrent.py,sha256=vp5EELIesoTUpzs0H40LHuJeX_rTrIrj9_xBHRGQV0c,4935
attn_gym/linear/kda/impl/__init__.py,sha256=kDMF1XH0X8OjXKa_zrdrusBFttn-hoOZzMtm7_qeq7Q,257
attn_gym/linear/kda/impl/cudnn.py,sha256=HPG22zjLwcar_MWhksGWqfjLKoQgJt8GPfSmC8PVrZg,9865
attn_gym/linear/kda/impl/cudnn_ops.py,sha256=BEdJdqtymUOBK2zVTqpyPl4rvmLoZqvxqNa-ip6hXto,9571
attn_gym/linear/kda/impl/fused.py,sha256=xsUrN-yu0viqbj0vYG_vq8EiJgDgxmQ5tCMnTfoj9dI,9249
attn_gym/linear/kda/impl/reference.py,sha256=6v-raa8P3Ts9W-XRzb_yyJKvZ3jMxIM5Cnt3yCqycgc,2583
attn_gym/linear/short_conv/__init__.py,sha256=jwx3Hasd0740FDuH7GxE0T4gFaPjGzuPpq8KMZr7hVI,1540
attn_gym/linear/short_conv/activations.py,sha256=XH8vk1zP96-HBVMEBqEVuDi_9FZlWoLuz1nFAmF53J4,6251
attn_gym/linear/short_conv/cute.py,sha256=sTCTztOZ3x2rdHgNOoULjHArucXzW0jkaeZqWnMKom4,167946
attn_gym/linear/short_conv/ops.py,sha256=ZhI7yhD-gpErHUoccyjQ8vpQk60dOHxnmYuBovvFQmo,3781
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=ax4z5v8lEyEVucWnivnadHyn7RXYBC8-eTMsD79hvqI,8563
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=Tr4uaPm1SNeesKts8V_rx2q2TjpV87K7EtcFecfDGeM,267
attn_gym/sparse/indexer/__init__.py,sha256=Qx9YN9e6mK4yGdNVQmhxC1DgivIoab5M1XgZUwkJTMI,115
attn_gym/sparse/indexer/api.py,sha256=7RzOVhZ8CR7pc58R3-mqMdvZR7Ua2q45riDoU2gsHyM,8018
attn_gym/sparse/indexer/ops.py,sha256=W3SnaqnRx5ZJMn-Tn_5iged8NiGcb5z7R261JjxIDCQ,1621
attn_gym/sparse/indexer/impl/LICENSE.Apache-2.0,sha256=cJ5cgC1m_KtD_EMPMM0vTpK6jZiBshr_abakNiDd__Q,11437
attn_gym/sparse/indexer/impl/NOTICE.cudnn,sha256=3BeuD_qUPlGL3kJGvWinoxvmOC_ALNQZkoWK0liaC_A,575
attn_gym/sparse/indexer/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
attn_gym/sparse/indexer/impl/cute.py,sha256=C-83XQoWRgo30NCir9OXhFbaXevygdtmwDaD36yTAyQ,9896
attn_gym/sparse/indexer/impl/cute_score.py,sha256=xSj9mk7uFW3WSiPy9a17VU5YnrRYS30W2oAY8yntJGc,18990
attn_gym/sparse/indexer/impl/cute_score_generic.py,sha256=LKl5R1vkPpZwkxAbNtZkGGUwdAtNXAexIcDQSroOo4w,19077
attn_gym/sparse/indexer/impl/cute_topk.py,sha256=i1L2IS5MBo8BPJpH3d1F22v_Wp-ROiWpSEBcvV5osY0,13880
attn_gym/sparse/indexer/impl/reference.py,sha256=DST1PbyrQl2xtKHHrmDFRDdmNBe225Ty3gz-33jrQLU,2342
attn_gym/sparse/indexer/impl/triton.py,sha256=uS2CbCfGFySyD-9WtDGpompMqSUyLLyPBGa9sGvA28M,5974
attn_gym/sparse/selected_attention/__init__.py,sha256=iKeBPNwewL-7CgJXSKWDvjLVt0Wcy5NWaRR7q73aieA,219
attn_gym/sparse/selected_attention/api.py,sha256=9TG0apHeKxKAAOrqvW3u2hBJMcSiwYQgLdQ1ii_-Q1M,13670
attn_gym/sparse/selected_attention/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
attn_gym/sparse/selected_attention/impl/cute.py,sha256=C9mjvtTk-dPPqDZoOu6fh9aJAQLoH7JNiIWcFQV8fcM,7669
attn_gym/sparse/selected_attention/impl/reference.py,sha256=j1aPf8C8Zy2T7sgGl6mihtZKCVXs08A8lVatgnCklT4,8026
attn_gym/sparse/selected_attention/impl/triton/__init__.py,sha256=etMRcr78RuWuUpin9Lm4ZIlDEvX8_8dlHrKCy1gUKpg,4822
attn_gym/sparse/selected_attention/impl/triton/backward.py,sha256=HRUoEjbyeyxn-Qp1zNVrAoQ_VqaELrniMk82hJy5jpo,28785
attn_gym/sparse/selected_attention/impl/triton/forward.py,sha256=UmL145RPFGmphAHTN7o9w0F7XxVxx7xiE1JsX_bcGzw,19189
attn_gym/sparse/selected_attention/impl/triton/primitives.py,sha256=Z7rlI39bUULxYybbA1OhPe0ZxwI20Cbh8rUNUoPnyrQ,3699
attn_gym/sparse/selected_attention/impl/triton/shared_backward.py,sha256=PVIpcxGIQBDLOnt2h-r_KsxNWyToge4Uc0icWNPtm2k,16471
attn_gym/testing/__init__.py,sha256=aL1W-rA39UZA-aCOIjUEr94TJU3hwJUpvb2uAbPOUM8,568
attn_gym/testing/delta_rule.py,sha256=jyq3Z2Y6d7w9jDSB42PVBp87GfSuPpPLB__t_BIevF8,6599
attn_gym/testing/gdn.py,sha256=g7S5VbHblzwUqUoAyq7p2z-eAOuWOcEhb-30l-hnwGA,4294
attn_gym/testing/indexer.py,sha256=R6I6xZkkHQCyeeZJaga-Cwyt5LPvceBE3fvPOy7vCg4,5442
attn_gym/testing/kda.py,sha256=atzV9g1PMNot25OjyNNvA_VeLVkGXQLIcrCXPXCHDkk,16520
attn_gym/testing/profiling.py,sha256=tOzECIJlJF1xQcNlxUY7bBGEzackfW94Ivd7JAhbL50,7013
attn_gym-0.0.9.dist-info/METADATA,sha256=kG6Ipk7WpRLrPlmxMPMeUj9eYgyqvPW6C5qP6GPSkV4,6244
attn_gym-0.0.9.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
attn_gym-0.0.9.dist-info/licenses/LICENSE,sha256=zxEbkIPIQLB5X1Fmmc209oHgBh2TxbdupicAi0GVK2I,1501
attn_gym-0.0.9.dist-info/RECORD,,
