nlpstack.tasks.representation_learning.datamodules module#