跳到主要内容
返回顶部
Ctrl
+
K
快速入门
Flax 基础
指南
Flax 基础知识
JAX 101
Flax 基础
管理参数和状态
setup
vs
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
量化
量化
#
使用 FP8 的用户指南
