Remove grok-1 (#975)

NVIDIA · Aug 14, 2024 · d7bad4b · d7bad4b
1 parent 3cd303d
commit d7bad4b
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 69 deletions.
diff --git a/.github/container/Dockerfile.grok b/.github/container/Dockerfile.grok
diff --git a/.github/container/manifest.yaml b/.github/container/manifest.yaml
@@ -153,11 +153,6 @@ equinox:
   tracking_ref: main
   latest_verified_commit: 1e601672d38d2c4d483535070a3572d8e8508a20
   mode: git-clone
-grok-1:
-  url: https://github.com/xai-org/grok-1.git
-  tracking_ref: main
-  latest_verified_commit: 7207216386e07206b2083c5c0be88db1add8e631
-  mode: git-clone
 big-vision:
   url: https://github.com/google-research/big_vision.git
   tracking_ref: main

diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
@@ -173,21 +173,6 @@ jobs:
       BASE_LIBRARY: pax
     secrets: inherit
 
-  build-grok:
-    needs: [build-jax]
-    uses: ./.github/workflows/_build.yaml
-    with:
-      ARCHITECTURE: ${{ inputs.ARCHITECTURE }}
-      ARTIFACT_NAME: "artifact-grok-build"
-      BADGE_FILENAME: "badge-grok-build"
-      BUILD_DATE: ${{ inputs.BUILD_DATE }}
-      BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }}
-      CONTAINER_NAME: grok
-      DOCKERFILE: .github/container/Dockerfile.grok
-      EXTRA_BUILD_ARGS: |
-        URLREF_GROK_1=${{ fromJson(inputs.SOURCE_URLREFS).GROK_1 }}
-    secrets: inherit
-
   build-gemma:
     needs: build-jax
     uses: ./.github/workflows/_build.yaml
@@ -223,7 +208,6 @@ jobs:
       - build-upstream-pax
       - build-rosetta-t5x
       - build-rosetta-pax
-      - build-grok
       - build-gemma
     outputs:
       TAGS: ${{ steps.collect-tags.outputs.TAGS }}
@@ -243,7 +227,6 @@ jobs:
             {"flavor": "upstream-pax", "stage": "final",   "priority": 900,  "tag": "${{ needs.build-upstream-pax.outputs.DOCKER_TAG_FINAL }}"},\
             {"flavor": "t5x",          "stage": "final",   "priority": 900,  "tag": "${{ needs.build-rosetta-t5x.outputs.DOCKER_TAG_FINAL }}"},\
             {"flavor": "pax",          "stage": "final",   "priority": 900,  "tag": "${{ needs.build-rosetta-pax.outputs.DOCKER_TAG_FINAL }}"},\
-            {"flavor": "grok",         "stage": "final",   "priority": 900,  "tag": "${{ needs.build-grok.outputs.DOCKER_TAG_FINAL }}"},\
             {"flavor": "gemma",        "stage": "final",   "priority": 900,  "tag": "${{ needs.build-gemma.outputs.DOCKER_TAG_FINAL }}"},\
             {"flavor": "jax",          "stage": "mealkit", "priority": 500,  "tag": "${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }}"},\
             {"flavor": "triton",       "stage": "mealkit", "priority": 500,  "tag": "${{ needs.build-triton.outputs.DOCKER_TAG_MEALKIT }}"},\
@@ -254,7 +237,6 @@ jobs:
             {"flavor": "upstream-pax", "stage": "mealkit", "priority": 500,  "tag": "${{ needs.build-upstream-pax.outputs.DOCKER_TAG_MEALKIT }}"},\
             {"flavor": "t5x",          "stage": "mealkit", "priority": 500,  "tag": "${{ needs.build-rosetta-t5x.outputs.DOCKER_TAG_MEALKIT }}"},\
             {"flavor": "pax",          "stage": "mealkit", "priority": 500,  "tag": "${{ needs.build-rosetta-pax.outputs.DOCKER_TAG_MEALKIT }}"},\
-            {"flavor": "grok",         "stage": "mealkit", "priority": 500,  "tag": "${{ needs.build-grok.outputs.DOCKER_TAG_MEALKIT }}"},\
             {"flavor": "gemma",        "stage": "mealkit", "priority": 500,  "tag": "${{ needs.build-gemma.outputs.DOCKER_TAG_MEALKIT }}"},\
 
             {}\

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -32,7 +32,7 @@ on:
         type: string
         description: |
           A comma-separated PACKAGE=URL#REF list to override sources used by build.
-          PACKAGE∊{JAX,XLA,Flax,transformer-engine,T5X,paxml,praxis,maxtext,levanter,haliax,grok,mujuco,mujuco-mpc,gemma,big-vision,common-loop-utils,flaxformer,panopticapi} (case-insensitive)
+          PACKAGE∊{JAX,XLA,Flax,transformer-engine,T5X,paxml,praxis,maxtext,levanter,haliax,mujuco,mujuco-mpc,gemma,big-vision,common-loop-utils,flaxformer,panopticapi} (case-insensitive)
         default: ''
         required: false
 
@@ -330,7 +330,6 @@ jobs:
             upstream-pax
             t5x
             pax
-            grok
             gemma
           )
           declare -a STAGES=(

diff --git a/README.md b/README.md
@@ -231,22 +231,6 @@
         </picture>
       </td>
     </tr>
-    <tr>
-      <td>
-        <picture>
-          <img style="height:1em;" src="https://img.shields.io/static/v1?label=&color=gray&logo=docker&message=Grok%3D%7Bcore%2CGrok-1%7D">
-        </picture>
-      </td>
-      <td>
-        <code>ghcr.io/nvidia/jax:grok</code>
-      </td>
-      <td>
-        <a href="https://gist.github.com/nvjax/913c2af68649fe568e9711c2dabb23ae/#file-final-grok-md"><img style="height:1em;" src="https://img.shields.io/endpoint?url=https%3A%2F%2Fgist.githubusercontent.com%2Fnvjax%2F913c2af68649fe568e9711c2dabb23ae%2Fraw%2Fbadge-grok-build-amd64.json&logo=docker&label=amd64"></a>
-        <a href="https://gist.github.com/nvjax/913c2af68649fe568e9711c2dabb23ae/#file-final-grok-md"><img style="height:1em;" src="https://img.shields.io/endpoint?url=https%3A%2F%2Fgist.githubusercontent.com%2Fnvjax%2F913c2af68649fe568e9711c2dabb23ae%2Fraw%2Fbadge-grok-build-arm64.json&logo=docker&label=arm64"></a>
-      </td>
-      <td>
-      </td>
-    </tr>
     <tr>
       <td>
         <picture>