JAX,Google开发的开源库,JAX简介 JAX 是一个由 Google 开发的开源库，专注于高效的数值计算，尤其适用于机器学习和深度学习。JAX 通过结合 NumPy 接口和自动微分（Autograd）技术，为用户提供了强大的计算能力。 JAX主要功能 JAX 的主要功能包括高性能的自动微分、函数转换和对 GPU/TPU 的支
如果没有自动跳转，请 点击这里