torch numpy pandas scikit-learn code_generator nlp_encoder speech_encoder text_generator vision_encoder