I’ve installed jax using pip3 install jax, and have also tried pip install -U "jax[cpu]". When I attempt to import jax in a Jupyter notebook, as follows:
from jax import numpy as np
import jax