note.wcoder.com
wcoder GitHub

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

← Previous Next →
Less
More