跳到主要内容
返回顶部
Ctrl
+
K
Flax 基础知识
MNIST 教程
为什么选择 Flax NNX？
指南
转换
在多个设备上扩展
使用过滤器
随机性
性能考虑
从 Flax Linen 到 NNX 的演变
一起使用 Flax NNX 和 Linen
模型手术
保存和加载检查点
Flax NNX 与 JAX 转换
从 Haiku 迁移到 Flax
示例：使用预训练的 Gemma 和 Flax NNX 进行推理
示例
核心示例
Flax NNX 术语表
Flax 的理念
如何贡献
API 参考
flax.nnx
graph
module
nn
激活函数
注意力机制
数据类型
初始化器
线性层
LoRA
归一化
循环
随机
rnglib
spmd
状态
训练
指标
优化器
转换
变量
助手
可视化
filterlib
bridge
flax.core.frozen_dict 包
flax.struct 包
flax.training 包
.rst
.pdf
示例
示例
#
核心示例
Transformer
玩具示例
