Checkpoints to paper https://arxiv.org/abs/2406.10322