Add multi-arch Docker image for x86-64 and arm64 (#1431)

<!-- .github/pull_request_template.md -->

## 📌 Description

<!-- What does this PR do? Briefly describe the changes and why they’re
needed. -->

## 🔍 Related Issues

<!-- Link any related issues here -->

## 🚀 Pull Request Checklist

Thank you for contributing to FlashInfer! Before we review your pull
request, please make sure the following items are complete.

### ✅ Pre-commit Checks

- [ ] I have installed `pre-commit` by running `pip install pre-commit`
(or used your preferred method).
- [ ] I have installed the hooks with `pre-commit install`.
- [ ] I have run the hooks manually with `pre-commit run --all-files`
and fixed any reported issues.

> If you are unsure about how to set up `pre-commit`, see [the
pre-commit documentation](https://pre-commit.com/).

## 🧪 Tests

- [ ] Tests have been added or updated as needed.
- [ ] All tests are passing (`unittest`, etc.).

## Reviewer Notes

<!-- Optional: anything you'd like reviewers to focus on, concerns, etc.
-->

Author: yongwww

.github/workflows/release-ci-docker.yml

@@ -5,25 +5,31 @@ on:
 
 jobs:
   build:
-    strategy:
-      matrix:
-        include:
-          - os: ubuntu-latest
-            arch: amd64
-          - os: [self-hosted, Linux, ARM64]
-            arch: arm64
-    runs-on: ${{ matrix.os }}
+    runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
       - name: Login to Docker Hub
         uses: docker/login-action@v3
         with:
           username: flashinfer
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - uses: docker/build-push-action@v4
+      - name: Build and push multi-arch image
+        uses: docker/build-push-action@v5
         with:
           context: docker
           file: docker/Dockerfile.ci_gpu
+          platforms: linux/amd64,linux/arm64
           push: true
-          tags: flashinfer/flashinfer-ci:latest
+          tags: |
+            flashinfer/flashinfer-ci:latest
+            flashinfer/flashinfer-ci:${{ github.sha }}
+          cache-from: type=registry,ref=flashinfer/flashinfer-ci:buildcache
+          cache-to: type=registry,ref=flashinfer/flashinfer-ci:buildcache,mode=max