JAX library for neural operator and PDE foundation model training.
Reference datasets for physics parameter estimation and inverse problem benchmarking. Includes synthetic datasets with known ground truth, experimentally collected trajectories, and paired MCMC and Sobol analysis example scripts.
Official repo for separable operator networks -- extreme-scale operator learning for parametric PDEs.
