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
d0032700d1c7a1353a3a8f2fadfbc73b2dc6b5dc
Switch branch/tag
flash-attention
tests
models
test_gpt_neox.py
13 Sep, 2023
1 commit
Add tests for Pythia, GPT-JT, and RedPajama models
· d0032700
Tri Dao
authored
Sep 13, 2023
d0032700
11 Sep, 2023
1 commit
[Gen] Fix calling update_graph_cache in tests
· 8a733cbd
Tri Dao
authored
Sep 10, 2023
8a733cbd
19 Aug, 2023
1 commit
Run isort and black on test files
· 0e8c46ae
Tri Dao
authored
Aug 18, 2023
0e8c46ae
19 Apr, 2023
1 commit
Implement LLaMa
· 96d10f65
Tri Dao
authored
Apr 18, 2023
96d10f65
31 Mar, 2023
1 commit
[LayerNorm] Implement LN with parallel residual, support dim 8k
· 393882bc
Tri Dao
authored
Mar 29, 2023
393882bc
29 Mar, 2023
1 commit
Implement GPT-NeoX
· 993d1244
Tri Dao
authored
Mar 29, 2023
993d1244