That is excellent! I have an almost identical line of code in my work (I am using deep galerkin methods, which, I believe, live in a neighbourhood not too far away from PINN’s). If you are happy to talk about it in more detail that would be brilliant as I’ve only just started down this road and any insights you can offer would be massively appreciated.
If you are happy to talk about it (I am a new user also so can’t PM) please just drop me an email here:
revoheb871@vewku.com
It’s a throwaway but I’ll give you my work email directly if that’s ok (just to spare posting my email publicly). Thank you for your fast reply and heavy detail!