Interface Problem by Variational Method - Discontinuous Formulation

Hello. I would like to solve a problem with two materials and interface conditions. I would like to know if it is possible to define a multi-domain PINN with NVIDIA modulus, which implies one NN per domain for each physical quantity. Should I have to define a new architecture with a new loss combining the losses of each domain?

I also find the interface problem example, which is treated by Variational Method. I find it very interesting and I intend to apply it on my test case. However the example only shows the case of continuous type formulation. Is it because the discontinuous formulation can not be implemented with NVIDIA modulus? If it is, is it possible to have the DG version code?

Thanks for your answer.