是否有将 tensorflow NN 转换为 Jax 的模块?

Is there a module to convert a tensorflow NN to Jax?

有一个libary to convert Jax functions to Tensorflow functions。是否有类似的库将 TensorFlow 函数转换为 Jax 函数?

不,JAX 团队不支持以类似于 jax.experimental.jax2tf 将 JAX 代码转换为 tensorflow 的方式将 tensorflow 转换为 JAX 的库,我也没有看到其他人开发的任何此类库。

据我所知,没有与您提到的将 TensorFlow 函数转换为 Jax 函数的库类似的库。对不起