Wheelodex » Projects » attn-gym

attn-gym

View on PyPI — Reverse Dependencies (0)

0.0.3 attn_gym-0.0.3-py3-none-any.whl

Wheel Details

Project: attn-gym
Version: 0.0.3
Filename: attn_gym-0.0.3-py3-none-any.whl
Download: [link]
Size: 16168
MD5: 7b582b5b61f96d4d991e9ac347938712
SHA256: e28bd28bdafa41241135c2daaeb308dc0208c484b39b4dbf4dbecae0cf7cfce3
Uploaded: 2024-12-09 19:03:52 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.3
Name: attn_gym
Version: 0.0.3
Summary: Helpful tools and examples for working with flex-attention
Author-Email: Driss Guessous <drisspguessous[at]gmail.com>
Classifier: License :: OSI Approved :: BSD License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.9
Requires-Dist: torch
Requires-Dist: bumpver; extra == "dev"
Requires-Dist: docstring-parser; extra == "dev"
Requires-Dist: jsonargparse; extra == "dev"
Requires-Dist: pip-tools; extra == "dev"
Requires-Dist: pre-commit (==3.6.0); extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Requires-Dist: docstring-parser; extra == "viz"
Requires-Dist: jsonargparse; extra == "viz"
Requires-Dist: matplotlib; extra == "viz"
Requires-Dist: numpy; extra == "viz"
Provides-Extra: dev
Provides-Extra: viz
Description-Content-Type: text/markdown
[Description omitted; length: 3704 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.26.3
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
attn_gym/__init__.py sha256=q1pnG6UQoynPcbB1qTMo8ELPK_I-Cc_M4EBnkXBAvUI 54
attn_gym/utils.py sha256=dnZOc9uE2K6CyW9IzCo4Zw72aIpzXgEIqaVp3j73xcU 6198
attn_gym/masks/__init__.py sha256=rSvo2Roc2bq_oKaMz4_l50IDqYsBNlMVC21v4-Kn1SA 318
attn_gym/masks/causal.py sha256=_fxaMeGgIAy1u2wNwwB6-9ZuW_H6vJ8CeLWpyldUVbQ 811
attn_gym/masks/dilated_sliding_window.py sha256=PBxW8ERi04D39cBYPEzn8VuZ_-kbNTlL5D79XmhBCVg 1827
attn_gym/masks/document_mask.py sha256=Fm3Lq-mD5NlGYkJyGBBMvuMeSN9IvxANowVWxYME-oo 3563
attn_gym/masks/flamingo.py sha256=RmjixPyPi8pEzcrWO8y-8Ha4B84nQKmWDlGShrJ5Okg 2843
attn_gym/masks/natten.py sha256=jjfTS5-cKNRKGLkYzKY6zOF8tLmSt6O84UZO_nF_Uoc 2954
attn_gym/masks/prefix_lm.py sha256=e7262LnzSnS4MsFhW7g-JtX-Hkz8WD_9uPTorpL1AFw 1637
attn_gym/masks/sliding_window.py sha256=o7Anp_RipH-UctAJLHI1i6I6IruOMiyp7BX9Kf6k69M 1577
attn_gym/mods/__init__.py sha256=T1vRoAx9ZyLZJ_qeWXa2kw2h4EyTY3RDomp69j6pIMA 112
attn_gym/mods/alibi.py sha256=emxvYnc5j6MTE_mdYYV9Way-YoVc5NWW4jM3UK66Gps 1668
attn_gym/mods/softcapping.py sha256=Q6m8yDPikgtMBmU96uXKl0V6js465hVGdQWVWnUFR4c 2805
attn_gym-0.0.3.dist-info/METADATA sha256=p4sU1t2Ed1TwEr02bkMnEKDQ_WuHpUfeqBblMix7gbE 4672
attn_gym-0.0.3.dist-info/WHEEL sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug 87
attn_gym-0.0.3.dist-info/licenses/LICENSE sha256=zxEbkIPIQLB5X1Fmmc209oHgBh2TxbdupicAi0GVK2I 1501
attn_gym-0.0.3.dist-info/RECORD — —