我第一次加载这个库时被难住了。在我的本地计算机上，我在本地 Jupyter Notebook 上加载至少需要 40trax秒，在共享的 Colab 环境中加载它需要超过 1 分钟。
import trax
我不确定这是我的安装问题还是我正在使用的trax版本中的 BUG？
我是 trax 的新手，事实上我的经验是使用 Keras 和 TensorFlow，所以我想从 trax 社区的某个人那里得到意见，如果这是否正常。
提前非常感谢！
顺便说一句：我正在使用trax 1.4.1，Python 3.9.6并且我的本地计算机具有以下规格：
Intel(R) Core(TM) i5-6200U CPU @ 2.30GHz, 4 cores and 16GB RAM.
经过一段时间的搜索，这似乎是正常行为。事实上，官方 Trax 存储库中对此提出了一个问题：import trax 需要 17 秒 #1368。
显然fastmath，包含数学运算的 trax 重新实现的模块有很多依赖项。
从问题线程：
这是一个众所周知的问题，基本上在所有设置（本地、colab、gpu 集群）上都会发生，对于运行长时间的实验来说这不是一个大问题，但是它确实使本地调试变得困难。我曾尝试使用探查器调试导入图，但尚未成功。看起来即使 from trax import fastmath 也有很多依赖项 - 这是 importlab 库为 trax.fastmath 生成的树。初始化模块：importlab --tree init .py 输出： https ://gist.github.com/syzymon/3bb6f59063f918b4b62b77cdb223da72
所以总而言之，无论是像我一样需要 17 秒还是 40 秒，都是 Trax 中的已知行为。