Skip to content

Instantly share code, notes, and snippets.

@saqibameen

saqibameen/jax-m1-install.sh

Last active March 13, 2024 03:49
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save saqibameen/e33612ca731407639dd31776d2f532ab to your computer and use it in GitHub Desktop.
Save saqibameen/e33612ca731407639dd31776d2f532ab to your computer and use it in GitHub Desktop.
Download ZIP
Installing Jax on Macbook — Apple Silicon (M1 Mac)
Raw
jax-m1-install.sh
# jax officially support m1 mac.
pip install jax jaxlib
# if running into issues.
conda create --name jax python=3.9
conda activate jax
pip install jax jaxlib
@saqibameen
Copy link
Author

saqibameen commented Oct 17, 2022 •
edited

Follow this issue for GPU support: google/jax#8074

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment