Phase field models using JAX.
This code is an implementation of Cahn-Hilliard and Allen-Cahn using JAX for grid-based phase field model simulations. The free energy functionals implemented in this package are related to self-assembly principles (see GitHub here for more direct references).
