JAX：Autograd和XLA | | | | | 新闻： 什么是JAX？ JAX是和结合，用于高性能机器学习研究。 通过其更新版本的 ，JAX可以自动区分本机Python和NumPy函数。 它可以通过循环，分支，递归和闭包来区分，并且可以采用派生类的派生类。 它支持通过反向模式区分（aka反向传播）以及正向模式区分，并且两者可以任意组合为任意顺序。 新功能是JAX使用在GPU和TPU上编译和运行您的NumPy程序。 默认情况下，编译是在后台进行的，而库调用会及时进行编译和执行。