Thank you for your inspiring work on latent reasoning! While reproducing your code, I encountered an issue during training and inference: the process consistently ...