diff --git a/.github/actions/build_cmake/action.yml b/.github/actions/build_cmake/action.yml
index 3fad247b6d..d55f5d5b24 100644
--- a/.github/actions/build_cmake/action.yml
+++ b/.github/actions/build_cmake/action.yml
@@ -16,6 +16,10 @@ inputs:
     description: 'Enable ROCm support.'
     required: false
     default: OFF
+  dnnl:
+    description: 'Enable DNNL support.'
+    required: false
+    default: OFF
 runs:
   using: composite
   steps:
@@ -62,6 +66,11 @@ runs:
           conda install -y -q libraft cuda-version=12.4 cuda-toolkit -c rapidsai-nightly -c "nvidia/label/cuda-12.4.0" -c conda-forge
         fi
 
+        # install DNNL package
+        if [ "${{ inputs.dnnl }}" = "ON" ]; then
+          conda install -y -q conda-forge::onednn
+        fi
+
         # install test packages
         conda install -y pytest
         if [ "${{ inputs.rocm }}" = "ON" ]; then
@@ -131,6 +140,7 @@ runs:
               -DFAISS_ENABLE_RAFT=${{ inputs.raft }} \
               -DFAISS_ENABLE_ROCM=${{ inputs.rocm }} \
               -DFAISS_OPT_LEVEL=${{ inputs.opt_level }} \
+              -DFAISS_ENABLE_DNNL=${{ inputs.dnnl }} \
               -DFAISS_ENABLE_C_API=ON \
               -DPYTHON_EXECUTABLE=$CONDA/bin/python \
               -DCMAKE_BUILD_TYPE=Release \
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index a17f0838b1..e478c655cd 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -67,6 +67,15 @@ jobs:
         uses: ./.github/actions/build_cmake
         with:
           opt_level: avx512
+  linux-x86_64-AMX-cmake:
+    name: Linux x86_64 AMX (cmake)
+    runs-on: faiss-aws-m7i.large
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+      - uses: ./.github/actions/build_cmake
+        with:
+          dnnl: ON
   linux-x86_64-GPU-cmake:
     name: Linux x86_64 GPU (cmake)
     needs: linux-x86_64-cmake