Wheelodex » Projects » flash-linear-attention

flash-linear-attention

View on PyPI — Reverse Dependencies (0)

0.1.2 flash_linear_attention-0.1.2-py3-none-any.whl

Wheel Details

Project: flash-linear-attention
Version: 0.1.2
Filename: flash_linear_attention-0.1.2-py3-none-any.whl
Download: [link]
Size: 475888
MD5: 64716d7c1e98deae16f1567f05a83f72
SHA256: c823924cb1e8f53f27fe3588a5f34792ddd6545a0212ae462da2e4d79cac1f59
Uploaded: 2025-03-31 06:31:08 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.4
Name: flash-linear-attention
Version: 0.1.2
Summary: Fast Triton-based implementations of causal linear attention
Author: Songlin Yang, Yu Zhang
Author-Email: Songlin Yang <yangsl66[at]mit.edu>, Yu Zhang <yzhang.cs[at]outlook.com>
Home-Page: https://github.com/fla-org/flash-linear-attention
Project-Url: Homepage, https://github.com/fla-org/flash-linear-attention
License: MIT License Copyright (c) 2023-2025 Songlin Yang Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: torch (>=2.5)
Requires-Dist: transformers (>=4.45.0)
Requires-Dist: datasets (>=3.3.0)
Requires-Dist: einops
Requires-Dist: ninja
Requires-Dist: causal-conv1d (>=1.4.0); extra == "conv1d"
Requires-Dist: pytest; extra == "dev"
Provides-Extra: conv1d
Provides-Extra: dev
Description-Content-Type: text/markdown
Dynamic: author
Dynamic: home-page
Dynamic: license-file
Dynamic: requires-python
License-File: LICENSE
[Description omitted; length: 27874 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (78.1.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
fla/__init__.py sha256=Nvn_zpglXVZb06l4ZlQkNPY03X6927GTYtuZZn9buIA 2151
fla/utils.py sha256=pWB_2pAnMggw032zfAiyUmIqzT5Ac77BCS_PySI5Sl8 7220
fla/layers/__init__.py sha256=LA88WU_CbHK_OliSS42j56APYxPEptnlkQwDPRIO734 1119
fla/layers/abc.py sha256=W9IsM5cftFTiJAx1SH4If_3hrskox6q68sEI6P88NO4 8339
fla/layers/attn.py sha256=IEEY_u_rTDvdJPJwMvd8mP9Hm5hF7go8-8RyHUR3KiQ 8290
fla/layers/based.py sha256=A5W2zkD5yE4lkAZDn7hL9Tmwu4hO8iUIJsaefLTC_vU 3947
fla/layers/bitattn.py sha256=o3qASACsYJhTX-uBvzoUhjARzhdcaAw1USOd0ujSIvY 7872
fla/layers/delta_net.py sha256=SZHZtgIeiLku1Lqo7Ruces5-uq7JlIBEn6RRlK-6o-U 11499
fla/layers/gated_deltanet.py sha256=ncxrKXfjHG5RDoKUdFUE-8YlO62baOSel-kbaBcC2Bg 11768
fla/layers/gla.py sha256=dktYlPmFasJ6CA5Bb6MCeEwx_kNWXvObvyipYnli6_4 12317
fla/layers/gsa.py sha256=tEXX7akLcGG2Bi-Fl4MfDBI9FCEWuzjkll0Ke-lXs8A 8882
fla/layers/hgrn.py sha256=URDj3C-qheCsxhp1A-_EdWihVZl7aBDHwG_6vdbLPEQ 5949
fla/layers/hgrn2.py sha256=vdMWqg_7cS8WOPKm_zCh7-v3nEBSoCrc_GylGIveASc 8123
fla/layers/lightnet.py sha256=124wlYH-frHopc8o4nuyxuXVNAtadaTAQUzvKoOPj64 8026
fla/layers/linear_attn.py sha256=HfgUjktnZ4dhE2lFpXbGfQ3pKeB-hx9MLHT_TK_2Ro4 6152
fla/layers/multiscale_retention.py sha256=tk_eMiDc1iZz0_Pu-LMrc2dJrWjo_gs1WX0kohB7n7E 12469
fla/layers/nsa.py sha256=NGARicCKc8bdsXZzRMnFb_AGZgeh4LhHfZawspjY3Cg 5303
fla/layers/rebased.py sha256=r0ncRpkkwzzsGyDiDQMSscy4UpTU9c90Bbx_Vw1p8VQ 4673
fla/layers/rwkv6.py sha256=-Q4agXJ6LmpwRAjEbubwoqCg6k4zbKGxVQ4DZRXb2RQ 10963
fla/layers/rwkv7.py sha256=gLW6i_uPWz11kzYPw2pzW6gLNTISxJv6tRvmQuGi7Pc 8532
fla/layers/simple_gla.py sha256=KmSnVYxzfPWd0JJ5nrugdAMfZEzCl6o9IeARNKl0IAE 10902
fla/models/__init__.py sha256=9g4c8semznNXNKr709GVv5BNnjj8NF1HG7fPcScMhaI 2437
fla/models/utils.py sha256=p3NXi5v4kNaYXx5PKXhs8Do59Th-0NQBii9PlV0Bzxs 5712
fla/models/abc/__init__.py sha256=LAaIlrfzQnsBwzUZzZ5vj1aTjiH2l-mrVhg_qZMFACA 422
fla/models/abc/configuration_abc.py sha256=nfZDBmM_mlXsiUqiRzbyhf7e170USJSpFlwmzz9Kdzk 3264
fla/models/abc/modeling_abc.py sha256=naA5wKC9OIoHG0pbB8oMGyrFgnI8Ca71FGbOCPvZJCk 17632
fla/models/bitnet/__init__.py sha256=QkE6xnQ49Se2dcjcrmMgIq9amHjK_PqTPj9-BrO5Kfk 470
fla/models/bitnet/configuration_bitnet.py sha256=Rh0LBuAbGf1AFjQstNBtgIqjgGJ-faQiFZr6T--vAcE 2125
fla/models/bitnet/modeling_bitnet.py sha256=LuzgdhHOa3oHc9TDw5czhNjprgFY2alNk5f2yHHww4U 17316
fla/models/delta_net/__init__.py sha256=hlvejYUtfNqD1w9xQURnA-I4hF_CpZiuHJpo8GQWb-I 505
fla/models/delta_net/configuration_delta_net.py sha256=i2r9atymPMqrNr6ntnQbm8Snc-ciyB22TA8yT1Rlfso 3229
fla/models/delta_net/modeling_delta_net.py sha256=Ds168ja-QJS_rZRXasb-dUv8Fz8vVcWxIb931ldtrzY 17693
fla/models/gated_deltanet/__init__.py sha256=Lb9tsi6EQo9kBFvCRsWNVbfQ7tsKw83XdnqBzoDu4EQ 585
fla/models/gated_deltanet/configuration_gated_deltanet.py sha256=D-P_yV16wgfBVRRIqwl-DE220bqhJcYDz4M882btKEU 2950
fla/models/gated_deltanet/modeling_gated_deltanet.py sha256=Sb_97rthjghfrwubK9RIT9TSC_Qy_14_dA9lva6_wys 17609
fla/models/gla/__init__.py sha256=RJVHn5zvaul52vBNEwQRqb_PKNbQgQWMLmLEQPTBbbU 422
fla/models/gla/configuration_gla.py sha256=-aq7OooqvGCpBJR5tzReLecu9PELkUYuCXZdmF6Lu-I 3417
fla/models/gla/modeling_gla.py sha256=ne_cnijyeCxMCNR6Dr6Fn4gXxmTVRo51r-eHhWSU1A4 17734
fla/models/gsa/__init__.py sha256=gFtWIDLdBo_Vhr4zhMw1P7aa1EQ10TAkpmGG7NRM6FY 422
fla/models/gsa/configuration_gsa.py sha256=5a2zS4V7g0EPGbqDy1nwypL74O3cva900oxTUagJ3EE 3552
fla/models/gsa/modeling_gsa.py sha256=ZhQN4O_QM3U9hLxSB2lVSIJ1Pcul5RR5sYxQhX-zIqg 17841
fla/models/hgrn/__init__.py sha256=uxwlyy7jH3r2Yv2poGLdNSAV0xOpt5SNihl12THtY04 438
fla/models/hgrn/configuration_hgrn.py sha256=bMYPAWRKPPbqrAzo2HDE0hc1PTk7s47i1Vs-U1kKd8U 2954
fla/models/hgrn/modeling_hgrn.py sha256=J6gdkepWifG3Y3nES22oPUwVX6OnkcPsdQvv_VPcJQA 17959
fla/models/hgrn2/__init__.py sha256=pGbf2-pcCL6XTcgEWg-TqGIAjy7YAGHl_dBSc9iuQo0 454
fla/models/hgrn2/configuration_hgrn2.py sha256=jlFQJhevd9Gq71hXpJvbjeUnpRykhbFEo90omoOQpqA 3408
fla/models/hgrn2/modeling_hgrn2.py sha256=GN8b5bjUXUaiU60iJXE8MpQv8y2IepfTtQ6BXhqYX6I 18024
fla/models/lightnet/__init__.py sha256=Ud2dFB9wRe8KiUVPNYKeHvDUFyxbAbj8NGPt0oVgOaI 502
fla/models/lightnet/configuration_lightnet.py sha256=iTvnoNGlEScGMzQeqqnN6z68TIsh8uu_i0q7QLO2kPw 3044
fla/models/lightnet/modeling_lightnet.py sha256=Ju4Z_ql7kgzuqsJd1GnErwG11I_37srL3wT3DJXGyxg 17507
fla/models/linear_attn/__init__.py sha256=PXKyTTtwrGvgtB4I2ok7-vvdqM1e_ta3RZ2rPJ_Jq9k 597
fla/models/linear_attn/configuration_linear_attn.py sha256=NiR6U-ycIO2pJPFElDIoJEW7DrHx7o5cmgtSpX00J8M 3287
fla/models/linear_attn/modeling_linear_attn.py sha256=QJWARm4KO01Sx4nkuLBNzOkdq8U82BGBzBel837Sc0g 17793
fla/models/mamba/__init__.py sha256=7ueim4K7LiLr81XEZezfnuahtWFWkBzHSEk3vvGL90A 498
fla/models/mamba/configuration_mamba.py sha256=Qy28dHQYLldX3bsU_0ZYaSxrvk-ko5hXfg88qkWt-qY 7375
fla/models/mamba/modeling_mamba.py sha256=tm17irAjAwnNo2dqy-oJ0M8xJNmzY7vSWrn48oUIjIY 37844
fla/models/mamba2/__init__.py sha256=2TD_ttYHF66bi92Pc6gk55e8Sxd2pM7wFG6QZ3PUrRQ 501
fla/models/mamba2/configuration_mamba2.py sha256=C5BA9kbO8PKU5JKwy3s5zKmxK_1rlOK9sJ4pH4bYwpI 7870
fla/models/mamba2/modeling_mamba2.py sha256=QgF_3TTBkmR7IbDSuGQAW_JKdNAIX3nfdttFPF7Ko9U 49329
fla/models/nsa/__init__.py sha256=ayYGweJAduPIhuxvvpKzQ31Avq-tUqjkBXa0CI5Op0c 429
fla/models/nsa/configuration_nsa.py sha256=fcOwtaRMh7c6fK07fcNYvV_zTsrCLQkeb4WoEJORUfw 2391
fla/models/nsa/modeling_nsa.py sha256=O0XRBv6JJoGk8ciKD4ZtjTloVjvmQ3FBr1cWD_CeV2Y 16830
fla/models/retnet/__init__.py sha256=EEHU0Y2Y2PQzwd4hTAotKxGW0IRusGo50OwjA9Z1lRM 470
fla/models/retnet/configuration_retnet.py sha256=9VovB4mQM69-Jxkp0d-r1KKB9OOA62rijbcoGLiwlgg 3315
fla/models/retnet/modeling_retnet.py sha256=NbEHGRLfIslcB-mmwEXYONToUo9ZE-7Q0LvFTaQgDRs 17911
fla/models/rwkv6/__init__.py sha256=MBEylIChCkvStj33te2g3b8RM3sZ9qdlRqYq2r7rqYg 472
fla/models/rwkv6/configuration_rwkv6.py sha256=Lm-oSZbBBk38FDCs0Zs52knpyxC-Hc2_ZeBn4yZzUpo 2966
fla/models/rwkv6/modeling_rwkv6.py sha256=3ipC0kDeSKJxuBpBLW4fiB710LyHB_WA0qtpCEXabrA 19727
fla/models/rwkv7/__init__.py sha256=k8xUAwwQARw8g6o_qB1MlkrnE1erN_jskbo9bfPicmY 472
fla/models/rwkv7/configuration_rwkv7.py sha256=T8N7xebzsDojGZ46zcXrJPyktMbpdpcs20uPeW9kO-s 4123
fla/models/rwkv7/modeling_rwkv7.py sha256=qJUJbDBxk04B0SSqQJP70Ta4mGjxWJQEMsf__NZs7aU 20089
fla/models/samba/__init__.py sha256=5bXJeZOhyAH3VLfcWGn0ifi7zBrd4tA4pW1dFWHbC6w 498
fla/models/samba/configuration_samba.py sha256=N13gkIjPUTIcJENOq3yRllq95-3sUZlI3rE4Dvnww9w 3151
fla/models/samba/modeling_samba.py sha256=HREtsf3RfkwTVg4CstPEbilsfpwoimSPc89Uvq0HFlk 17747
fla/models/transformer/__init__.py sha256=nxPVpEu4-_KlxtjF8BfBBLgaIJ9WbuqDFcXmj73ceOk 550
fla/models/transformer/configuration_transformer.py sha256=CtpcLt_eSXAUwhs7tG7fmscUmALHnUbHrMp-rFjhfG0 2262
fla/models/transformer/modeling_transformer.py sha256=RKdg529fPNqTGebFn2tqO2ML4j7gfl-41DjIsZzye0A 16150
fla/modules/__init__.py sha256=D4TsXbbzfNpRilZGLaiLFf6USXpMKQlLg9DOvMH5j_U 1295
fla/modules/activations.py sha256=3uDujArhw5TCtT1ZaUFve828h8WuhsZF09rZ0GNVsBY 12536
fla/modules/convolution.py sha256=qAq7aliVTcP6r0KFzy5V6gYNhqfZAulVsY3vMQDSxdk 15009
fla/modules/feature_map.py sha256=bmYn8KnvjSgQX16lq69N0GDSobBLKwSRtOeJo6MEZxY 8972
fla/modules/fused_bitlinear.py sha256=NiYOuVKZW6JME5tsFW0zgdhy2l8CtVp9Irgc6pgFULM 20778
fla/modules/fused_cross_entropy.py sha256=-DFRzW-4dDrVlThQuSBn6EjWVBhMYRyYjk6c6Zhfn8Y 15492
fla/modules/fused_kl_div.py sha256=o1kcdEpdOJYMdpduSF8mM-gJsYN1-lRNiTjli9HYs18 9814
fla/modules/fused_linear_cross_entropy.py sha256=Ht1NVEFqsz2QBTxsKNLsDJMJ5QuFsAlp3KGN168swx8 20107
fla/modules/fused_norm_gate.py sha256=ThtW8r3bKx9bBF4BPLfIjwxyPb3kb4DOPD_Tbwifo8A 30046
fla/modules/grpo.py sha256=-56nyzBD_stq_DiBJaji_ak9llSN5s0phb5hPhaz9ZU 15363
fla/modules/l2norm.py sha256=7O16IW8iZHDDPl_agB1zeOw41Pd6xmS6VY74QpirGUE 4215
fla/modules/layernorm.py sha256=RolijimDRRZVDFZO5KcczN3VasxqDx1QC8-_R5D3m1c 33700
fla/modules/layernorm_gated.py sha256=N1mPxvdeMEZs9ZTw8VJs5LEN0omGYL1i6c32GQiXW9U 18157
fla/modules/mlp.py sha256=i6VCuudGUKioHBH5pll-0YUOju-_3oVsYQyblHCvJ88 4746
fla/modules/parallel.py sha256=9h96Zgk0Z1Dx2czlyym2C2JvvplyTwDOizrFVIuNleg 1163
fla/modules/rotary.py sha256=qc4kAYXqQ-cVH1PBKYxkruWu3YTIRkB_zpLvQac2mi8 18814
fla/ops/__init__.py sha256=-cks8PGXRvESq1Uwpi7JFY4Mt7mnbHRU1ksTMWxEMwY 2077
fla/ops/abc/__init__.py sha256=QUEHkdGKRnmnKxkbsXRQjyb2KW7IVO77hmdEoGWZwR0 85
fla/ops/abc/chunk.py sha256=2N5yneczhxsT_yubrPyxLfjwP4hb76cDaX_mysMX4wM 44112
fla/ops/abc/naive.py sha256=p8FN8IFk8BWJNyGE94a05LvjsKFDoTWddqV0FkijUSE 3066
fla/ops/attn/__init__.py sha256=ljDLWNKcuMJN9BU6YpMVjQ9tTNHtp-1GEDQlNYxjHr4 96
fla/ops/attn/parallel.py sha256=_VzxQDS1ANlu46ntQ2VVy92tnn7D69j3AbPSbJRJgsI 20398
fla/ops/based/__init__.py sha256=p82p7VOLu7yJOVlZk_vhDqkfR_taCam1mDOvpJqCt5I 166
fla/ops/based/fused_chunk.py sha256=78a3jTBlpJkJ16jf6J2Dy_62uZifhiNsFnzQ1ldJDEA 13150
fla/ops/based/naive.py sha256=QCjeQ7ELaAzo_HwAmhPk9Hu4rd10z-L_mcxWCxVUUBQ 2268
fla/ops/based/parallel.py sha256=-vVBmSCwMlLXQuKM6slGCg-8RrAjNZr3i6PNCg6Ffss 13551
fla/ops/common/__init__.py sha256=iwhKnzeBJLKxpRVjvzwiRE63_zNpIBfaKLITauVph-0 24
fla/ops/common/chunk_delta_h.py sha256=NxO3z4fu-v9h7KOHJFLHlGqma7vb9VA4F9o1h3S2APE 15309
fla/ops/common/chunk_h.py sha256=x4H7piGgck76OvLL-HEbbm0hYSiDn9sNv20Wempx2hQ 15544
fla/ops/common/chunk_h_parallel.py sha256=rv_9W7LORpl2BPco7szOFn37Y8xveuL0G-8C0XNJAgs 23342
fla/ops/common/chunk_h_split.py sha256=LfUaJZ2c7wECWLGsJvF649l6JNa1Jx5WRPSh7Ygl3eM 24738
fla/ops/common/chunk_o.py sha256=p9C-76ZaW-Ugz2KWMazwbZw7xq5j-Z9CGbfg1V9Lqlc 22021
fla/ops/common/fused_recurrent.py sha256=BvE4Qutc0IoCkMMkgP4hou4mQoc3BpG1aInvDv6HzUc 20615
fla/ops/common/utils.py sha256=a0qyp-Jb5x1YBZuvmbG9m_EXubdfNJscdjwIfUq6Qvs 1901
fla/ops/delta_rule/__init__.py sha256=AwLQ3i8BflftL5BYoAt1xZ4GhLoyIvUUifKECBV5BKQ 267
fla/ops/delta_rule/chunk.py sha256=u3NPCC7VFWRB9d1e1cqW4zY5aA_craIFuRlHxq7F5Vs 11908
fla/ops/delta_rule/fused_chunk.py sha256=A9l5hK0awZ-m9S9D00azj14tDFNLYlUL1iuaz2XyosM 177
fla/ops/delta_rule/fused_recurrent.py sha256=7VvT0yBfx9IyM_cvOOMCCOzmUZ5nrWtFk8uBlxrDa6U 21942
fla/ops/delta_rule/naive.py sha256=bOmyp-XijFlVjeD3wsXXXs2kfimGh3wi-oGIzm6vup8 4588
fla/ops/delta_rule/parallel.py sha256=AYYeH4-YIlk3OJBrwfZZr0ybvALQJif8uLlHBjv0fmo 13990
fla/ops/delta_rule/wy_fast.py sha256=uws6UlGAagTCfs1YKITYG1jPiLZ9wn3cdZD9HQmuD6s 19905
fla/ops/gated_delta_rule/__init__.py sha256=KQ-vtlVpGSfLpLCfMwP-p6bhyqvfLEls8l5_LLoYmHM 188
fla/ops/gated_delta_rule/chunk.py sha256=XcTPLgc9m4er4SiKrcsPwPbnT94CEYIZWlcLZlHgF8k 12685
fla/ops/gated_delta_rule/fused_recurrent.py sha256=SubNyJ251jxn1F5mS3VQHggwbdoDNmPJStmU9QNO_pU 11817
fla/ops/gated_delta_rule/wy_fast.py sha256=Cp_cssS8AeGwoR0vAIuI0yqLf4bJYcWiYqGgRnJKF6Y 25703
fla/ops/generalized_delta_rule/__init__.py sha256=l1dPdZ7SQONF2V8FJYxG37i3lxTH0ujSxG0MAiKLphk 296
fla/ops/generalized_delta_rule/dplr/__init__.py sha256=R_61Vpu8z3TmHrfRJsiNd-NwaAbX2AUhkmIxSMMjeRA 184
fla/ops/generalized_delta_rule/dplr/chunk.py sha256=KWIrDKcsotd7Xek1PzXfbS_5c7L5ArWk5dXH-EEnJoA 11789
fla/ops/generalized_delta_rule/dplr/chunk_A_bwd.py sha256=yhqdanKSSie-VmfKE7_TmJPd8iqCdbWdTBJNUMCQSJI 18085
fla/ops/generalized_delta_rule/dplr/chunk_A_fwd.py sha256=_clFrN6iOpag52By6E15lyYXJvehO3TitXGojAWcCYE 14456
fla/ops/generalized_delta_rule/dplr/chunk_h_bwd.py sha256=JZmgt4PeRvvUwCn2h4Z3o4a-WahSa3TQFFkMKoDembk 7357
fla/ops/generalized_delta_rule/dplr/chunk_h_fwd.py sha256=ufaOEQgOHHABwVrg-OV_yxAR8QyUE7w1dYDzDKFXZCM 7502
fla/ops/generalized_delta_rule/dplr/chunk_o_bwd.py sha256=e3gqx3AKN-B8Fsma-kzLUlFktDFcPMAvg3HMmowpfuE 16378
fla/ops/generalized_delta_rule/dplr/chunk_o_fwd.py sha256=92a-OGM0TgJPj5qEyFy7pWv2cosIoidgNhaTz4g-uLc 4895
fla/ops/generalized_delta_rule/dplr/fused_recurrent.py sha256=9f7Qd0XXeB-chXy627_KRg0hs4H0-ZZxhhNG_QCFvOs 9746
fla/ops/generalized_delta_rule/dplr/naive.py sha256=16yEmXHx2krNZOjGKCARnNOamLFIu0OwzyEtr8bt5ZI 4037
fla/ops/generalized_delta_rule/dplr/wy_fast_bwd.py sha256=LQsCw_9iSfSP7GHBXw4XEGqIdFXqVeToVuskJAdO4MQ 7819
fla/ops/generalized_delta_rule/dplr/wy_fast_fwd.py sha256=SEWR9Eugau98C7CZBS3OIwmqrQPes9GBwItSsViqr6M 12040
fla/ops/generalized_delta_rule/iplr/__init__.py sha256=17PKuyeOEivsooLULz4oOt-Fgzj3Z1kbg-RAWaCbEWc 184
fla/ops/generalized_delta_rule/iplr/chunk.py sha256=WQ7ggsdmt52DdQyh_hrNZpjuORy4u0JYrpGEjt2oUno 18356
fla/ops/generalized_delta_rule/iplr/fused_recurrent.py sha256=J1e04LPJOrzyNqT-yIz_ZEt6MTs_CjT2kU-tmS6jlws 17298
fla/ops/generalized_delta_rule/iplr/naive.py sha256=iJ6dOCnu5ayvfwqqhi6vECvmRCfOowrWEFWOuACA0A8 2681
fla/ops/generalized_delta_rule/iplr/wy_fast.py sha256=59R_etU4dmAtBcpW-GmG4AWOmzM6MV-nfrXx-HcKGbo 12665
fla/ops/gla/__init__.py sha256=fWIInNb2TzIEikB48TqDxzuQpmfc-yuudzyCWTDr9KI 225
fla/ops/gla/chunk.py sha256=aWv5Ypta4TuLdD0GFEGcTzLRAlRoWUA2rbw1-3uh0uA 52743
fla/ops/gla/fused_chunk.py sha256=gwqVWw0gGFJyj1G3dVzl1j-NsL7qqvoY8G8zs-KkdnM 21585
fla/ops/gla/fused_recurrent.py sha256=Q-u1CzMJc8VTwyXmGzB7oQTmeORC2xLjqNt0aEkmc5g 5197
fla/ops/gla/naive.py sha256=4a2aOvcXrUtDc-3n65CTmVdDuY2sIPOLXoMDbPVRgVc 959
fla/ops/gsa/__init__.py sha256=8l3frZMFLZYWrVcjJ-0UusJoHkaaUOLAS9zovbc11RU 161
fla/ops/gsa/chunk.py sha256=J1UBzhHbC8UXM5syVehzqmyoYm0ZjhQPLABloZHAnpM 45596
fla/ops/gsa/fused_recurrent.py sha256=wa1LWsY3Pdx1srQpqJQbOxWeku9zAP2N_Co44OSd0q4 18096
fla/ops/gsa/naive.py sha256=8jPogc6WcdMQMvz_RD8-5_4jLa7fj39hCDv3Es2mwz8 2091
fla/ops/hgrn/__init__.py sha256=Dz15ecN3QlZcaxO7H7wqcnp1D9C-xbyOcESV12nQ6xg 165
fla/ops/hgrn/chunk.py sha256=ynNHleIv3pXXFV2FvrOlJKFkqtQE63P8UiyXqFOs5MY 9396
fla/ops/hgrn/fused_recurrent.py sha256=xn2pxz3fwBPKvJTigeU6RiccQsNee0Eg6GEHIkXniJI 9055
fla/ops/hgrn/naive.py sha256=7JIHYJ2mvC5WST9zdmJjxzStw3MLsSVWmNWcMGMxUE0 1582
fla/ops/lightning_attn/__init__.py sha256=1UpOY3yb4lEOUAObZL4ytAbrvx10pKhnaDLeGN21N7M 205
fla/ops/lightning_attn/chunk.py sha256=V5ZbzXMC48DF5jlNiJdWjuGJcEmUW6pfzcqmpltpnGQ 2874
fla/ops/lightning_attn/fused_recurrent.py sha256=cKuIe5LRrjRKSDgenULa41jGXkLcR6Wth-VSpy9t7Lc 2942
fla/ops/linear_attn/__init__.py sha256=54S8TKrm3tXjWkaa1lM37m-00XDaGjMzrHo7qfCvw8U 273
fla/ops/linear_attn/chunk.py sha256=MVe4ZO-UgMcwdrRYh5j9GpLM8mTyHpG8LVHKQeR0U10 2172
fla/ops/linear_attn/fused_chunk.py sha256=ShSTYvobCdLxvaLQYXiNcxZWLAAVPV1mTrdFvbD8bxY 12542
fla/ops/linear_attn/fused_recurrent.py sha256=sdj6_R4bokq_ERQ3TM6AIDIbQSdqcHiBM9s7f66kazo 8415
fla/ops/linear_attn/naive.py sha256=Dj8v7-ZW3r1UvKeQ96idqOIsvMgF0u8VDaLHGGDC5TE 1098
fla/ops/linear_attn/utils.py sha256=AElV0wSGe8S2MAORPWWbmF2yd8jfAPE5nTNpuahuRrQ 175
fla/ops/nsa/__init__.py sha256=BM8WYjgZguPm3xRx_upTAV-e7zOzf-yQ4KpfJhnihbc 140
fla/ops/nsa/naive.py sha256=O6asTbtMNsEUj9q46zQMjJvuCUU0u7MuQGM0N_u0c78 3732
fla/ops/nsa/parallel.py sha256=OJQG7-UFdfJTI-NQT-5JoJNKbnooaVrA8ZbZ-Hhzkr4 44370
fla/ops/nsa/utils.py sha256=S_flR21OxQUJ7Vud4r1d16lrsnWZhlEkXDV8o-AdvLc 3261
fla/ops/rebased/__init__.py sha256=q8947lHsR_q7huKqtlU5DQTkjXUsYwlunZfSxnbF0rQ 102
fla/ops/rebased/naive.py sha256=p7MXUav8MdEWuSVIBzeO5jdtTQc-rfBG5F_6rq26TNk 661
fla/ops/rebased/parallel.py sha256=LvHHgoOyQ-ENZ6aBONLRFIPAItAEc08Wq622ssStkYY 14182
fla/ops/retention/__init__.py sha256=AAUJH_NpTvaRZ6HYKdf_gquVbkj6JTeut_t2EYYcfss 328
fla/ops/retention/chunk.py sha256=BDjSTXL64m9O4vB5VdeciGIiyq4QDLrZvvCGTRlwHMQ 2683
fla/ops/retention/fused_chunk.py sha256=XP-Vzmme7Xr2Iy_5Mgf-pXr6N-nNBQQdE5Nn1RMCJTk 13263
fla/ops/retention/fused_recurrent.py sha256=Pr9ScdZtn9bXKU0GX9rNAQ8bByNHITXdWJtlrUtCsgs 1253
fla/ops/retention/naive.py sha256=jdvU8CoMlpZmeEcX2lOY5pkBM1QC7HJkQxYxMqZtNVg 594
fla/ops/retention/parallel.py sha256=iHJRv2OEUdqIitoUPEZlcA_QVU6el1xkSi1-WR1sAYo 2283
fla/ops/rwkv4/__init__.py sha256=dDcbCFnE6qSPne7oEOrXR-sC75xZ5CBpAxCQZj9PVoA 119
fla/ops/rwkv4/fused_recurrent.py sha256=lXE9yhu_-1CdkaNdbzy0dS5shgGWGvdFYap2WXgEFDQ 13202
fla/ops/rwkv6/__init__.py sha256=407X-jUUyUSZso_wxQ_EqyGZhss4ByCgnPa3iJLgRxs 169
fla/ops/rwkv6/chunk.py sha256=k1juh0GzrFKJI_b0DkDktC85-VXq4hzJT4sBszzREZ4 52400
fla/ops/rwkv6/chunk_naive.py sha256=HWqwgpFI08CfEUUuI9CR40CkF2yzG76V7houwbB8pMw 1481
fla/ops/rwkv6/fused_recurrent.py sha256=toN95XaGJ-OpH1UUP33rczpNucmTuW63fgW7ycz3SGM 26159
fla/ops/rwkv6/recurrent_naive.py sha256=GA_47uLlMxXm8q4lU739sedFXeA8LR52J3jU7dXOhcY 3116
fla/ops/rwkv7/__init__.py sha256=KapdWK9xpcCQjU5GuiupY-AGfyKNN8ZwUS_vN6fLUII 169
fla/ops/rwkv7/chunk.py sha256=HJa6yqZ9F2HdK_MGrpYZXReCDI-2qwm_xd_SrEF0Ve0 2189
fla/ops/rwkv7/fused_recurrent.py sha256=AWZ1Hz5_qqQW17ZJKN0XaEHS3NLdJBO8M-Xj4buhtvo 2099
fla/ops/simple_gla/__init__.py sha256=vThuTaKk2Dx0la0GPDL3E0jjxRf82N2Hy22jwO22dI4 258
fla/ops/simple_gla/chunk.py sha256=ahi4ib5xSYRwkOKmFifLbuJ602x_hGZQUJbYo5fYQL8 10250
fla/ops/simple_gla/fused_recurrent.py sha256=uJHN1G1t08QWChHIPamvQuovYIXdL02JXuedGSUTAN0 5111
fla/ops/simple_gla/naive.py sha256=KTSVrA4dNRM7l275GQLKmLrOClEODie5xJxmjjHs1-E 1917
fla/ops/simple_gla/parallel.py sha256=mDiGLIoKUCeNKbHAyMa-IYgJpUSKTK_yZy8F7y67lVA 24369
fla/ops/titans/__init__.py sha256=3tL7-aSBRXQr6DAH8gnQF0OJmjjT2lGy9DkIPoYKHBQ 105
fla/ops/titans/log_impl.py sha256=lTdkhoXUhKL0bid6gnSRPgvJkstfO5Kt81ieXz0i1fA 4752
fla/ops/titans/naive.py sha256=NdzT2VkZVV2E6yI66-o0hJ7MuJOYTA_nKgkxuxq94oM 13827
fla/ops/ttt/__init__.py sha256=USHjg-yO51-Hs3aR5TN7v_VidhRzf9vgr7vdoFphNFc 177
fla/ops/ttt/chunk.py sha256=ODULVN5wPV65OzWs459jWIDuw6UfAjHH22XKvX0B2sc 56192
fla/ops/ttt/fused_chunk.py sha256=6piCeuLni-myiATf-I_2Xf7Hj3RLSY4Kha1ge9oirOg 33982
fla/ops/ttt/naive.py sha256=thICsCZyKW79-X2U2pOQC3D-phoMfH_oCwxUax0Hffs 4209
fla/ops/utils/__init__.py sha256=5pCd2HXmbZC6fGWg8Ctv3cm_k5RDEpKkFGZfWjPzhio 1315
fla/ops/utils/asm.py sha256=1fnG4EQR6cMSUJWLvC0PEF2lrD05ofJ2W_X-Sgt6Gj0 401
fla/ops/utils/cumsum.py sha256=6VEd2t9UbcmaAL7ILSTtAF_bhXZkhnoHcVbLgnXPiwA 12531
fla/ops/utils/exp.py sha256=Q_PGyhe5jH152XVHmoB-PksqrqqJTOmh5AeQ9sA_Df8 364
fla/ops/utils/logcumsumexp.py sha256=WnZeAx7h95unDDbNTmMT_tt8O7GYFG5rUfLsSZI9aZ8 1470
fla/ops/utils/logsumexp.py sha256=BzzkVeQDeoiJo3sYSZVnxQ-WTkr-35s8gRMF_-KBCO4 1870
fla/ops/utils/matmul.py sha256=5HKQMp7oTgOTSO_cPq68MmEdYp-Dko2IwJaHyIYeBPk 9128
fla/ops/utils/pooling.py sha256=2g6c6YsdrMhqegJ20UjmG0kEszhJwKeuM1B_CUbkVSw 6410
fla/ops/utils/softmax.py sha256=S19_aXTV0Yl_zHsVHbp7vKMNr4BetrKR7nMW9YXRNmk 2351
fla/ops/utils/testing.py sha256=COwaqHLs5FRzRjMPp4rY3QzOiNOG08zFUuA5E8IyUp8 792
flash_linear_attention-0.1.2.dist-info/licenses/LICENSE sha256=mNkTdA3RZGLXnKlMiCaaFmZzBY2VyUH4_JjLP7Oae0U 1074
flash_linear_attention-0.1.2.dist-info/METADATA sha256=LEQQbSVyRt7b04XJTx4lDoy8CR_E_DL-Ylnm-NcQwPY 30174
flash_linear_attention-0.1.2.dist-info/WHEEL sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0 91
flash_linear_attention-0.1.2.dist-info/top_level.txt sha256=0tmKzbo6mWd_gq5H-Kr3_6PQSx7juwxCBu6lQcz9P78 4
flash_linear_attention-0.1.2.dist-info/RECORD — —

top_level.txt

fla