From e75eacf6329c165fc4f9235a4d69530a61c6488d Mon Sep 17 00:00:00 2001 From: Terry Kong Date: Fri, 8 Sep 2023 15:22:07 -0700 Subject: [PATCH 1/2] Update patchlist-t5x.txt to PRs created from nvjax-svc-0 --- rosetta/patchlist-t5x.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/rosetta/patchlist-t5x.txt b/rosetta/patchlist-t5x.txt index 91c11e278..81c59e6b4 100644 --- a/rosetta/patchlist-t5x.txt +++ b/rosetta/patchlist-t5x.txt @@ -5,6 +5,6 @@ # - External Pull Requests (These are pull requests with upstream t5x and are of the form "pull/$PULLID/head") # - Note: Only the first column is used as a git-ref, so anything after is a comment -pull/1357/head -pull/1358/head -pull/1385/head # https://github.com/google-research/t5x/pull/1385: Adds transformer engine support and GPU optimizations to T5x (enables H100) +pull/1392/head # https://github.com/google-research/t5x/pull/1392: Add support for partial checkpoint restore +pull/1393/head # https://github.com/google-research/t5x/pull/1393: Adds DALI support to t5x +pull/1391/head # https://github.com/google-research/t5x/pull/1385: Adds transformer engine support and GPU optimizations to T5x (enables H100) From b9bb057c294fe47a9cce30a25403094a8e63ea80 Mon Sep 17 00:00:00 2001 From: Terry Kong Date: Thu, 14 Sep 2023 21:06:27 -0700 Subject: [PATCH 2/2] Updates flax patches --- rosetta/patchlist-flax.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rosetta/patchlist-flax.txt b/rosetta/patchlist-flax.txt index 3de35cfeb..335d9f0f3 100644 --- a/rosetta/patchlist-flax.txt +++ b/rosetta/patchlist-flax.txt @@ -5,4 +5,4 @@ # - External Pull Requests (These are pull requests with upstream flax and are of the form "pull/$PULLID/head") # - Note: Only the first column is used as a git-ref, so anything after is a comment -pull/3268/head +pull/3340/head # Add Sharding Annotations to Flax Modules