python training machine-learning computer-vision deep-learning neural-network inference pytorch dataset transformer image-classification ai-model multimodal-ai research-code pepagi