/tensorflow-jax-experimental

TensorFlow XLA backend of experimental JAX on Mk2

Primary LanguageC++Apache License 2.0Apache-2.0

Stargazers