jax

Python ★ 1 updated 5y ago ⑂ fork

GPU- and TPU-backed NumPy with differentiation and JIT compilation.

No plain-English explanation yet — one is being written right now. Check back in a minute.