Hi! Awesome work! I was wondering, how can I easily get the pythia losses fro the original paper? ref: https://arxiv.org/abs/2304.01373 <img width="710" height="296" alt="Image" src="https://github.com/user-attachments/assets/97bd6a1c-482e-400a-a87c-73d93db8a1c4" />