lightning

Author	SHA1	Message	Date
Adrian Wälchli	3142ed5e44	Integration tests for XLA precision (#18286 )	2023-08-13 09:20:26 -04:00
Adrian Wälchli	03ca31c3d3	Avoid updating the device for XLA FSDP in `Fabric.setup()` [TPU] (#18276 )	2023-08-11 22:00:23 -04:00
Gerson Kroiz	d7c2e597a1	[TPU] Add Fabric support for PyTorch XLA FSDP (#18126 ) Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: awaelchli <aedu.waelchli@gmail.com> Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>	2023-08-02 12:56:00 -04:00