We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
A few months ago, I noticed that JAX (https://github.com/google/jax) and Trax (https://github.com/google/trax) have been getting more popular.
JAX functions which are compiled (https://github.com/google/jax#compilation-with-jit) can be turned into an XLA HLO proto (see google/jax#1871) which can be run from C++
Trax can use TF, Numpy, or JAX under the hood so I don't think we need to do much additional work to add support for it.
Concretely, we'd need to add a backend for XLA and packagers for JAX and Trax
The text was updated successfully, but these errors were encountered:
Note: Flax (https://github.com/google/flax) is another DL library built on top of JAX
Sorry, something went wrong.
Is it right understanding that this is GPU/TPU optimization only?
No branches or pull requests
A few months ago, I noticed that JAX (https://github.com/google/jax) and Trax (https://github.com/google/trax) have been getting more popular.
JAX functions which are compiled (https://github.com/google/jax#compilation-with-jit) can be turned into an XLA HLO proto (see google/jax#1871) which can be run from C++
Trax can use TF, Numpy, or JAX under the hood so I don't think we need to do much additional work to add support for it.
Concretely, we'd need to add a backend for XLA and packagers for JAX and Trax
The text was updated successfully, but these errors were encountered: