JAX

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

JAX的主要功能是包括：