JAX

JAX

分类： AI开放平台
访问次数： 0 次
收录时间：

网站简介

JAX是Google推出的高性能数值计算库，提供类似NumPy的API，支持GPU/TPU加速、自动微分、即时编译（JIT）和向量化等功能。

访问网站