Make it easier to restore simulations from sharded checkpoints.