Bump pytorch-lightning[extra] from 1.9.4 to 2.0.0 in /requirements

[dependabot]

Bumps pytorch-lightning[extra] from 1.9.4 to 2.0.0.

Release notes

Sourced from pytorch-lightning[extra]'s releases.

Lightning 2.0: Fast, Flexible, Stable

Lightning AI is excited to announce the release of Lightning 2.0 ⚡

• Highlights
• Backward Incompatible Changes
  • PyTorch
  • Fabric
• Full Changelog
  • PyTorch
  • Fabric
  • App
• Contributors

Over the last couple of years PyTorch Lightning has become the preferred deep learning framework for researchers and ML developers around the world, with close to 50 million downloads and 18k OSS projects, from top universities to leading labs.

With the help of over 800 contributors, we have added many features and functionalities to make it the most complete research toolkit possible, but some of these changes also introduced issues:

• API changes to the trainer
• Trainer code became harder to follow
• Many integrations made Lightning appear bloated
• The trainer became harder to customize / takes away what I instead need to tweak / have control over.

To make the research experience better, we are introducing 2.0:

• No API changes - We commit to backward compatibility in the 2.0 series
• Simplified abstraction layers, removed legacy functionality, integrations out of the main repo. This improves the project's readability and debugging experience.
• Introducing Fabric. Scale any PyTorch model with just a few lines of code. Read-on!

Highlights

PyTorch 2.0 and torch.compile

Lightning 2.0 is best friends with PyTorch 2.0. You can torch.compile your LightningModules now!

import torch
import lightning as L
model = LitModel()
This will compile forward and {training,validation,test,predict}_step
compiled_model = torch.compile(model)
trainer = L.Trainer()
trainer.fit(compiled_model)

PyTorch reports that on average, "models runs 43% faster in training on an NVIDIA A100 GPU. At Float32 precision, it runs 21% faster on average and at AMP Precision it runs 51% faster on average" (source). If you want to learn more about torch.compile and how such speedups can be achieved, read the official PyTorch 2.0 blog post.

... (truncated)

Commits

• 01834c8 Exclude some examples from docs navigation (#17081)
• f25dfbe update links to Discord (#17087)
• 2f39bb6 docs: migration guide to the latest [1/n] (#17034)
• 077106d Prepare 2.0.0 release (#17056)
• 888db1a Update README (#17082)
• a22242b Improve the error message for installing tensorboardx (#17053)
• 550fa5f Organize app examples (#17045)
• 5d58736 Rename PL installation to pip install lightning (#17074)
• 281efc6 Update data docs (#16839)
• 67b94ef Avoid inference_mode with FSDP (#17064)
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)

[adamjstewart]

PyTorch Lightning was renamed to Lightning back in v1.8, but the old package name is still updated on PyPI. The old name is deprecated and may be removed someday. Also, the docs now only document the new name, causing the docs to fail to build. I switched to the new name, which bumps our min lightning version (and radiant-mlhub too), but simplifies the import name and makes things future proof.

Source: Lightning-AI/pytorch-lightning#17095

[adamjstewart]

pytorch_lightning type hints are public, but lightning type hints aren't. We can fix that: Lightning-AI/pytorch-lightning#17100