SOLsTIS

The jinns python package allows to solve differential equations -ODEs,  PDEs, or systems of thereof - using the framework of physics-informed neural network and the JAX ecosystem (automatic differentiation, just-in-time compilation, and more). Its interface is focused on user flexibility, allowing to easily define a "physics" loss and handling the rest.