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 files for gemseo-jax 2.1.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| gemseo_jax-2.1.1.tar.gz | 281.3 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| gemseo_jax-2.1.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 316.4 kB
Release files / gemseo_jax-2.1.1.tar.gz
| Download URL | gemseo_jax-2.1.1.tar.gz |
|---|---|
| Size | 281.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
6e3dcba21b8ec445c530aa8ecb31cf772782ce623ff9814ef694d1ebd0ad694b
|
|
BLAKE2b-256 checksum How to use checksums |
c8f22ae83822c75318c23544840916778d2862ecaa0f1b79f07a646b1e016c1a
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.9.7
|
Release files / gemseo_jax-2.1.1-py3-none-any.whl
| Download URL | gemseo_jax-2.1.1-py3-none-any.whl |
|---|---|
| Size | 35.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
f772f6feba613c832db01fb9eaf135c0584b0d33da95c1c500a7f1829c0bdaf7
|
|
BLAKE2b-256 checksum How to use checksums |
4cddff678256e1b76060f4bcb28ccf4cb84455d8b37de37bea4e6835ababd07c
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.9.7
|
