Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Flax] Add remat (gradient checkpointing) #17843

Merged
merged 11 commits into from
Jul 1, 2022
Merged

[Flax] Add remat (gradient checkpointing) #17843

merged 11 commits into from
Jul 1, 2022
+414 −96

Commits on Jun 30, 2022

  1. [Flax] Add remat (gradient checkpointing)

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    7dd4c58 View commit details
    Browse the repository at this point in the history

  2. fix variable naming in test

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    03606c1 View commit details
    Browse the repository at this point in the history

  3. flip: checkpoint using a method

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    9b6e164 View commit details
    Browse the repository at this point in the history

  4. fix naming

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    395221f View commit details
    Browse the repository at this point in the history

  5. fix class naming

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    7b165d5 View commit details
    Browse the repository at this point in the history

  6. apply PVP's suggestions from code review

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    d6040e0 View commit details
    Browse the repository at this point in the history

  7. make fix-copies

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    9972f38 View commit details
    Browse the repository at this point in the history

  8. fix big-bird, electra, roberta

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    0a0b6bf View commit details
    Browse the repository at this point in the history

  9. cookie-cutter

    @sanchit-gandhi
    sanchit-gandhi committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    70b7175 View commit details
    Browse the repository at this point in the history

Commits on Jul 1, 2022

  1. fix flax big-bird

    @sanchit-gandhi
    sanchit-gandhi committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    0665a33 View commit details
    Browse the repository at this point in the history

  2. move test to common

    @sanchit-gandhi
    sanchit-gandhi committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    66c4b14 View commit details
    Browse the repository at this point in the history