pytorch_lightning/accelerators - Lightning-AI/pytorch-lightning - Morph

SIGN IN SIGN UP

Lightning-AI / pytorch-lightning UNCLAIMED

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

0 0 0 Python

	__init__.py	1.7 KB
	accelerator_connector.py	18.3 KB
	accelerator.py	8.9 KB
	cpu_accelerator.py	2.8 KB
	ddp_accelerator.py	13.8 KB
	ddp_cpu_hpc_accelerator.py	1.6 KB
	ddp_cpu_spawn_accelerator.py	10.7 KB
	ddp_hpc_accelerator.py	9.1 KB
	ddp_spawn_accelerator.py	12.0 KB
	ddp2_accelerator.py	9.4 KB
	dp_accelerator.py	6.5 KB
	gpu_accelerator.py	3.4 KB
	horovod_accelerator.py	7.2 KB
	tpu_accelerator.py	13.2 KB