How was the stop-gradient operator in pi0.5 report realized?

[ChenyuZhangX]

I was wondering how the stop-gradient in the pi0.5 paper was realized. I found this part was hard to find in this repo.