Optax#
Optax 是一个用于 JAX 的梯度处理和优化库。它旨在通过提供可以以自定义方式重新组合的构建模块来促进研究，以便优化参数模型，例如但不限于深度神经网络。
我们的目标是
提供可读性强、经过良好测试、高效的核心组件实现，
通过将低级组件组合成自定义优化器（或其他梯度处理组件），提高研究人员的工作效率。
通过让任何人都能轻松贡献，加速新想法的采纳。
我们倾向于专注于小的可组合的构建块，这些构建块可以有效地组合成定制的解决方案。其他人可以在这些基本组件的基础上构建更复杂的抽象。在合理的情况下，实现优先考虑可读性和结构化代码，以匹配标准方程，而不是代码重用。
安装#
Optax的最新版本可以从PyPI安装，使用以下命令：
pip install optax
你也可以直接从GitHub安装，使用以下命令。这可以用来获取Optax的最新版本：
pip install git+git://github.com/google-deepmind/optax.git
请注意，Optax 是建立在 JAX 之上的。 有关安装 JAX 的说明，请参见 这里。
支持#
如果您在使用此软件时遇到问题，请通过在我们的问题跟踪器上提交问题来告知我们。我们也非常乐意接受错误修复和其他贡献。有关如何贡献的更多信息，请参阅开发指南。
许可证#
Optax 是根据 Apache 2.0 许可证 授权的。