- 12 May, 2021 1 commit
-
-
anj-s authored
* rename files * add newly renamed file * rename and move checkpoint activations related files * add test files to ci list * fix lint errors * modify docs * add changelog * retain old path for now * fix lint errors * add another import test case * fix merge conflict * add missing test file
-
- 15 Apr, 2021 1 commit
-
-
anj-s authored
* modify doc string * add offload docs * add tutorial * remove print * remove print statement * modify import * modify constants * modify README and add Offload symbol * fix lint * smaller mods * lint errors * Update README.md added the references at the bottom of the readme * address comments * doc changes * add blank line Co-authored-by:
Anjali Sridhar <anj@devfair0443.h2.fair> Co-authored-by:
Vittorio Caggiano <caggiano@gmail.com>
-
- 02 Apr, 2021 3 commits
-
-
anj-s authored
-
Anjali Sridhar authored
-
Anjali Sridhar authored
-
- 25 Mar, 2021 1 commit
-
-
Benjamin Lefaudeux authored
-
- 08 Mar, 2021 1 commit
-
-
Sam Shleifer authored
* Document FSDP tips and tricks in a separate file
-
- 23 Feb, 2021 1 commit
-
-
Min Xu authored
-
- 19 Feb, 2021 1 commit
-
-
Min Xu authored
* [docs]: add checkpoint_wrapper and many small fixes * update copyright year
-
- 28 Dec, 2020 1 commit
-
-
Benjamin Lefaudeux authored
-
- 21 Nov, 2020 1 commit
-
-
Benjamin Lefaudeux authored
* rewrite using autograd and Variable execution queue to make the reduce automatic * share buckets with OSS to remove duplication * some speed still likely on the table since the speed vs. bucketing does not match expectations, could be a follow up
-
- 18 Nov, 2020 1 commit
-
-
Benjamin Lefaudeux authored
* adding a shard-aware GradScaler wrap, credits to Sean Naren for the idea * adding stubs & explanations in the documentation
-
- 17 Nov, 2020 1 commit
-
-
Min Xu authored
- removed experimental warning as we have validated it on cifar and imagenet, transformer is looking good so far too. - fixed API doc formatting - make it consistent with the other code in the repo - tested by making the doc locally and inspect the results
-
- 22 Sep, 2020 1 commit
-
-
Benjamin Lefaudeux authored
* various fixes, no more issues with `make html` and more API fields should be populated
-
- 12 Sep, 2020 1 commit
-
-
msbaines authored
-
- 09 Sep, 2020 1 commit
-
-
msbaines authored
-