This website requires JavaScript.
Explore
Help
Register
Sign In
squall
/
flash-attention
Watch
1
Star
0
Fork
0
You've already forked flash-attention
Code
Issues
Pull Requests
Actions
Packages
Projects
Releases
Wiki
Activity
b0eac3297f
flash-attention
/
flash_attn
/
losses
History
Tri Dao
343492ec30
Make nccl operations async in CrossEntropyLossParallel
2022-11-13 17:27:26 -08:00
..
cross_entropy_apex.py
Add fused cross entropy loss
2022-11-12 21:58:41 -08:00
cross_entropy_parallel.py
Make nccl operations async in CrossEntropyLossParallel
2022-11-13 17:27:26 -08:00