Perform rescan for github.com/google-deepmind/torax/pull/2132?