runtime error

Traceback (most recent call last): File "/home/user/app/app.py", line 9, in <module> import jax.numpy as jnp ModuleNotFoundError: No module named 'jax'

Container logs: