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

There is a libary to convert Jax functions to Tensorflow functions. Is there a similar library to convert TensorFlow functions to Jax functions?

CodePudding user response：

No, there is no library supported by the JAX team to convert tensorflow into JAX in a manner similar to how jax.experimental.jax2tf converts JAX code to tensorflow, and I have not seen any such library developed by others.

CodePudding user response：

To my knowledge there is no library similar to the one you mentioned to convert TensorFlow functions to Jax functions. I'm sorry