PyPI page
Home page
Author:
None
Summary:
Fast kernel for triangle self attetion.
Latest version:
0.1.13
Required dependencies:
einops
|
jaxtyping
|
numpy
|
platformdirs
|
pyyaml
|
setuptools
|
torch
|
triton
Optional dependencies:
deepspeed
|
pytest
Downloads last day:
64
Downloads last week:
558
Downloads last month:
2,472