Is Jax supported in Replit?

I have a homework project that requires me to use Jax Array instead of Numpy Array. But I was unable to install Jaxlib or Jax Package into my IDE.

Does Replit support the Jax Package? If yes, how do I install it?


Yes, run this command in the shell:

poetry add jax

If this does not work, please post any errors you receive as text.