/pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

Primary LanguagePythonApache License 2.0Apache-2.0

Watchers