Look like jax comes from conda-forge. You might uninstall jax first before using pip. Or you can install jax first, then using conda to install numpyro.
Look like jax comes from conda-forge. You might uninstall jax first before using pip. Or you can install jax first, then using conda to install numpyro.