The save option is just for saving your entire game progress. There’s no way to save your progress in an individual puzzle unfortunately.