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