如果您阅读jax 源代码,您会遇到名为xla_client. 经常这样导入
xla_client
from . import xla_client
这意味着这xla_client是一个 python 模块,但我找不到任何具有该名称的文件或对该名称的变量的引用。
我假设它与https://pypi.org/project/jaxlib/相关,但这个包只是链接回 jax 源代码。
有人可以告诉我吗?
tensorflow jax
jax ×1
tensorflow ×1