Google Jax
16 Dec 2018, Prathyush SPJAX is NumPy on the CPU and GPU with autodiff and JIT compilation. It’s just been released, try it out!
For more details, visit the source.
JAX is NumPy on the CPU and GPU with autodiff and JIT compilation. It’s just been released, try it out!
For more details, visit the source.