conda create -n getting-started-with-jax python=3.10 ipykernel jupyter conda activate getting-started-with-jax pip install jax # Optional (Apple Silicon) pip install jax-metal