diff --git a/.github/workflows/build_linux_wheels.yaml b/.github/workflows/build_linux_wheels.yaml
index a0e9b505f0..f2104df92b 100644
--- a/.github/workflows/build_linux_wheels.yaml
+++ b/.github/workflows/build_linux_wheels.yaml
@@ -28,7 +28,7 @@ jobs:
       with-cuda: enable
       with-rocm: enable
       build-python-only: enable
-  test:
+  filter-python-versions:
     needs: generate-matrix
     runs-on: ubuntu-latest
     steps:
@@ -52,23 +52,20 @@ jobs:
 
           with open(github_output_file, "w") as handle:
             handle.write(f"matrix={json.dumps(new_dict)}")
-  # build:
-  #   if: ${{ needs.generate-matrix.outputs.python_version != '3.9' }}
-  #   needs: generate-matrix
-  #   name: ${{ matrix.repository }}
-  #   uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
-  #   strategy:
-  #     fail-fast: false
-  #   with:
-  #     repository: pytorch/torchtune
-  #     ref: ""
-  #     package-name: torchtune
-  #     build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-  #     pre-script: .github/scripts/pre_build_script.sh
-  #     trigger-event: ${{ github.event_name }}
-  #     build-platform: 'python-build-package'
-  #     pip-install-torch-extra-args:
-  #       torchvision
-  #       torchao
-      # - name: output-python-version
-      #   run: echo PYTHON_VERSION=$env.PYTHON_VERSION >> "$GITHUB_OUTPUT"
+  build:
+    needs: filter-python-versions
+    name: ${{ matrix.repository }}
+    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
+    strategy:
+      fail-fast: false
+    with:
+      repository: pytorch/torchtune
+      ref: ""
+      package-name: torchtune
+      build-matrix: ${{ needs.filter-python-versions.outputs.matrix }}
+      pre-script: .github/scripts/pre_build_script.sh
+      trigger-event: ${{ github.event_name }}
+      build-platform: 'python-build-package'
+      pip-install-torch-extra-args:
+        torchvision
+        torchao