Skip to content

gemseo-jax

PyPI - License PyPI - Python Version PyPI Codecov branch

Overview

GEMSEO plugin for JAX (jit compilation, autodiff, XLA).

The power of JAX (a Python library for high-performance array computing) to accelerate MDO.

JAX is heavily used for large-scale machine learning research, but many of its benefits can also be used to leverage scientific computing as a whole. In the context of Multidisciplinary Optimization (MDO), we use JAX to avoid manual implementation of derivatives of objective functions and constraints wrt optimization variables, which allows for using gradient-based optimizers without an extra implementation cost.

Installation

Install the latest version with pip install gemseo-jax.

See pip for more information.

Bugs and questions

Please use the gitlab issue tracker to submit bugs or questions.

Contributing

See the contributing section of GEMSEO.

Contributors

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