JAX

Google推出的用於變換數值函式的機器學習框架

標籤:

GoogleJAX是一個用於變換數值函式的機器學習框架,Google稱其為為結合了修改版本的Autograd(透過函式微分自動獲得梯度函式)和
的XLA(加速線性代數)。該框架的設計儘可能遵循
的結構和工作流程,並與TensorFlow和
等各種現有框架協同工作。

數據統計

相關導航