from nextstrain/docker-base#119 (comment)
This is an unfinished attempt to compile an AARCH64 version of JAX on a GitHub-hosted Linux runner (AMD64).
I briefly tried building via Docker under emulation, which was unsuccessful. Even if it were successful, it would probably take longer than the maximum allowed GitHub Actions workflow run time.
Then, I focused on cross-compiling since that has been done before. I attempted this first with then without Docker (for easier debugging of individual commands as GitHub Actions steps). With a 3.5 hour run time, I was able to build a file jaxlib-0.4.9-cp310-cp310-manylinux2014_aarch64.whl
(a downloadable artifact of the linked run). However, I'm not sure that this wheel works because auditwheel repair
failed on the file. I think the failure makes sense because I didn't use a manylinux Docker image. To address that, I tried building with messense/manylinux2014-cross:x86_64
but had troubles running the setup commands before the actual JAX build.