nlpstack.integrations.torch.generation.beam_search module#