tests/tests_pytorch/strategies - Lightning-AI/pytorch-lightning

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

30975 0 0 Python

	launchers
	scripts
	__init__.py
	test_common.py	3.3 KB
	test_custom_strategy.py	1.9 KB
	test_ddp_integration_comm_hook.py	8.2 KB
	test_ddp_integration.py	17.4 KB
	test_ddp.py	8.8 KB
	test_deepspeed.py	49.4 KB
	test_fsdp.py	39.2 KB
	test_model_parallel_integration.py	22.1 KB
	test_model_parallel.py	12.0 KB
	test_registry.py	5.5 KB
	test_single_device.py	4.6 KB
	test_xla.py	2.5 KB