'Is there a module to convert a tensorflow NN to Jax?
There is a libary to convert Jax functions to Tensorflow functions. Is there a similar library to convert TensorFlow functions to Jax functions?
Solution 1:[1]
To my knowledge there is no library similar to the one you mentioned to convert TensorFlow functions to Jax functions. I'm sorry
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
| Solution | Source |
|---|---|
| Solution 1 | Erling Olsen |