/pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers