支援的作業系統和硬體

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

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

GPU 和 TPU 則適用下列條件：

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