GitHub repository: kaiyaok2/pytorch-lightning
Description: Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.