跳至主要內容
返回頂部
Ctrl
+
K
Flax 基礎
MNIST 教學
為何選擇 Flax NNX？
指南
轉換
在多個裝置上擴展
使用篩選器
隨機性
效能考量
從 Flax Linen 到 NNX 的演變
同時使用 Flax NNX 和 Linen
模型手術
儲存和載入檢查點
Flax NNX 與 JAX 轉換
從 Haiku 遷移到 Flax
範例：使用預訓練的 Gemma 與 Flax NNX 進行推論
範例
核心範例
Flax NNX 詞彙表
Flax 哲學
如何貢獻
API 參考
flax.nnx
graph
module
nn
激活函數
注意力機制
資料類型
初始化器
線性
LoRA
正規化
遞迴
隨機
rnglib
spmd
state
訓練
指標
優化器
轉換
變數
輔助函數
視覺化
filterlib
bridge
flax.core.frozen_dict 套件
flax.struct 套件
flax.training 套件
.rst
.pdf
範例
範例
#
核心範例
Transformer
玩具範例
