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.2 KB
	accelerator.py	2.4 KB
	cpu.py	3.0 KB
	gpu.py	5.3 KB
	hpu.py	2.4 KB
	ipu.py	1.8 KB
	registry.py	4.5 KB
	tpu.py	2.5 KB