Merge pull request #838 from ko3n1g/ko3n1g/ci/build-arm-wheels

ci: Build arm wheels for release

Author: ko3n1g

.github/workflows/publish.yaml

@@ -42,7 +42,7 @@ jobs:
       matrix:
         # Using ubuntu-22.04 instead of 24.04 for more compatibility (glibc). Ideally we'd use the
         # manylinux docker image, but I haven't figured out how to install CUDA on manylinux.
-        os: [ubuntu-22.04]
+        os: [ubuntu-22.04, ubuntu-22.04-arm]
         python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
         torch-version: ["2.4.0", "2.5.1", "2.6.0", "2.7.1"]
         cuda-version: ["11.8.0", "12.9.1"]