GoogleJAX是一個用于變換數值函數的機器學習框架，Google稱其為為結合了修改版本的Autograd（通過函數微分自動獲得梯度函數）和TensorFlow的XLA（加速線性代數）。該框架的設計盡可能遵循NumPy的結構和工作流程，并與TensorFlow和PyTorch等各種現有框架協同工作，大家快來AI365導航！
JAX的主要功能是包括：
- grad：自動微分
- jit：編譯
- vmap：自動矢量化
- pmap：SPMD編程
??版權聲明：若無特殊聲明，本站所有文章版權均歸奇想AI導航網原創和所有，未經許可，任何個人、媒體、網站、團體不得轉載、抄襲或以其他方式復制發表本站內容，或在非我站所屬的服務器上建立鏡像。否則，我站將依法保留追究相關法律責任的權利。
類似于JAX的工具
奇想AI導航網收錄了國內外數百個不同類型的AI工具，每日更新和添加最新AI工具，奇想AI導航網還推薦了AI學習開發的常用網站、框架和模型，幫助你加入人工智能浪潮，自動化高效完成任務！
Ctrl + D 或 ? + D 收藏本站到瀏覽器書簽欄。