fa4_jax/__init__.py,sha256=-c4QSDHrjiEGEGsrnbVNpar5bWSwhobzjMnC8vJlAsw,1787
fa4_jax/_reference_backend.py,sha256=Ozs5Xz6Bd0ZJHclMf1nU4rAbQ-Us7YGajZwwDRwSeHY,2558
fa4_jax/_version.py,sha256=n_5vdJsPNu7wZ57LGuRL585uvll-hiuvZUBWzdG0RQU,520
fa4_jax/attention.py,sha256=2jerGXvRACfkAoki1Tf01Jjr80IK4SNgP2O_y0x5zY8,26122
fa4_jax/kernels.py,sha256=LZnzwE5AYyPmlYysrjj1_qCFI6P1QPM0-J-YUaudnzA,56339
fa4_jax/masks.py,sha256=EvuF0lu06n40knL5P_r3fdmVEMz4z2rujGrkvWcqND8,8435
fa4_jax/reference.py,sha256=dIT7R_2AxRDQ9JMjXSSu-79JMIbtlvxndS_FUO68b5M,11067
fa4_jax/cute/VENDOR.md,sha256=xm25ssg1yek0asqD6PbBEofLppfJr4ArlvsjvMOmFV8,10273
fa4_jax/cute/__init__.py,sha256=Mjro3LMTK-ZNM5cqzPPlIEAqQ1iZT7KM98Bdu6daUvs,351
fa4_jax/cute/ampere_helpers.py,sha256=ytrBcKfaN54nw20EQ2BIDXEM-ojQgtrSud7NWOSEcRE,3706
fa4_jax/cute/barrier.py,sha256=dnr4B6p1IbjRXubJU09HRaetIiSw2iuBbMScy56mwRk,2054
fa4_jax/cute/blackwell_helpers.py,sha256=qsIZdx1TnyeCjw-o-Tdtri7PLjfkriJXByoCiPx5gPo,49891
fa4_jax/cute/block_info.py,sha256=r2N5yOl35Ics5WTFd1sb39xE66Ht9QQogc1tBy7_rKk,7664
fa4_jax/cute/block_sparse_utils.py,sha256=q_sLE2jJbgizWkyWsHhZHUZvpGG0Bxs9JmpUwzXoGmY,72282
fa4_jax/cute/block_sparsity.py,sha256=-OLVEkJ-Zkfg3rkIpjGXirUoWkVi2ffO-mHaDdTQXCs,28520
fa4_jax/cute/copy_utils.py,sha256=kMyU5vbDHquZdqQisee2MgP9YI6xjsa4lVieAj6jC0Q,15878
fa4_jax/cute/cute_dsl_utils.py,sha256=QOdA60SRFRaCguwfx8FKIxBPl4Ky3kRaOnnTRYHYly8,10250
fa4_jax/cute/fa_logging.py,sha256=vHHt-ksct69kt43vCeeQ_0xv5kq0_Pjj2WmfROjTwKA,2915
fa4_jax/cute/fast_math.py,sha256=fl-CK_TQujaWdVjzrpoUCLQQy3JsMSIs1hcDVp0b5zg,492
fa4_jax/cute/flash_bwd.py,sha256=HpxGfzzXiFXS-LzdjQsU-EFp_WaYl9nKgwF6V8W7zRU,72102
fa4_jax/cute/flash_bwd_postprocess.py,sha256=EaKmUte8KzQSAZd6Qt3BM261tDQHaxuDH7wTv_ds-e4,32347
fa4_jax/cute/flash_bwd_preprocess.py,sha256=-pZBOM9u1GAEuDX7ZKvmY6Bo2zE4QU6VAJn-WaDv86w,21113
fa4_jax/cute/flash_bwd_sm100.py,sha256=PE-TWxjKWBipQIqB3Ls3GIAXzrGX1AXVCGYFVmxDX70,192225
fa4_jax/cute/flash_bwd_sm120.py,sha256=xgVFt1T-dmfz6d0HoYOHLJB2kSiyLuJdfwxmyMjQU4Q,1303
fa4_jax/cute/flash_bwd_sm90.py,sha256=3SxVaoYTE0FhJpMAuwidkzhqRcH9Kr4E3-17yCua99I,89544
fa4_jax/cute/flash_fwd.py,sha256=IchP4ay3qOizxWu9TbhJ9tEM-1JM9_ERA_CtlKjtfbA,56584
fa4_jax/cute/flash_fwd_sm100.py,sha256=z99yHMnHptNdBPIT5macKLTTxFMhEyAKB-ui71iKwFQ,167476
fa4_jax/cute/flash_fwd_sm120.py,sha256=84BHaXJOJXwLXp8lCU1yA8n_GGrT400XvFZ0ROj_jFo,1461
fa4_jax/cute/flash_fwd_sm90.py,sha256=gG5mf7CYxlkhFFKy1tpWyYmum0GeW5oBN69msht6A3M,73654
fa4_jax/cute/flash_jvp.py,sha256=sHnUPJQaPS_ilVEg1FkwC5PUAUOxYx0Zjex4ahbqFeo,34299
fa4_jax/cute/mask.py,sha256=nW4V9Kf9yIGNUN1tqXal2JlIJqVEYdPAqN_2JUKN77k,82007
fa4_jax/cute/mma_sm100_desc.py,sha256=rG7Ciy8aS3l8ORp4B7iPg7QwRS1BgmLD36ew81FuOfk,10900
fa4_jax/cute/named_barrier.py,sha256=H4D5a2kKKIjuEwHTd2ZdtunddLQC-A6qhn049ylNgTs,1753
fa4_jax/cute/pack_gqa.py,sha256=XLBLD2NzEObcS0RGBqesmZzq808pocC0_vkwHtWuCjE,11575
fa4_jax/cute/paged_kv.py,sha256=fcJlXHEWbcOuM6kt4W2teftOtcihA2jO4-xpVkzDCMs,9625
fa4_jax/cute/pipeline.py,sha256=hJvOkccr4KXHRYl54QyCkPReBSU6g2AOMPuVQMYe1Wo,14543
fa4_jax/cute/seqlen_info.py,sha256=IVV4PHVczNg4FMj74s8tYTswzAJDorBTYW4Qq5WnyVg,12279
fa4_jax/cute/softmax.py,sha256=TAySB7UqlaB7WCBUXBQAFTY6icB2pL5ELBxlefVRqtk,29397
fa4_jax/cute/tile_scheduler.py,sha256=bhbSJ7C0zl8KgR8WE6qdDg4YiuHfX_5mDd9S22nswFs,85458
fa4_jax/cute/utils.py,sha256=byDOpDio8G-kENVbqlg9RuZEgzOEa7y5ZxALeIAFhUQ,36915
fa4_jax/cute/quack/__init__.py,sha256=b9cbCBamWA_fHGjo7d4VUtojHUkxAwMKhzzGfgxpSZc,385
fa4_jax/cute/quack/copy_utils.py,sha256=7CwKybJkL-ksJjIoT1RElmhN30FFJwFKX-iqFmr8_FY,71932
fa4_jax/cute/quack/cute_dsl_utils.py,sha256=YDr1T1gH-gBB5_uwIeDGGD2CQib0HLBCtIXIp4GmtXQ,3086
fa4_jax/cute/quack/layout_utils.py,sha256=w8U-k0-bOJFgUDKEZEM2heHz2c6mPj_5xNDjgV3SfWI,17370
fa4_jax/cute/quack/sm90_utils.py,sha256=iy3_CS8d641ek9qKUEisZmD3vR9wRhK5V3B4EaCxtEM,11193
fa4_jax/cute/quack/utils.py,sha256=VDTFzjzuMu__a59xa4ZIjElkHdpFuigvK3rzXzrPxFQ,6992
fa4_jax-0.1.0.dist-info/METADATA,sha256=wLOLUuP-i-NYmO2aWUEIfP6IHfyRS9XQP4m9L9wt-EM,8717
fa4_jax-0.1.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
fa4_jax-0.1.0.dist-info/licenses/LICENSE,sha256=9CZ6gyWJiBxgFM6Aj1UBOwFB7-IUqcX8016lPpS76Zs,1071
fa4_jax-0.1.0.dist-info/licenses/LICENSE.flash-attention,sha256=jJzLlsBl5wYTW2y60nm3IdphVuUfOl8nxrMymvlBbXM,1558
fa4_jax-0.1.0.dist-info/licenses/LICENSE.quack,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
fa4_jax-0.1.0.dist-info/RECORD,,
