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
f68d41ec77
flash-attention
/
flash_attn
/
losses
History
Tri Dao
c6ecd40a59
Tweak CrossEntropyLoss to take process_group in init
2022-12-27 10:47:43 -08:00
..
__init__.py
Add __init__.py files to subdirectories for installation
2022-11-17 16:55:44 -08:00
cross_entropy.py
Tweak CrossEntropyLoss to take process_group in init
2022-12-27 10:47:43 -08:00