jax-ml/jax 是一个 Python 开源项目，星火仓评分 85 分，A 级。JAX 是一个用于高性能数值计算和机器学习的 Python 库，能够对 Python+NumPy 程序进行自动微分、向量化、JIT 编译到 GPU/TPU 等变换。
jax (jax-ml) 开源项目详情