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
054816177e5bfe8f1e9883d9e03fe9bb8473410c
Switch branch/tag
flash-attention
flash_attn
losses
cross_entropy_parallel.py
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
