跳到主要內容
返回頂端
Ctrl
+
K
快速入門
Flax 基礎知識
指南
Flax 基礎
JAX 101
Flax 基礎知識
管理參數和狀態
setup
相對於
compact
處理 Flax 模組參數
Flax 中的 randomness 和 PRNG
資料前處理
處理整個資料集
載入資料集
訓練技巧
批次正規化
Dropout
學習率排程
轉移學習
儲存和載入檢查點
並行訓練
在多個裝置上組建
在多個裝置上提升 Flax 模組
模型檢查
模型手術
提取中間值
轉換和升級
從 Haiku 遷移到 Flax
將 PyTorch 模型轉換為 Flax
將檢查點遷移至 Orbax
將我的程式碼庫升級到 Optax
將我的程式碼庫升級到 Linen
RNNCellBase 升級指南
遷移到常規字典
量子化
使用 FP8 的使用者指南
技術要點
範例
核心範例
Google Research 範例
使用 Flax 的儲存庫
社群範例
名詞對照表
常見問題集 (FAQ)
開發人員筆記
Flax 模組生命週期
提升轉換
FLIP
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
量子化
量子化
#
使用 FP8 的使用者指南
