nlpstack.integrations.torch.training.optimizers module#