Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes. - GitHub - Lightning-AI/pytorch-lightning at 1.6.0