Folax (Finite Operator Learning with JAX) is a framework for solving and optimizing PDEs by integrating machine learning with numerical methods in computational mechanics.
jax pdes sci-ml physics-informed-neural-networks operator-learning numerical-simulation-optimization
-
Updated
Oct 22, 2025 - Python