模块 trax.fastmath 是否已弃用？

问题描述

TRAX 包的模块 trax.fastmath 是否已弃用？我正在使用此模块，但返回 ModuleNotFoundError。

解决方法

我用 trax.math 替换了 trax.fastmath 并且一切正常。我仍然不确定这是否已在 Trax 文档中更新。