PipelineStage: Improve error logging and debuggability

[H-Huang]

Add try-except around the forward to also log the stage, shapes, etc. before reraising the exception. Look into which debug flags can be used to handle the hang cases. Document this so it is easy to use.