sci-ml/flash-attn
Flash Attention: Fast and Memory-Efficient Exact Attention (Python component).
Screenshots
-
flash-attn-2.8.3_p1-r1~amd64cuda rocm python_single_target_python3_12 python_single_target_python3_13 python_single_target_python3_14 debug
View 
Download 
Browse License: BSD Overlay: tatsh-overlay
Runtime Dependencies
flash-attn-2.8.3_p1-r1
python_single_target_python3_12?
( dev-lang/python:3.12 )
python_single_target_python3_13?
( dev-lang/python:3.13 )
python_single_target_python3_14?
( dev-lang/python:3.14 )
python_single_target_python3_12?
( sci-ml/einops[python_targets_python3_12
(-)
] )
python_single_target_python3_13?
( sci-ml/einops[python_targets_python3_13
(-)
] )
python_single_target_python3_14?
( sci-ml/einops[python_targets_python3_14
(-)
] )
>=sci-ml/pytorch-2.13.0[python_single_target_python3_12
(-)
?
,python_single_target_python3_13
(-)
?
,python_single_target_python3_14
(-)
?
,cuda?
,rocm?
]
cuda?
( >=sci-ml/pytorch-2.13.0[flash] )
python_single_target_python3_12?
( dev-lang/python:3.12 )
python_single_target_python3_13?
( dev-lang/python:3.13 )
python_single_target_python3_14?
( dev-lang/python:3.14 )