JAX是一种高性能的数组计算库，提供了类似于NumPy的接口，支持自动矢量化和微分，可以在多个设备和多个主机上进行并行计算，JAX官网入口网址

NumPy是一款强大的科学计算库，提供了高效的n维数组操作和数值计算工具，适用于数据分析、统计计算、机器学习和深度学习等领域，NumPy官网入口网址