nlpstack.tasks.representation_learning.types module#