跳至主内容
返回顶部
Ctrl
+
K
Flax 基础
MNIST 教程
为何选择 Flax NNX？
JAX/Flax 关键概念
指南
基础
变换
过滤器
随机性
保存和加载检查点
Flax NNX 与 JAX 变换
高级
在多台设备上扩展
性能注意事项
同时使用 Flax NNX 和 Linen
模型修改
迁移
从 NNX 0.10 迁移到 NNX 0.11
从 Flax Linen 迁移到 Flax NNX
从 Haiku 迁移到 Flax NNX
示例
示例：使用预训练的 Gemma 和 Flax NNX 进行推理
核心示例
Flax NNX 术语表
Flax 理念
如何贡献
API 参考
flax.nnx
图
对象
模块
神经网络
激活函数
注意力机制
数据类型
初始化器
线性层
LoRA
归一化
循环层
随机层
随机数生成库
spmd
状态
训练
指标
优化器
变换
变量
辅助函数
可视化
过滤器库
桥接
flax.core.frozen_dict 包
flax.struct 包
flax.training 包
.rst
.pdf
基础
基础
#
基础
变换
过滤器
随机性
保存和加载检查点
Flax NNX 与 JAX 变换
