Revert "Revert "update patches""

This reverts commit d2a94f4.

Revert "Revert "switch back to original TE patch and update manifest""

This reverts commit 090dfeb.

Author: ashors1

.github/container/manifest.yaml

@@ -2,31 +2,31 @@
 jax:
   url: https://github.com/google/jax.git
   tracking_ref: main
-  latest_verified_commit: 2356e57af47baa37286177676a5a4bcd0c7b8c40
+  latest_verified_commit: 3a0b495faa212b9b415b30f71f9283a81be24261
   mode: git-clone
 xla:
   url: https://github.com/openxla/xla.git
   tracking_ref: main
-  latest_verified_commit: 17ec7b4031608d3ddaca83e0661ed3aa0393e9d9
+  latest_verified_commit: 3bcf05e59b1c46b617c8510152bb407f1e1a49a7
   mode: git-clone
 flax:
   url: https://github.com/google/flax.git
   mirror_url: https://github.com/nvjax-svc-0/flax.git
   tracking_ref: main
-  latest_verified_commit: 7cbc3c13ecd8a1713732113dec7939d0b75342c6
+  latest_verified_commit: 93d3eb788fc130a3446681458fda0e00954fdbcd
   mode: git-clone
   patches:
     pull/3340/head: file://patches/flax/PR-3340.patch # Add Sharding Annotations to Flax Modules
 transformer-engine:
   url: https://github.com/NVIDIA/TransformerEngine.git
   tracking_ref: main
-  latest_verified_commit: bb759adc2a53c57cc387f531c580a16149e3d4ac
+  latest_verified_commit: bacefdbb6815159c42d5ca501a1400697b98a1e3
   mode: git-clone
 t5x:
   url: https://github.com/google-research/t5x.git
   mirror_url: https://github.com/nvjax-svc-0/t5x.git
   tracking_ref: main
-  latest_verified_commit: 18f74b7891223faf96825d02ef7d7b23630a4c3c
+  latest_verified_commit: ed43961aebb179870b829e59cdd39a707f796656
   mode: git-clone
   patches:
     mirror/patch/partial-checkpoint-restore: file://patches/t5x/mirror-patch-partial-checkpoint-restore.patch # pull/1392/head  # https://github.com/google-research/t5x/pull/1392: Add support for partial checkpoint restore
@@ -36,15 +36,15 @@ paxml:
   url: https://github.com/google/paxml.git
   mirror_url: https://github.com/nvjax-svc-0/paxml.git
   tracking_ref: main
-  latest_verified_commit: e666687c20ed64383485853cc7691228b4f4d302
+  latest_verified_commit: 19db52eed85ae0d2365339b83a97cd0b873bbf73
   mode: git-clone
   patches:
     pull/46/head: file://patches/paxml/PR-46.patch # adds Transformer Engine support
 praxis:
   url: https://github.com/google/praxis.git
   mirror_url: https://github.com/nvjax-svc-0/praxis.git
   tracking_ref: main
-  latest_verified_commit: d126ac2cf4bc4808280d2842bbd3af872c6f682b
+  latest_verified_commit: 43e9e2ce1179efdc064277b04e16eedcd47e9558
   mode: git-clone
   patches:
     pull/27/head: file://patches/praxis/PR-27.patch # This PR allows XLA:GPU to detect the MHA pattern more easily to call fused kernels from cublas.
@@ -53,7 +53,7 @@ lingvo:
   # Used only in ARM pax builds
   url: https://github.com/tensorflow/lingvo.git
   tracking_ref: master
-  latest_verified_commit: 17d1b70d67a3ba05773750adecf7ac3296e619d5
+  latest_verified_commit: c33b460ae9208323b664c5bd565899dcd9c2e42c
   mode: git-clone
 tensorflow-text:
   # Used only in ARM pax builds
@@ -68,13 +68,13 @@ pydantic:
 fiddle:
   url: https://github.com/google/fiddle.git
   tracking_ref: main
-  latest_verified_commit: 26b9a44620038f906af8fc4947763d9abba0bfd2
+  latest_verified_commit: dbe479bf6ccb2b6edaa6fe0843d5d0d9182f8d57
   mode: pip-vcs
 # Used by t5x
 airio:
   url: https://github.com/google/airio.git
   tracking_ref: main
-  latest_verified_commit: e472ba33fee92f412663cf2a9bef55161a87e3ce
+  latest_verified_commit: 823bc5be0b6620c40b4d7558281d4b2357bae480
   mode: pip-vcs
 clu:
   url: https://github.com/google/CommonLoopUtils.git
@@ -94,26 +94,26 @@ jestimator:
 optax:
   url: https://github.com/google-deepmind/optax.git
   tracking_ref: main
-  latest_verified_commit: bc22961422eb2397a4639ec945da0bea73d624d6
+  latest_verified_commit: 1decd3e4918f2b4529265cb444689a57f7be59d3
   mode: pip-vcs
 seqio:
   url: https://github.com/google/seqio.git
   tracking_ref: main
-  latest_verified_commit: 36d3ef0d3f027d40d0600e11a7e17b0c88c20cff
+  latest_verified_commit: 1d3e909d464002baa3df9a6de65559999870df91
   mode: pip-vcs
 # used by Pallas
 openxla-triton:
   url: https://github.com/openxla/triton.git
   tracking_ref: llvm-head
-  latest_verified_commit: 1a9969fadc0a4cf694e751d24f3e40802602c3b1
+  latest_verified_commit: 09ddf3b32b7ac777c5f57c3c8e9ba18d8adf7ccb
   mode: git-clone
 jax-triton:
   url: https://github.com/jax-ml/jax-triton.git
   tracking_ref: main
-  latest_verified_commit: 4a5791d00809ac96f92d42a80a4f52501d1421be
+  latest_verified_commit: 4af0ecb709f46633de59342a16e15454f5fcdcfa
   mode: git-clone
 maxtext:
   url: https://github.com/google/maxtext.git
-  tracking_ref: main 
-  latest_verified_commit: 123ec5ad07b85a501ccdcdac6860c19e69f422a2
+  tracking_ref: main
+  latest_verified_commit: ec9cdce1928fb36f93e498c8f74f8f130bad3dc2
   mode: git-clone