autokeras icon indicating copy to clipboard operation
autokeras copied to clipboard

Published 20 hours ago • verified publisher icon keras-team

Would it be possible to add jax?

Open Anselmoo opened this issue 3 years ago • 0 comments

In according to Google's JAX, would it be possible to include JAX into AutoKeras, please?

I guess, if no array-slicing would be performed, JAX should be fully working?

https://github.com/keras-team/autokeras/blob/c51da2dd87b195ab3bd0941ea70862c3cf66e9a9/autokeras/adapters/input_adapters.py#L25-L29

Anselmoo avatar May 02 '22 20:05 Anselmoo