跳至主要内容
返回顶部
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
数据预处理
数据预处理
#
处理整个数据集
加载数据集