Google JAX是一個用於轉換數值函數的機器學習框架，谷歌稱其為Autograd（梯度函數通過函數微分自動獲得）和TensorFlowXLA（加速線性代數）的修改版的結合。 該框架旨在盡可能遵循NumPyas的結構和工作流程，並與各種現有框架集成在一起工作。
JAX的主要功能包括：
- grad：自動微分
- jit: 編譯
- vmap：自動矢量化
- pmap：SPMD 編程
相關導航
暫無評論...
AiWindVane 收集了來自全球的各種 AI 工具，包括全面的 AI 轉錄工具、AI 圖像工具、AI 視頻工具、AI 聲音工具、AI 編輯工具、AI 音樂工具、AI 設計工具和 AI 討論工具。 提供AI學習和開發常用網絡、框架和模型全表，支持和理解AI時間，並有效應用到工作和生活中。