JAX

AI 编程

简介JAX是Google推出的一个高性能、跨平台的Python数组计算库，适用于大规模机器学习和数值计算