Skip to content

Getting Started¤

picardax logo

Picardax is a library for time-implicit, differentiable PDE solvers in JAX. It is a sister project to Exponax (which provides explicit and exponential time steppers). Where Exponax sidesteps nonlinear solves entirely, Picardax embraces them: it solves the implicit system arising from theta-method (and similar) time discretizations using iterative nonlinear solvers (Picard iteration, Newton's method) that themselves rely on iterative linear solvers (CG, GMRES, BiCGStab).

Installation¤

pip install picardax

Requires Python 3.10+ and JAX 0.4.13+. See the JAX install guide.

License¤

MIT, see here

fkoehler.site  ·  GitHub @ceyron  ·  X @felix_m_koehler