Enregistré dans:
Détails bibliographiques
Auteurs principaux: Gangloff, Hugo, Jouvin, Nicolas
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:https://arxiv.org/abs/2412.14132
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915069914251264
author Gangloff, Hugo
Jouvin, Nicolas
author_facet Gangloff, Hugo
Jouvin, Nicolas
contents jinns is an open-source Python library for physics-informed neural networks, built to tackle both forward and inverse problems, as well as meta-model learning. Rooted in the JAX ecosystem, it provides a versatile framework for efficiently prototyping real-problems, while easily allowing extensions to specific needs. Furthermore, the implementation leverages existing popular JAX libraries such as equinox and optax for model definition and optimisation, bringing a sense of familiarity to the user. Many models are available as baselines, and the documentation provides reference implementations of different use-cases along with step-by-step tutorials for extensions to specific needs. The code is available on Gitlab https://gitlab.com/mia_jinns/jinns.
format Preprint
id arxiv_https___arxiv_org_abs_2412_14132
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle jinns: a JAX Library for Physics-Informed Neural Networks
Gangloff, Hugo
Jouvin, Nicolas
Machine Learning
jinns is an open-source Python library for physics-informed neural networks, built to tackle both forward and inverse problems, as well as meta-model learning. Rooted in the JAX ecosystem, it provides a versatile framework for efficiently prototyping real-problems, while easily allowing extensions to specific needs. Furthermore, the implementation leverages existing popular JAX libraries such as equinox and optax for model definition and optimisation, bringing a sense of familiarity to the user. Many models are available as baselines, and the documentation provides reference implementations of different use-cases along with step-by-step tutorials for extensions to specific needs. The code is available on Gitlab https://gitlab.com/mia_jinns/jinns.
title jinns: a JAX Library for Physics-Informed Neural Networks
topic Machine Learning
url https://arxiv.org/abs/2412.14132