社区示例

除了 GitHub 上官方 Flax 示例的精选列表之外，越来越多的人正在使用 Flax 构建新型机器学习模型。我们很乐意在此展示社区构建的任何示例！

如果您想提交自己的 Flax 示例，可以从 fork GitHub 上官方 Flax 示例之一开始。

模型

链接	作者	任务类型	参考
matthias-wright/flaxmodels	@matthias-wright	各种	GPT-2、ResNet、StyleGAN-2、VGG、…
DarshanDeshpande/jax-models	@DarshanDeshpande	各种	Segformer、Swin Transformer、… 还有一些独立的层
google/vision_transformer	@andsteing	图像分类、图像/文本	https://arxiv.org/abs/2010.11929, https://arxiv.org/abs/2105.01601, https://arxiv.org/abs/2111.07991, …
jax-resnet	@n2cholas	各种 resnet 实现	torch.hub
Wav2Vec2 微调	@vasudevgupta7	自动语音识别	https://arxiv.org/abs/2006.11477

示例

链接	作者	任务类型	参考
JAX-RL	@henry-prior	强化学习	不适用
BigBird 微调	@vasudevgupta7	问答	https://arxiv.org/abs/2007.14062
DCGAN	@bkkaggle	图像合成	https://arxiv.org/abs/1511.06434
denoising-diffusion-flax	@yiyixuxu	图像生成	https://arxiv.org/abs/2006.11239

教程

链接	作者	任务类型	参考

贡献策略

如果您有兴趣向“社区示例”部分添加项目，请考虑以下事项

• 代码示例：示例必须包含一个有帮助、清晰且解释如何运行代码的 README。代码本身应该易于理解。

• 教程：这些文档最好是 Jupyter Notebook 格式（请参考贡献，了解如何使用jupytext将 Jupyter Notebook 转换为 Markdown 文件）。您的教程应该写得很好，并讨论/描述一个有趣的主题/任务。为避免重复，这些文档的内容必须与 Flax 文档站点上的现有文档或本文档中提到的其他社区示例不同。

• 模型：移植到 Flax 的模型的存储库必须提供以下至少一项

  • 当模型训练完成时，与原始工作相当的指标。强烈建议提供训练期间指标历史的可用图表。

  • 用于验证针对已知实现的数值等效性的测试（相同的输入 + 权重 = 相同的输出），最好使用预训练的权重。

在上述所有情况下，代码必须与以下软件包的最新稳定版本一起使用： jax、flax 和 optax，并大量使用 Flax。请注意，jax 和 optax 都是 flax 的必需软件包（有关更多详细信息，请参阅安装说明）。