trifast

View on PyPIReverse Dependencies (0)

0.1.12 trifast-0.1.12-py3-none-any.whl

Wheel Details

Project: trifast
Version: 0.1.12
Filename: trifast-0.1.12-py3-none-any.whl
Download: [link]
Size: 14855
MD5: 60302bdab060747597cbdb9153fe96c2
SHA256: 7f8792fe536ad0a9a9f50315decb57199995de511e18c204d9faaab488da9ee8
Uploaded: 2025-03-17 20:10:56 +0000

dist-info

METADATA

Metadata-Version: 2.4
Name: trifast
Version: 0.1.12
Summary: Fast kernel for triangle self attetion.
Author-Email: Liam Atkinson <hello[at]liamatkinson.com>
Requires-Python: >=3.11
Requires-Dist: einops (>=0.8.0)
Requires-Dist: jaxtyping (>=0.2.36)
Requires-Dist: numpy (>=1.26.3)
Requires-Dist: platformdirs (>=4.3.6)
Requires-Dist: pyyaml (>=6.0.2)
Requires-Dist: setuptools (>=75.6.0)
Requires-Dist: torch (>=2.6.0)
Requires-Dist: triton (>=3.1.0)
Requires-Dist: deepspeed (>=0.16.0); extra == "benchmark"
Requires-Dist: pytest (>=8.3.4); extra == "test"
Provides-Extra: benchmark
Provides-Extra: test
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 1375 characters]

WHEEL

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

RECORD

Path Digest Size
trifast/__init__.py sha256=5g0mk6ScxfVMxIfDdiflab7YAi0nXa_rQAN816JEkb4 79
trifast/autotune.py sha256=J1B6XPYV2ce24EDEiZzy4UMRqcNMBrjDTbGqXcQTfyA 9249
trifast/autotune_helpers.py sha256=p41bRDOFjCAtqDEo2XSRmRzIckWecXshM-9Ae-msd9U 11752
trifast/equiv.py sha256=2LKT-ApgXUkScErjXptguNvhjIPoHIvoVlevxYoZcJc 3124
trifast/torch.py sha256=HIGOh2lHaB1TN12EwSmSx5qfPfJhDjVff6GC9iRbErc 7144
trifast/triton.py sha256=-25DhcGzRcOWBEFoKM-wYAeyhtEvupqNoz0CPp5pV0M 19136
trifast/tune.py sha256=6WDPXgB0yEgSEddU2NskgxtrfZ91_KtYm3u-c8Nh4TI 2945
trifast/utils.py sha256=Wb5Glxp4LK4K3oF0e1U857RmmHmJDQI-DK8POFLc8hI 2367
trifast-0.1.12.dist-info/METADATA sha256=s85hPBs74Ei_WBlaHlSfqyFIseSf1scHrUzcds8h_5o 2010
trifast-0.1.12.dist-info/WHEEL sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ 87
trifast-0.1.12.dist-info/entry_points.txt sha256=zVm5_Sr2ym5p78XIxYT-jejHzqvv6pfUYk-ar6jTq50 51
trifast-0.1.12.dist-info/licenses/LICENSE sha256=afOWukWGfV3X55y5qQFt-cb8bzlrAH3NemoahUGKQ0A 1070
trifast-0.1.12.dist-info/RECORD

entry_points.txt

trifast-tune = trifast.tune:main