Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
flash-attention
Commits
0bf5e50038ee341ece03bfd0c8ff45a6c57aed5a
Switch branch/tag
flash-attention
training
src
losses
cross_entropy_parallel.py
29 Nov, 2022
1 commit
Release training code
· 0bf5e500
Tri Dao
authored
Nov 28, 2022
0bf5e500
21 Nov, 2022
1 commit
Bump version to 0.2.1
· 05481617
Tri Dao
authored
Nov 20, 2022
05481617
14 Nov, 2022
1 commit
Make nccl operations async in CrossEntropyLossParallel
· 343492ec
Tri Dao
authored
Nov 13, 2022
343492ec
13 Nov, 2022
1 commit
Add fused cross entropy loss
· 7c995381
Tri Dao
authored
Nov 12, 2022
7c995381