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

user_14720380 0
Viewed 354
jax tensorflow python
2 Answers

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.

user_29378310

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

user_171383220
Related