Optimization and Differentiable Simulation for Pattern-forming PDEs
mosaix-pde is a software package for optimizing pattern forming PDEs that appear in different areas of physics, written in JAX. It has code for PDE optimization and control with gradient-based methods and reinforcement learning.
mosaix-pde is hosted on GitHub: https://github.com/acoh64/mosaix-pde 2
(Courtesy of Alex Cohen)