跳至主要內容
返回頂端
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
資料前處理
資料前處理
#
處理整個資料集
載入資料集