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