JAX TensorFlow 简化库

授权协议：Apache-2.0操作系统：跨平台开发语言：Python 查看源码 »

JAX 是一个 TensorFlow 的简化库，它结合了 Autograd 和 XLA，专门用于高性能机器学习研究。凭借 Autograd，JAX 可以求导循环、分支、递归和闭包函数，并且它可...