pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

What is pytorch-lightning?

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.