JAXTensorFlow 简化库认领

JAX 是一个 TensorFlow 的简化库，它结合了 Autograd 和 XLA，专门用于高性能机器学习研究。凭借 Autograd，JAX 可