nlpstack.integrations.torch.modules.scalarmix module#