nlpstack.integrations.torch.training.callbacks module#