Update GitHub workflows

Author: raphaelsty

.github/workflows/pypi.yaml

@@ -60,20 +60,12 @@ jobs:
           echo "CUDA_LIB_DIR=$LIB_DIR" >> $GITHUB_ENV
         shell: bash
 
-      - name: Build wheels
-        uses: pypa/cibuildwheel@v2.23.3
+      - name: Build wheel
         env:
-          CIBW_SKIP: "*-manylinux_i686 *-musllinux_* *-win32"
-          CIBW_BUILD_VERBOSITY: 1
-          LIBTORCH_BYPASS_VERSION_CHECK: 1
-          CIBW_MANYLINUX_X86_64_IMAGE: manylinux_2_28
-          CIBW_MANYLINUX_ARM64_IMAGE: manylinux_2_28"
-          CIBW_ENVIRONMENT: |
-            PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu/torch_stable.html"
-          CIBW_PIP_ARGS: --no-cache-dir
-          CIBW_BEFORE_BUILD: "pip install torch==2.7.0 numpy maturin delvewheel"
-          CIBW_REPAIR_WHEEL_COMMAND_LINUX: >
-            LD_LIBRARY_PATH=$(python -c 'import torch, os; print(os.path.join(os.path.dirname(torch.__file__), "lib"))'):$LD_LIBRARY_PATH auditwheel repair -w {dest_dir} {wheel} --exclude libtorch.so --exclude libtorch_cpu.so --exclude libtorch_python.so
+          LD_LIBRARY_PATH: ${{ env.CUDA_LIB_DIR }}:${LD_LIBRARY_PATH}
+          CUDA_COMPUTE_CAP: ${{ matrix.variant.compute-caps }}
+        run: maturin build --release --out dist
+        shell: bash
 
       - name: Upload wheel artifact
         uses: actions/upload-artifact@v4