跳至主要内容
返回顶部
Ctrl
+
K
Flax 基础
MNIST 教程
指南
变换
在多个设备上扩展
使用过滤器
随机性
从 Linen 到 NNX 的演变
一起使用 Flax NNX 和 Linen
模型手术
Flax NNX 与 JAX 变换
从 Haiku 迁移到 Flax
示例
核心示例
术语表
Flax 理念
如何贡献
API 参考
flax.nnx
graph
module
nn
激活函数
注意力
初始化器
线性
规范化
随机
rnglib
spmd
state
training
指标
优化器
transforms
variables
helpers
visualization
filterlib
bridge
flax.core.frozen_dict 包
flax.struct 包
flax.training 包
.rst
.pdf
示例
示例
#
核心示例
变换
玩具示例
