On PC, escape menu brings "Load from last checkpoint" option. On mobile, click on 3 lines located at top-left corner of the screen to get this option.
This is usually enough, but if you want to completely reset the stage in Demo, you can reset your save data from options menu. In full version you can restart a stage from 'Stage Select' menu.