3D plot for linear partial differential system with PINN

See Comparison of plotting packages - #2 by ChrisRackauckas.

It’s really your choice. I’d use Plots or Makie. Some of the tutorials show examples like this one:

https://docs.sciml.ai/NeuralPDE/dev/tutorials/gpu/