Help with installing Python JAX

Hi, I’ve been having problem with installing JAX to be used with python. I use Anaconda, which (along with other languages and IDEs) contains python and spyder.

If anyone uses Python JAX, plz help. Thanks.

How did you try to install it? What happened when you tried that?

1 Like

Thanks for the reply. I was installing it in Windows (laptop) using pip install. But later (in the lab) I tried downloading JAX (and equinox and diffrax) in Ubuntu. It reported that Ubuntu’s in built python is 2.7 which is incompatible with the JAX’s requirement. It advised something like using an alternate environment.