GPU- and TPU-backed NumPy with differentiation and JIT compilation. - View it on GitHub