2.0.1 patch release

App

No changes

---

Fabric

Changed

• Generalized Optimizer validation to accommodate both FSDP 1.x and 2.x (#16733)

---

PyTorch

Changed

• Pickling the LightningModule no longer pickles the Trainer (#17133)
• Generalized Optimizer validation to accommodate both FSDP 1.x and 2.x (#16733)
• Disable torch.inference_mode with torch.compile in PyTorch 2.0 (#17215)

Fixed

• Fixed issue where pickling the module instance would fail with a DataLoader error (#17130)
• Fixed WandbLogger not showing "best" aliases for model checkpoints when ModelCheckpoint(save_top_k>0) is used (#17121)
• Fixed the availability check for rich that prevented Lightning to be imported in Google Colab (#17156)
• Fixed parsing the precision config for inference in DeepSpeedStrategy (#16973)
• Fixed issue where torch.compile would fail when logging to WandB (#17216)

---

Contributors

@Borda @williamFalcon @lightningforever @adamjstewart @carmocca @tshu-w @saryazdi @parambharat @awaelchli @colehawkins @woqidaideshi @md-121 @yhl48 @gkroiz @idc9 @speediedan

If we forgot someone due to not matching commit email with GitHub account, let us know :]