JAX加速MDO的强大功能。

gemseo-jax是基于GEMSEO和JAX的Python库，用于高性能数组计算的一个插件。

JAXDiscipline类封装了JAX函数，内置自动微分。此类提供以下有用功能

• 对特定输入/输出的雅可比计算图的过滤，
• 函数和雅可比调用进行jit编译以降低重新评估的成本，
• 执行预运行以触发和记录编译时间。

AutoJAXDiscipline是一个特殊的JAXDiscipline，从JAX函数的签名中推断输入名称、输出名称和默认输入值，类似于AutoPyDiscipline。

JAXChain是一个JAXDiscipline，允许组装一系列JaxDiscipline并在JAX中执行它们。这对于避免无意义的JAX与NumPy之间的转换非常有用。

文档

如何获取文档？

错误/问题

请使用gitlab 问题跟踪器提交错误或问题。

许可证

《gemseo-jax》的源代码遵循GNU LGPL v3.0许可协议。您可以在LICENSE.txt文件中找到它的副本。GNU LGPL v3.0许可协议是对GNU GPL v3.0许可协议的一种例外。GNU GPL v3.0许可协议的副本可以在LICENSES文件夹中找到。《gemseo-jax》产品依赖于其他具有不同许可协议的软件。依赖项及其许可列表可以在CREDITS.md文件中找到。

贡献者

• Ian Costa-Alves
• François Gallard
• Matthias De Lozzo
• Antoine DECHAUME