Split up docker deployment

.github/workflows/docker_deploy.yml

@@ -16,7 +16,13 @@ on:
 
 jobs:
   docker:
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest]
+        arch: [linux/amd64, linux/arm64]
+        python-version: [3.10.8]
+        julia-version: [1.8.2]
     steps:
       - name: Checkout
         uses: actions/checkout@v3
@@ -50,9 +56,12 @@ jobs:
         uses: docker/build-push-action@v3
         with:
           context: .
-          platforms: linux/amd64,linux/arm64
+          platforms: ${{ matrix.arch }}
           push: ${{ github.event_name != 'pull_request' }}
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
           cache-from: type=registry,ref=mcranmer/pysr:buildcache
           cache-to: type=registry,ref=mcranmer/pysr:buildcache,mode=max
+          build-args: |
+            PYVERSION=${{ matrix.python-version }}
+            JLVERSION=${{ matrix.julia-version }}