Liyang90 47726391ad [TPU] Add support for PJRT from PyTorch/XLA 2.0 (#17352) ... Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>		2023-04-18 18:52:36 +02:00
..
conf	copyright Lightning AI team (#16647)	2023-02-06 15:26:51 +01:00
__init__.py	…
test_amp.py	Skip flaky ddp-spawn test on windows (#16942)	2023-03-03 15:26:28 +01:00
test_cpu.py	Remove `Trainer(multiple_trainloader_mode)` in favor of `CombinedLoader(mode)` (#16800)	2023-02-20 13:32:06 +01:00
test_ddp_fork_amp.py	Introduce new precision layout in PL (#16783)	2023-02-17 17:58:14 +01:00
test_fabric_integration.py	copyright Lightning AI team (#16647)	2023-02-06 15:26:51 +01:00
test_gpu.py	Make DDP subprocess the default launcher for multi-device (#16780)	2023-02-20 11:20:50 +00:00
test_hooks.py	Simplify the Trainer core logic (#17017)	2023-03-13 13:43:57 +00:00
test_hparams.py	Move `Trainer._log_hyperparams` to an utility (#16712)	2023-02-10 09:22:56 +01:00
test_onnx.py	PL: Test PyTorch 2.0 pre-release on CPU and CUDA (#16764)	2023-02-28 18:38:40 +01:00
test_restore.py	Remove `Trainer(multiple_trainloader_mode)` in favor of `CombinedLoader(mode)` (#16800)	2023-02-20 13:32:06 +01:00
test_torchscript.py	copyright Lightning AI team (#16647)	2023-02-06 15:26:51 +01:00
test_tpu.py	[TPU] Add support for PJRT from PyTorch/XLA 2.0 (#17352)	2023-04-18 18:52:36 +02:00