跳轉至主要內容
返回頂端
Ctrl
+
K
快速入門
Flax 基礎知識
指南
Flax 基本原理
JAX 101
Flax 基礎知識
管理參數和狀態
setup
與
compact
處理 Flax 模組引數
Flax 中的隨機性和 PRNG
資料前處理
處理整個資料集
載入資料集
訓練技巧
批次正規化
丟棄
學習率規劃
轉移學習
儲存和載入檢查點
平行訓練
透過多個裝置進行整體化
在多個裝置上擴充 Flax 模組
模型檢視
模型手術
擷取中間值
轉換和升級
從 Haiku 遷移到 Flax
將 PyTorch 模型轉換成 Flax
將檢查點遷移到 Orbax
將我的程式庫升級為 Optax
將我的程式庫升級為 Linen
RNNCellBase 升級指南
遷移到一般的字典
量化
使用 FP8 的使用者指南
重點
範例
核心範例
Google Research 範例
使用 Flax 的存放庫
社群範例
詞彙
常見問題集 (FAQ)
開發人員備註
Flax 模組的生命週期
提升轉換
FLIPs
Flax 哲學
如何參與貢獻
API 參考
flax.config 套件
flax.core.frozen_dict 套件
flax.cursor 套件
flax.errors 套件
flax.jax_utils 套件
flax.linen
模組
初始化/應用
層級
啟動函式
初始化函式
轉換
檢視
變數字典
SPMD
裝飾符
分析
flax.serialization 套件
flax.struct 套件
flax.traceback_util 套件
flax.training 套件
flax.traverse_util 套件
Flax NNX
.rst
.pdf
開發人員備註
開發人員備註
#
Flax 模組的生命週期
提升轉換
FLIPs