GoogleJAX是一個用于變換數值函數的機器學習框架,Google稱其為為結合了修改版本的Autograd(通過函數微分自動獲得梯度函數)和TensorFlow的XLA(加速線性代數)。該框架的設計盡可能遵循NumPy的結構和工作流程,并與TensorFlow和PyTorch等各種現有框架協同工作。
JAX的主要功能是包括:
- grad:自動微分
- jit:編譯
- vmap:自動矢量化
- pmap:SPMD編程

數據統計
數據評估
關于JAX特別聲明
本站OpenI提供的JAX都來源于網絡,不保證外部鏈接的準確性和完整性,同時,對于該外部鏈接的指向,不由OpenI實際控制,在2023年 5月 9日 上午11:23收錄時,該網頁上的內容,都屬于合規合法,后期網頁的內容如出現違規,可以直接聯系網站管理員進行刪除,OpenI不承擔任何責任。
相關導航
暫無評論...