JAX

JAX

5|0 評價|0 收藏
造訪網站
產品介紹
JAX 是一個用於高效能陣列運算和程式轉換的 Python 函式庫，專為數值運算和機器學習而設計。
收錄時間
2026年7月22日
免費網站AI 框架AI SDKAI 計算機視覺項目AI 大模型 APIAI 機器學習平臺AI 模型部署AI 模型微調AI 模型管理AI 開源大模型AI 工具庫
JAX 產品預覽
造訪網站

什麼是 JAX？

JAX 是一個用於加速器導向陣列運算和程式轉換的 Python 函式庫，專為高效能數值運算和大規模機器學習而設計。它提供了熟悉的 NumPy 風格 API，並包含可組合的函式轉換，用於編譯、批次處理、自動微分和並行化。相同的程式碼可以在多種後端上執行，包括 CPU、GPU 和 TPU。

如何使用 JAX？

  • 透過 pip 或 conda 安裝 JAX，然後匯入 jax 和 jax.numpy。使用 jax.jit 進行即時編譯，jax.vmap 進行自動向量化，jax.grad 進行自動微分，jax.pmap 進行並行化。有關教學課程和 API 參考，請參閱官方文件。

JAX 的核心功能

  • 熟悉的 NumPy 風格 API
  • 可組合的函式轉換（jit、vmap、grad、pmap）
  • 自動微分（前向和反向模式）
  • 即時編譯
  • 自動向量化
  • 跨多個裝置的並行化
  • 支援 CPU、GPU 和 TPU 後端

用戶評價

5.0

/ 5

0 則評價

5 星
0
4 星
0
3 星
0
2 星
0
1 星
0

暫無評價，來寫第一條吧

相似產品

JAX 推廣元件

使用網站徽章展示您的產品已收錄於 Hootool，推動社群對工具發現與傳播的支持。推廣元件可輕鬆新增到首頁或頁尾。

Featured on

Hootool.ai