cs.LG, cs.NA, math.NA, physics.comp-ph

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

arXiv:2605.10159v1 Announce Type: new
Abstract: 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 design is a tracing system …