JAX
AI开发框架
JAX

Google推出变换数值函数的机器学习框架

JAX是一种自动微分的NumPy。所以JAX并不是一个深度学习框架,而是一个科学计算框架。深度学习是JAX功能的一个子集。

既然是NumPy,那就可以用NumPy接口做各类科学计算。

而且还带自动微分,科学计算世界中,微分是最常用的一种计算。JAX的自动微分包含了前向微分、反向微分等各种接口。反正各类花式微分,几乎都可以用JAX实现。

相关导航

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注