pytorch_lightning/plugins - 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
	apex.py	5.2 KB
	ddp_plugin.py	6.4 KB
	ddp_sequential_plugin.py	16.5 KB
	native_amp.py	2.7 KB
	plugin_connector.py	6.7 KB
	plugin.py	947 B
	precision_plugin.py	1.3 KB
	rpc_plugin.py	4.3 KB
	sharded_native_amp_plugin.py	1.3 KB
	sharded_plugin.py	3.8 KB