支援的作業系統和硬體
我們目前支援下列 GPU 核心:
tokamax.dot_product_attention(FlashAttention)。tokamax.gated_linear_unit(閘控線性單元 (SwiGLU 等))。tokamax.layer_norm(層正規化和 均方根正規化)。
GPU 和 TPU 則適用下列條件:
我們目前支援下列 GPU 核心:
tokamax.dot_product_attention
(FlashAttention)。tokamax.gated_linear_unit
(閘控線性單元 (SwiGLU 等))。tokamax.layer_norm
(層正規化和
均方根正規化)。GPU 和 TPU 則適用下列條件:
除非另有註明,否則本頁面中的內容是採用創用 CC 姓名標示 4.0 授權,程式碼範例則為阿帕契 2.0 授權。詳情請參閱《Google Developers 網站政策》。Java 是 Oracle 和/或其關聯企業的註冊商標。
上次更新時間:2025-11-21 (世界標準時間)。