新闻首页 / 标签为“JAX”的新闻

0

谷歌开源TensorFlow的简化库JAX

谷歌开源了一个 TensorFlow 的简化库 JAX。 JAX 结合了 Autograd 和 XLA,专门用于高性能机器学习研究。 凭借 Autograd,JAX 可以求导循环、分支、递归和闭包函数,并且它可以进行三阶求导。通过 grad,它支持自动模式反向求导(反向传播)和正向求导,且二者可以任

最新评论