Página Principal Explorar Ajuda
Registe-se Iniciar Sessão
david
/
flash-attention
mirror de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Ficheiros Problemas 0 Wiki
Árvore: 6807b1ea37
Ramos Etiquetas
flash-attention
/
tests
/
losses
Tri Dao c7f32a8409 [CrossEntropy] Support precomputed LSE há 3 meses atrás
..
test_cross_entropy.py c7f32a8409 [CrossEntropy] Support precomputed LSE há 3 meses atrás
test_cross_entropy_parallel.py c7f32a8409 [CrossEntropy] Support precomputed LSE há 3 meses atrás