yoziru / jax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
License: Apache License 2.0
This project forked from google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
License: Apache License 2.0
first noted in nextstrain/docker-base#173 (comment)
This happens upon importing jax
:
Traceback (most recent call last):
File "<string>", line 1, in <module>
File "/usr/local/lib/python3.10/site-packages/jax/__init__.py", line 35, in <module>
from jax import config as _config_module
File "/usr/local/lib/python3.10/site-packages/jax/config.py", line 17, in <module>
from jax._src.config import config # noqa: F401
File "/usr/local/lib/python3.10/site-packages/jax/_src/config.py", line 24, in <module>
from jax._src import lib
File "/usr/local/lib/python3.10/site-packages/jax/_src/lib/__init__.py", line 92, in <module>
import jaxlib.xla_client as xla_client
File "/usr/local/lib/python3.10/site-packages/jaxlib/xla_client.py", line 26, in <module>
from . import xla_extension as _xla
ImportError: /usr/local/lib/python3.10/site-packages/jaxlib/xla_extension.so: undefined symbol: PyObject_Vectorcall
This looks like an issue with the binary provided by this fork repo: jaxlib-0.4.13-cp310-cp310-manylinux2014_aarch64.whl. I confirmed that the official 0.4.13 works locally:
pip install jaxlib==0.4.13 jax==0.4.13
python -c "import jax" # this works
and the custom built 0.4.13 does not:
FROM python:3.10-slim-bullseye
RUN pip install https://github.com/yoziru/jax/releases/download/jaxlib-v0.4.13/jaxlib-0.4.13-cp310-cp310-manylinux2014_aarch64.whl jax==0.4.13
RUN python -c "import jax" # this fails
No response
No response
No response
No response
First off, thanks for providing these unofficial wheels. We are using them in nextstrain/docker-base. Without these pre-built wheels, we'd probably have to resort to using emulated JAX in our linux/arm64 image variant.
It's not clear to me how you are making the releases. Are you running Docker builds locally on an Apple silicon machine and uploading files to GitHub Releases?
I am trying to do this myself using GitHub Actions runners. I first started off with your Dockerfile snippet, but it doesn't work on the AMD64 GitHub Actions runner. I think cross-compilation is the way to go, but haven't got that working yet.
A declarative, efficient, and flexible JavaScript library for building user interfaces.
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. ๐๐๐
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google โค๏ธ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.