This type of plot is a surface plot and you could use matplotlib for it. I don’t know what the current recommended technique is to create this loss surface from a DL model, but e.g. this paper 15 might be useful.