Merge branch 'alechan/add-te' of github.com:NVIDIA/JAX-Toolbox into alechan/add-te

Author: aybchan

.github/container/Dockerfile.triton

@@ -71,11 +71,6 @@ seqio:
   tracking_ref: main
   latest_verified_commit: 11706e4a1e01a81ea6b3e02c5ad147028d5b94bb
   mode: pip-vcs
-jax-triton:
-  url: https://github.com/jax-ml/jax-triton.git
-  tracking_ref: main
-  latest_verified_commit: 1999d9b116bf7c5c94f70de4a45b414255366fbe
-  mode: git-clone
 maxtext:
   url: https://github.com/google/maxtext.git
   tracking_ref: main

.github/container/manifest.yaml

@@ -178,10 +178,10 @@ jobs:
       max-parallel: 1
       matrix:
         PARALLEL_CONFIG:
-        - [1, 1, 1, 1]
-        - [1, 1, 8, 1]
-        - [1, 1, 1, 8]
-        - [1, 1, 4, 2]
+        # - [1, 1, 1, 1]
+        # - [1, 1, 8, 1]
+        # - [1, 1, 1, 8]
+        # - [1, 1, 4, 2]
         - [1, 2, 2, 2]
         # 2-node config temporarily disabled
         # - [1, 4, 2, 2]