Cannot import Trax
Author: Nuna7Created Apr 3, 2023Updated Sep 8, 2024
Description
I am trying to use trax following NLP specialization but couldn't import it. I have search for solution and ask help on ChatGPT but could not solve it. ...
Environment information
OS: <Mac Ventura 13.1>
$ pip freeze | grep trax
# trax==1.2.4
$ pip freeze | grep tensor
# mesh-tensorflow==0.1.21
tensor2tensor==1.15.7
tensorboard @ file:///usr/local/conda/envs/tf210/conda-bld/tensorboard_1670959059865/work/tensorboard-2.10.0-py3-none-any.whl
tensorboard-data-server @ file:///var/folders/nz/j6p8yfhx1mv_0grj5xl4650h0000gp/T/abs_91lm1y0ip6/croot/tensorboard-data-server_1670853587549/work/tensorboard_data_server-0.6.1-py3-none-macosx_11_0_arm64.whl
tensorboard-plugin-wit==1.6.0
tensorflow @ file:///usr/local/conda/envs/tf210/conda-bld/tensorflow-base_1671120030233/work/tensorflow_pkg/tensorflow-2.10.0-cp310-cp310-macosx_12_0_arm64.whl
tensorflow-addons==0.19.0
tensorflow-dataset==0.1.1
tensorflow-datasets==4.8.3
tensorflow-estimator @ file:///usr/local/conda/envs/tf210/conda-bld/tensorflow-estimator_1670959621277/work/tensorflow_estimator-2.10.0-py2.py3-none-any.whl
tensorflow-gan==2.1.0
tensorflow-hub==0.13.0
tensorflow-metadata==1.12.0
tensorflow-probability==0.7.0
$ pip freeze | grep jax
# jax==0.4.8
jaxlib==0.4.7
$ python -V
# Python 3.10.9For bugs: reproduction and error logs
# Steps to reproduce:
...import trax# Error logs:
...ValueError: no signature found for builtin function <built-in function asarray>Source: google/trax