在Jax中分析时，跟踪文件夹无效

Question

我正在尝试通过运行文档中提供的示例来测试Jax分析器：

import jax 

with jax.profiler.trace("/tmp/jax-trace", create_perfetto_link=True):
  # Run the operations to be profiled
  key = jax.random.PRNGKey(0)
  x = jax.random.normal(key, (5000, 5000))
  y = x @ x
  y.block_until_ready()

但是，我反复得到了ValueError: Invalid trace folder: /tmp/jax-trace/plugins/profile/{timestamp}。基于源(src/profiler.html)，这似乎意味着不存在跟踪转储，但我不确定为什么会出现这种情况。

Answer 1

在github中有一个关于这个问题的未决问题。TensorFlow分析器似乎不再转储trace.json文件。