Inicio Explorar Ayuda
Registro Iniciar sesión
david
/
flash-attention
espejo de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Archivos Incidencias 0 Wiki
Rama: main
Ramas Etiquetas
flash-attention
/
flash_attn
/
losses
Tri Dao c7f32a8409 [CrossEntropy] Support precomputed LSE hace 3 meses
..
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation hace 2 años
cross_entropy.py c7f32a8409 [CrossEntropy] Support precomputed LSE hace 3 meses