Google Colab link for the mini-workshop part I at our Weekly today: Physics Informed Machine Learning led by Dr. Adrian Buganza Tepole. This builds to a degree from the previous two sessions (
Part
I,
Part II) on Solving ODEs and PDEs with Dr. Linlin Li.
Mini workshop: Physics
informed machine learning
Machine learning has impacted all fields of engineering. Artificial neural
networks are universal function approximators. Thus, it is not surprising that they can be used to represent the types of functions that arise as solution of ordinary or partial differential equations (ODEs or PDEs respectively).
ODEs and PDEs are natural mathematical models for many physical phenomena
such as diffusion, heat transfer, and mechanical equilibrium. Previous workshops
as part of EMBRIO have already focused on the solution of ODEs and PDEs with more traditional methods, namely finite differences.
This
workshop builds on that knowledge and explores the use of artificial neural networks for the solution of the same kinds of problems. We will introduce JAX, a numerical linear algebra package which is an alternative to other, perhaps more popular machine learning
tools such as pytorch.
The
reason to do the workshop around JAX is that this library allows for just-in-time compilation of code and vectorization which make it very efficient. Additionally, if attendees are familiar with the standard python packages numpy and scipy then JAX will hopefully
require a less steep learning curve. Using JAX we will show that the PDEs of interest can be used as part of the loss function such that minimization of this objective yields the PDE solution.