nlpstack.tasks.representation_learning.torch module#