Cool you might want to look at hugging face and pytorch lightning they should jump start you if you haven't already.
<Microsoft Open Source Engineer> I am an AI enthusiast with a passion for engaging with new technologies, history, and computational medicine.
