JAX

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

#Ai工(gōng)具箱 #Ai寫作對話(huà) #AI視頻(pín)生(shēng)成

JAX

GoogleJAX是一(yī)個用于變換數值函數的機器學習框架，訊船Google稱其爲爲結合了修改版本的Autograd（通過函數微分(fēn)制短自動獲得梯度函數）和TensorFlow的XLA（加速線性代數）。該框架和船的設計盡可能遵循NumPy的結構和工(gōng)作流程，并與Tensor頻紅Flow和PyTorch等各種現有框架協同工(gōng)作。

JAX的主要功能是包括：

  • grad：自動微分(fēn)
  • jit：編譯
  • vmap：自動矢量化
  • pmap：SPMD編程

  • 上一(yī)個應用：

  • 下(xià)一(yī)個應用：

    • 與JAX相關工(gōng)具

    TensorFlow
    TensorFlow
    Google推出的機器學習和人工(gōn如錯g)智能開(kāi)源庫
    PyTorch
    PyTorch
    開(kāi)源機器學習框架 技文
    Scikit-learn
    Scikit-learn
    Python機器學習庫