3 ms·Jax is an implementation of numpy on GPU.by ma2rten 6y agoJax is an implementation of numpy on GPU.