DeepXDE possess tremendous API to implement Physics Informed Neural Network. Inside the whole program, we can choose Tensorflow, Pytorch, Or the recently famous JAX as backend.
Click inside the file to acquire more detail !
The most intriguing .ipynb file is Driving Shue et al. model , don't miss it !
Citation
@KozakHou, email:[email protected]
Tel : +886-905804898
Affiliation : Department of Space Science and Engineering, National Central University