nlpstack.tasks.causal_language_modeling.types module#