Jax issues with osx-arm64 (M2 core)

tequilahub / tequila

A High-Level Abstraction Framework for Quantum Algorithms

MIT License

362 stars 101 forks source link

Jax issues with osx-arm64 (M2 core) #270

Closed kottmanj closed 1 year ago

kottmanj commented 1 year ago

RuntimeError: This version of jaxlib was built using AVX instructions, which your CPU and/or operating system do not support. You may be able work around this issue by building jaxlib from source.

current fix: uninstall jax and jaxlib and install autograd (almost the same functionality).

e.g.

pip install tequila-basic
pip uninstall jax
pip uninstall jaxlib
pip install autograd

kottmanj commented 1 year ago

Can't fix this on tequila side. Will at some point provide M2 wheels on PyPi.