jax-ml/jax — Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and m…

jax-ml/jax: jax-ml/jax may be useful when its README and activity match a concrete workflow.. Use it for Misc. Medium: useful for prototypes or internal workflo