No module named 'trax'

Question

Guys, Help we with this, I installed trax with pip install trax, but it showing error, after restarting the kernel also Getting this error.

Answer 1

Try running followings

pip install --upgrade pip

pip install --upgrade "jax[cpu]"

pip install trax