Either! You can sprinkle the marshmallows on right after baking and they will melt slightly from the hot cake, add them with about 10 seconds left of cook time so they get more melty, or you can add at the very beginning so they completely melt into the hot cocoa mug cake. Like in the picture below!