Commits · 1e712ea8b07850b0875c12afcfa4ddb67ed08c43 · gaoqiong / flash-attention

25 Dec, 2022 1 commit
- Implement TensorParallel for MHA · 1e712ea8
  Tri Dao authored Dec 24, 2022
  
  1e712ea8
23 Dec, 2022 1 commit
- Simplify FusedDense · e68ebbe8
  Tri Dao authored Dec 22, 2022
  
  e68ebbe8
21 Dec, 2022 1 commit
- Implement XPos (Sun et al.) · 496e4f52
  Tri Dao authored Dec 21, 2022
  
  496e4f52
17 Dec, 2022 1 commit
- add asserts for sin shape · ee8984d2
  Alexander Ploshkin authored Dec 17, 2022
  
  ee8984d2
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