Skip to content
This repository was archived by the owner on Oct 31, 2025. It is now read-only.
This repository was archived by the owner on Oct 31, 2025. It is now read-only.

Cannot import Trax #1776

Description

@Nuna7

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.9

For bugs: reproduction and error logs

# Steps to reproduce:
...import trax
# Error logs:
...ValueError: no signature found for builtin function <built-in function asarray>

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions