jaxmat

Read full text
Authors: Bleyer, Jeremy, Bouteiller, Paul, Hale, Jack S.
Cited by: 0

Abstract

jaxmat is an open-source library for implementing material constitutive models in a way that integrates seamlessly with modern machine learning frameworks and existing finite element software. The library is implemented in JAX and heavily relies on additional features provided by the JAX ecosystem, in particular the equinox, diffrax, and optimistix libraries.

Publication details

Journal / source
Zenodo (CERN European Organization for Nuclear Research)
Publication date
8 July 2026