PyPI page
Home page
Author:
None
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
...
Summary:
Fast and easy-to-use implementation of flash attention.
Latest version:
0.1.0
Required dependencies:
chex
|
jax
|
jax-triton
|
triton
Optional dependencies:
click
|
pre-commit
|
pytest
Downloads last day:
1
Downloads last week:
7
Downloads last month:
13