Perform rescan for github.com/google/brax/pull/598?