drh0111 / jax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Home Page: http://jax.readthedocs.io/
License: Apache License 2.0