GoogleJAX是一個用于變換數值函數的機器學習框架，Google稱其為為結合了修改版本的Autograd（通過函數微分自動獲得梯度函數）和TensorFlow的XLA（加速線性代數）。該框架的設計盡可能遵循NumPy的結構和工作流程，并與TensorFlow和PyTorch等各種現有框架協同工作。
JAX的主要功能是包括：
- grad：自動微分
- jit：編譯
- vmap：自動矢量化
- pmap：SPMD編程
數據統計
數據評估
關于JAX特別聲明
本站OpenI提供的JAX都來源于網絡，不保證外部鏈接的準確性和完整性，同時，對于該外部鏈接的指向，不由OpenI實際控制，在2023年 5月 9日 上午11:23收錄時，該網頁上的內容，都屬于合規合法，后期網頁的內容如出現違規，可以直接聯系網站管理員進行刪除，OpenI不承擔任何責任。
相關導航
暫無評論...
粵公網安備 44011502001135號