JAX

用于加速器导向的数组计算和程序转换的Python库