跳至主要內容
返回頂端
Ctrl
+
K
快速上手
Flax基礎
指南
Flax基礎知識
JAX入門
Flax基礎
管理參數和狀態
setup
比較
compact
處理Flax模組自變數
處理Flax中的隨機性和PRNG
資料預處理
處理整個資料集
載入資料集
訓練技巧
批次正規化
中斷
學習速率排程
遷移學習
儲存和載入檢查點
平行訓練
裝置融合
裝置融合提升Flax模組
模型檢查
模型修復
提取中間值
轉換和升級
從Haiku遷移至Flax
轉換PyTorch模型至Flax
將Checkpointing遷移至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
模型檢查
模型檢查
#
模型修復
提取中間值
提取中間值的梯度