jax2tf

jax2tf is a conversion tool in the JAX project that translates JAX functions into TensorFlow graphs or SavedModel-compatible functions. It supports interoperability between JAX and TensorFlow, including exporting computations for TensorFlow-based serving and deployment.