示例# 此目录包含使用optax库的示例。 对抗训练 使用Flax和Optax的CIFAR10上的ResNet。 使用Flax的简单神经网络。 梯度累积 L-BFGS 线性分配问题 最优运输 MNIST上的前瞻优化器 元学习 多层感知机 MNIST 小型莎士比亚的字符级变换器 超参数和数据集下载 数据准备 NanoLM模型定义 状态、优化器和损失定义 模型训练 文本生成 双线性最小-最大问题中的乐观梯度下降 扰动优化器 最大值的独热编码 可微排名 一般输入/输出(Pytrees)