nlpstack.integrations.torch.training.trainer module#