tests/tests_pytorch/plugins/precision - 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 104 Python

	__init__.py
	test_all.py	949 B
	test_amp_integration.py	3.2 KB
	test_amp.py	6.2 KB
	test_bitsandbytes.py	2.2 KB
	test_deepspeed_precision.py	2.4 KB
	test_double.py	6.6 KB
	test_fsdp.py	7.0 KB
	test_half.py	3.0 KB
	test_transformer_engine.py	3.2 KB
	test_xla.py	2.6 KB