flmath / jax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from jax-ml/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