GEMSEO plugin for JAX (jit compilation, autodiff, XLA)
Project description
gemseo-jax
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
File details
Details for the file gemseo_jax-2.0.0-py3-none-any.whl.
File metadata
- Download URL: gemseo_jax-2.0.0-py3-none-any.whl
- Upload date:
- Size: 30.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 12aaa568ca4f927dac78dfcabe9dd5539e445634d6e9d29deea49417a1e4df82 |
|
| MD5 | e7a6bc5e151ea3e8fbb3baa023200deb |
|
| BLAKE2b-256 | c1df4e9c7da3c68d117ecdd73217fdf89a3e4ca337cf5370b6303a258953d31d |
