tf2jax

PyPI page
Home page
Author: DeepMind
License: Apache 2.0
Summary: TF2JAX: Convert TensorFlow to JAX
Latest version: 0.3.8
Required dependencies: absl-py | dm-tree | jax | jaxlib | numpy | tensorflow | tensorflow-macos

Downloads last day: 635
Downloads last week: 2,856
Downloads last month: 6,361