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

github.com/borisdayma/jax

Problem

Audience

Updates

No founder updates yet. Check back soon.