JAX

开发公司：JAX 开源项目

支持自动微分、即时编译和向量化的数值计算库，采用接近NumPy的编程方式并可利用加速设备。

产品简介

支持自动微分、即时编译和向量化的数值计算库，采用接近NumPy的编程方式并可利用加速设备。适合机器学习及科学计算开发，用变换组合优化计算流程。自动微分：在数值函数上计算所需导数；编译与向量化：把计算组织为适合加速的形式。

在数值函数上计算所需导数。

把计算组织为适合加速的形式。

对于机器学习研究者，可从“自动微分”进入任务，并利用“编译与向量化”继续整理结果；具体价值在于把计算组织为适合加速的形式。

可用自动微分处理相关工作：在数值函数上计算所需导数。

可用编译与向量化处理相关工作：把计算组织为适合加速的形式。