Cannot import Trax about trax HOT 6 OPEN

Similar background, the same error

MacOS: 13.3.1

trax==1.2.4

mesh-tensorflow==0.1.21
tensor2tensor==1.15.7
tensorboard==2.13.0
tensorboard-data-server==0.7.0
tensorflow==2.13.0rc0
tensorflow-addons==0.20.0
tensorflow-datasets==4.9.2
tensorflow-estimator==2.13.0rc0
tensorflow-gan==2.1.0
tensorflow-hub==0.13.0
tensorflow-macos==2.13.0rc0
tensorflow-metadata==1.13.1
tensorflow-probability==0.7.0
tensorstore==0.1.37

jax==0.4.10
jaxlib==0.4.10

Python 3.8.10

"import trax" results in "ValueError: no signature found for builtin function <built-in function asarray>"

from trax.

I got this error when using cpu environment.
Using a gpu-tensorflow environment works for me.

from trax.

I face the same issue on Mac OS Ventura with M2 chip

from trax.

I am facing the exact same error. This error doesn't happen when I use Google collab, but it is unusably slow when training on large datasets. This forced me back to using Jupyter notebooks. Any idea?

from trax.

+1 on Python 3.9.6 on Mac OS 14.1.2 (23B92)

from trax.

+1 on python 3.11.5, win11, cpu only.

seemed that I soved the problem. trax defined a decorator(in trax\tf_numpy\numpy\utils.py) to attach numpy docstring to its own function. that's the place where this problem is generated. I just added a try-finally in the definition of the decorator to ignore the raised error.

from trax.