跳至主要内容
返回顶部
Ctrl
+
K
快速入门
亚麻基础
指南
亚麻基础
JAX 101
亚麻基础
管理参数和状态
setup
vs
compact
处理亚麻模块参数
亚麻中的随机性和 PRNG
数据预处理
处理整个数据集
加载数据集
训练技巧
批量归一化
丢弃
学习率调度
迁移学习
保存和加载检查点
并行训练
在多个设备上集成
在多个设备上扩展亚麻模块
模型检查
模型手术
提取中间值
转换和升级
从 Haiku 迁移到亚麻
将 PyTorch 模型转换为亚麻
将检查点迁移到 Orbax
将我的代码库升级到 Optax
将我的代码库升级到 Linen
RNNCellBase 升级指南
迁移到常规字典
量化
使用 FP8 的用户指南
关键部分
示例
核心示例
谷歌研究示例
使用亚麻的存储库
社区示例
词汇表
常见问题解答 (FAQ)
开发人员笔记
亚麻模块生命周期
提升的转换
FLIPs
亚麻理念
如何贡献
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 包
亚麻 NNX
.rst
.pdf
亚麻基础
亚麻基础
#
JAX 101
亚麻基础
管理参数和状态
setup
vs
compact
处理亚麻模块参数
亚麻中的随机性和 PRNG