"No module Named jax" with jax installed

[CaptClockobob]

I'm trying to start the server on WSL Ubuntu 20.04, and I get this error:

--> Starting DALL-E Server. This might take up to two minutes. Traceback (most recent call last): File "app.py", line 18, in <module> from dalle_model import DalleModel File "/mnt/c/Users/mvaml/dallePlayground/dalle-playground/backend/dalle_model.py", line 5, in <module> import jax ModuleNotFoundError: No module named 'jax'

Jax is installed.

Please help.