awaelchli b3c869f636 Revise checkpoint consolidation with PyTorch 2.3 (#19561)		2024-03-04 10:13:31 -05:00
..
launchers	Update tests for PyTorch 2.2.1 (#19521)	2024-02-23 13:11:34 -05:00
__init__.py	…
test_ddp.py	…
test_ddp_integration.py	Update tests for PyTorch 2.2.1 (#19521)	2024-02-23 13:11:34 -05:00
test_deepspeed.py	…
test_deepspeed_integration.py	Refactor BoringFabric in tests (#19364)	2024-01-30 23:32:45 +01:00
test_dp.py	…
test_fsdp.py	precommit: ruff-format (#19434)	2024-02-15 13:39:17 -05:00
test_fsdp_integration.py	Revise checkpoint consolidation with PyTorch 2.3 (#19561)	2024-03-04 10:13:31 -05:00
test_registry.py	Support shortcut name for DeepSpeed stage 1 offload (#19075)	2024-02-05 20:53:18 -05:00
test_single_device.py	…
test_strategy.py	…
test_xla.py	Refactor BoringFabric in tests (#19364)	2024-01-30 23:32:45 +01:00
test_xla_fsdp.py	…
test_xla_fsdp_integration.py	Refactor BoringFabric in tests (#19364)	2024-01-30 23:32:45 +01:00