finite-difference-methods
Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to end.
A Numerical Analysis Package in Python
