jNO: A JAX Library for Neural Operator and Foundation Model Training

Leon Armbruster, Rathan Ramesh, Georg Kruse +1

jNO (jax Neural Operators) is a JAX-native library for neural operators and foundation models with unified support for both data-driven and physics-informed training. Its core desi…