SSealbook

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

jax

License & maintenance

Apache-2.0 is a permissive license: free to use commercially with attribution.

Last code push today.