TF2JAX

https://github.com/google-deepmind/tf2jax

An experimental library for converting TensorFlow functions and graphs to JAX functions.

Tokens:3,204
Snippets:46
Trust Score:9.4
License:Apache-2.0
Update:1 month ago
Tokens:
Raw