I confirm that saving tab groups does not work on STP 126 in Big Sur. Actually, this is main reason for me to use new Safari. Please fix this problem.

I also have updated today to 127 on Big Sur and don't see any sign of Tab Groups there.