JAX:支持GPU/TPU的NumPy包(Autograd/XLA),面向高性能机器学习

JAX:支持GPU/TPU的NumPy包(Autograd/XLA),面向高性能机器学习