跳至主要內容
OpenXLA Project
XLA
  • 安裝/使用
  • 說明文件
  • 原始碼
StableHLO
  • 說明文件
  • 原始碼
Shardy
  • 說明文件
  • 原始碼
PJRT
  • 說明文件
  • 原始碼
XProf
  • 說明文件
  • 原始碼
Tokamax
  • 說明文件
  • 原始碼
Community
  • 2025 年 JAX 和 OpenXLA 夏季開發人員實驗室
  • 事件和錄製內容
  • 社區會議
  • OpenXLA 討論
  • Discord
產業合作夥伴 聯絡資訊
/
  • English
  • Deutsch
  • Español – América Latina
  • Français
  • Indonesia
  • Italiano
  • Polski
  • Português – Brasil
  • Tiếng Việt
  • Türkçe
  • Русский
  • עברית
  • العربيّة
  • فارسی
  • हिंदी
  • বাংলা
  • ภาษาไทย
  • 中文 – 简体
  • 中文 – 繁體
  • 日本語
  • 한국어
登入
  • Tokamax
OpenXLA Project
  • XLA
    • 更多
  • StableHLO
    • 更多
  • Shardy
    • 更多
  • PJRT
    • 更多
  • XProf
    • 更多
  • Tokamax
    • 更多
  • Community
    • 更多
  • 產業合作夥伴
  • 聯絡資訊
  • Tokamax
  • 簡介
  • 開發人員指南
    • 支援的作業系統和硬體
    • 自動調整
    • 基準化和數字
    • 貢獻
  • 示範筆記本
    • Tokamax Demo
  • 作業專屬說明文件
    • Splash Attention
  • 安裝/使用
  • 說明文件
  • 原始碼
  • 說明文件
  • 原始碼
  • 說明文件
  • 原始碼
  • 說明文件
  • 原始碼
  • 說明文件
  • 原始碼
  • 說明文件
  • 原始碼
  • 2025 年 JAX 和 OpenXLA 夏季開發人員實驗室
  • 事件和錄製內容
  • 社區會議
  • OpenXLA 討論
  • Discord
本頁面由 Cloud Translation API 翻譯而成。
  • OpenXLA
  • Tokamax
透過集合功能整理內容 你可以依據偏好儲存及分類內容。

支援的作業系統和硬體

我們目前支援下列 GPU 核心:

  • tokamax.dot_product_attention (FlashAttention)。
  • tokamax.gated_linear_unit (閘控線性單元 (SwiGLU 等))。
  • tokamax.layer_norm (層正規化和 均方根正規化)。

GPU 和 TPU 則適用下列條件:

  • tokamax.ragged_dot (專家混合模型)。

除非另有註明,否則本頁面中的內容是採用創用 CC 姓名標示 4.0 授權,程式碼範例則為阿帕契 2.0 授權。詳情請參閱《Google Developers 網站政策》。Java 是 Oracle 和/或其關聯企業的註冊商標。

上次更新時間:2025-11-21 (世界標準時間)。

  • 保持連線

    • Twitter
    • YouTube
  • 條款
  • 隱私權
  • Manage cookies
  • English
  • Deutsch
  • Español – América Latina
  • Français
  • Indonesia
  • Italiano
  • Polski
  • Português – Brasil
  • Tiếng Việt
  • Türkçe
  • Русский
  • עברית
  • العربيّة
  • فارسی
  • हिंदी
  • বাংলা
  • ภาษาไทย
  • 中文 – 简体
  • 中文 – 繁體
  • 日本語
  • 한국어