Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.
/ build-jax Public archive
forked from jax-ml/jax

An unfinished attempt at building JAX from source using GitHub Actions runners.

Notifications You must be signed in to change notification settings

victorlin/build-jax

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 

Repository files navigation

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.

About

An unfinished attempt at building JAX from source using GitHub Actions runners.

Resources

Stars

Watchers

Forks

Languages

  • Starlark 100.0%