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
c7c66976cce3b1fa9281ddca02970bd383de9ebc
Switch branch/tag
flash-attention
flash_attn
layers
16 Dec, 2022
1 commit
fix slicing dimensions
· c7c66976
Alexander Ploshkin
authored
Dec 16, 2022
c7c66976
15 Dec, 2022
1 commit
Remove redundant shape asserts in rotary embeddings
· 96656b93
Alexander Ploshkin
authored
Dec 15, 2022
96656b93
18 Nov, 2022
2 commits
Add PatchEmbed
· 0fa5c0d7
Tri Dao
authored
Nov 17, 2022
0fa5c0d7
Add __init__.py files to subdirectories for installation
· ece539ab
Tri Dao
authored
Nov 17, 2022
ece539ab
17 Nov, 2022
1 commit
[Rotary] Customize base, support seqlen_offset
· 71f674ae
Tri Dao
authored
Nov 17, 2022
71f674ae
14 Nov, 2022
1 commit
Add MLP, MHA, Block, Embedding modules
· d4b320b3
Tri Dao
authored
Nov 13, 2022
d4b320b3