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.

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

Rework Flax and Paxml training tutorials #5205

Merged
merged 12 commits into from
Dec 1, 2023

Commits on Nov 27, 2023

  1. Updae code for Flax example

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Nov 27, 2023
    Configuration menu
    Copy the full SHA
    f0ce406 View commit details
    Browse the repository at this point in the history
  2. Update code for paxml training example

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Nov 27, 2023
    Configuration menu
    Copy the full SHA
    6671fa4 View commit details
    Browse the repository at this point in the history

Commits on Nov 29, 2023

  1. Configuration menu
    Copy the full SHA
    8c22431 View commit details
    Browse the repository at this point in the history
  2. Update Paxml example

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Nov 29, 2023
    Configuration menu
    Copy the full SHA
    4197591 View commit details
    Browse the repository at this point in the history
  3. Update Flax example

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Nov 29, 2023
    Configuration menu
    Copy the full SHA
    aa2dc0d View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    6d914fc View commit details
    Browse the repository at this point in the history

Commits on Nov 30, 2023

  1. Configuration menu
    Copy the full SHA
    6c427bd View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    00ecd67 View commit details
    Browse the repository at this point in the history

Commits on Dec 1, 2023

  1. Fix review comments

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Dec 1, 2023
    Configuration menu
    Copy the full SHA
    63bac54 View commit details
    Browse the repository at this point in the history
  2. Fix review comments

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Dec 1, 2023
    Configuration menu
    Copy the full SHA
    4f47b94 View commit details
    Browse the repository at this point in the history
  3. Fix review comments

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Dec 1, 2023
    Configuration menu
    Copy the full SHA
    fff2e80 View commit details
    Browse the repository at this point in the history
  4. Fix review comments

    Signed-off-by: Albert Wolant <awolant@nvidia.com>
    awolant committed Dec 1, 2023
    Configuration menu
    Copy the full SHA
    ba42b76 View commit details
    Browse the repository at this point in the history