谷歌JAX深度学习从零开始学
上QQ阅读APP看书,第一时间看更新

1.1 JAX来了

“工欲善其事,必先利其器”。人工智能或者其核心理论深度学习也一样。任何一个好的成果落地并在将来发挥其巨大作用,都需要一个能够将其落地并应用的基本框架工具。JAX是机器学习框架领域的新生力量,它具有更快的高阶微分计算方法,可以采用先编译后执行的模式,突破了已有深度学习框架的局限性,同时具有更好的硬件支持,甚至将来可能会成为谷歌的主要科学计算深度学习库。