nlpstack.tasks.causal_language_modeling.torch module#