https://github.com/google/JAX JAX 论文链接:https://www.sysml.cc/doc/146.pdf JAX 的创新之处在于,它基于 XLA 在 GPU 和 TPU 上编译和运行 NumPy 程序. 解释:NumPy计算只能在CPU上计算,所以tensorflow需要定义自己tensor