JAX - Google推出的用于变换数值函数的机器学习框架

产品名称：JAX

官方网址：https://jax.readthedocs.io/en/latest/

进入网站

JAX简介

JAX是Google推出的高性能数值计算库，提供类似NumPy的API，支持GPU/TPU加速、自动微分、即时编译（JIT）和向量化等功能。JAX通过XLA（加速线性代数）编译器优化代码，显著提升运行效率，在大规模数据处理和机器学习中表现突出。JAX支持自动微分，能轻松计算函数梯度，适用于优化算法。JAX的异步执行模式和不可变数组设计使其在性能和可靠性上优于传统NumPy，是现代科学计算和机器学习研究中的重要工具。

浏览统计(最近30天)

相似站点

JAX - Google推出的用于变换数值函数的机器学习框架

飞桨PaddlePaddle

Google AI Studio

JAX

言犀智能体平台

Refly

文心智能体平台

Coze

Dify