Является ли модуль trax.fastmath устаревшим?

Question

Устарел ли модуль trax.fastmath пакета TRAX? Я использую этот модуль, но возвращается ошибка ModuleNotFoundError.

Answer 1

Я заменил trax.fastmath на trax.math, и все заработало отлично. Я до сих пор не уверен, было ли это обновлено в документации Trax.