- 06 Sep, 2023 2 commits
- 05 Sep, 2023 4 commits
-
-
Tri Dao authored
-
Kyeongpil Kang authored
-
Tri Dao authored
-
Tri Dao authored
-
- 04 Sep, 2023 9 commits
- 03 Sep, 2023 4 commits
- 30 Aug, 2023 3 commits
-
-
dan_the_3rd authored
Co-authored-by: danthe3rd <danthe3rd>
-
dan_the_3rd authored
Co-authored-by: danthe3rd <danthe3rd>
-
GAOXinyu authored
when using checkpoint_lvl=2, we all_gather_raw(x) without async_op=True. So we don't need to wait for handle. Just skip.
-
- 29 Aug, 2023 2 commits
- 28 Aug, 2023 2 commits
- 27 Aug, 2023 1 commit
-
-
Tri Dao authored
-
- 26 Aug, 2023 3 commits
- 25 Aug, 2023 2 commits
- 24 Aug, 2023 1 commit
-
-
Aman Gupta Karmani authored
-
- 22 Aug, 2023 2 commits
- 21 Aug, 2023 1 commit
-
-
GAOXinyu authored
-
- 20 Aug, 2023 2 commits
-
-
Xuechen Li authored
* q * add comment.
-
Tri Dao authored
-
- 19 Aug, 2023 1 commit
-
-
Xuechen Li authored
* fix name. * set inv function. * add map back function. * handle gqa. * add type annotation to avoid confusion. * fix docstr. * test inverse remap logic.
-
- 18 Aug, 2023 1 commit
-
-
Tri Dao authored
-