Shut down compile workers after compilation (if torch.compile is enabled)

Shut down compile workers after compilation (if torch.compile is enabled) #573