lightning 2.6.0.dev20251012__tar.gz → 2.6.0.dev20251019__tar.gz

{lightning-2.6.0.dev20251012/src/lightning.egg-info → lightning-2.6.0.dev20251019}/PKG-INFO

@@ -1,17 +1,16 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: lightning
-Version: 2.6.0.dev20251012
+Version: 2.6.0.dev20251019
 Summary: The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
 Home-page: https://github.com/Lightning-AI/lightning
+Download-URL: https://github.com/Lightning-AI/lightning
 Author: Lightning AI et al.
 Author-email: developer@lightning.ai
 License: Apache-2.0
-Download-URL: https://github.com/Lightning-AI/lightning
 Project-URL: Bug Tracker, https://github.com/Lightning-AI/pytorch-lightning/issues
 Project-URL: Documentation, https://lightning.ai/lightning-docs
 Project-URL: Source Code, https://github.com/Lightning-AI/lightning
 Keywords: deep learning,pytorch,AI
-Platform: UNKNOWN
 Classifier: Environment :: Console
 Classifier: Natural Language :: English
 Classifier: Development Status :: 4 - Beta
@@ -27,26 +26,245 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
-Provides-Extra: pytorch-extra
-Provides-Extra: pytorch-strategies
-Provides-Extra: pytorch-test
-Provides-Extra: pytorch-examples
+License-File: LICENSE
+Requires-Dist: PyYAML<8.0,>5.4
+Requires-Dist: fsspec[http]<2027.0,>=2022.5.0
+Requires-Dist: lightning-utilities<2.0,>=0.10.0
+Requires-Dist: packaging<27.0,>=20.0
+Requires-Dist: torch<4.0,>=2.1.0
+Requires-Dist: torchmetrics<3.0,>0.7.0
+Requires-Dist: tqdm<6.0,>=4.57.0
+Requires-Dist: typing-extensions<6.0,>4.5.0
+Requires-Dist: pytorch-lightning
 Provides-Extra: fabric-extra
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-extra"
 Provides-Extra: fabric-strategies
-Provides-Extra: fabric-test
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-strategies"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
 Provides-Extra: fabric-examples
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-examples"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-examples"
+Provides-Extra: fabric-test
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-test"
+Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "fabric-test"
+Requires-Dist: coverage==7.10.7; extra == "fabric-test"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-test"
+Requires-Dist: pytest-cov==6.3.0; extra == "fabric-test"
+Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-test"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "fabric-test"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "fabric-test"
+Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-test"
+Requires-Dist: pytest==8.4.2; extra == "fabric-test"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-test"
+Provides-Extra: pytorch-extra
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-extra"
+Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-extra"
+Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-extra"
+Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
+Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
+Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
+Provides-Extra: pytorch-strategies
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
+Provides-Extra: pytorch-examples
+Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "pytorch-examples"
+Requires-Dist: requests<3.0; extra == "pytorch-examples"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-examples"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-examples"
+Provides-Extra: pytorch-test
+Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-test"
+Requires-Dist: coverage==7.10.7; extra == "pytorch-test"
+Requires-Dist: fastapi; extra == "pytorch-test"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-test"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-test"
+Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-test"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-test"
+Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-test"
+Requires-Dist: psutil<8.0; extra == "pytorch-test"
+Requires-Dist: pytest-cov==6.3.0; extra == "pytorch-test"
+Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-test"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "pytorch-test"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "pytorch-test"
+Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-test"
+Requires-Dist: pytest==8.4.2; extra == "pytorch-test"
+Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-test"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-test"
+Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "pytorch-test"
+Requires-Dist: uvicorn; extra == "pytorch-test"
 Provides-Extra: fabric-all
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-all"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-all"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-all"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-all"
 Provides-Extra: fabric-dev
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-dev"
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-dev"
+Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "fabric-dev"
+Requires-Dist: coverage==7.10.7; extra == "fabric-dev"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-dev"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-dev"
+Requires-Dist: pytest-cov==6.3.0; extra == "fabric-dev"
+Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-dev"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "fabric-dev"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "fabric-dev"
+Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-dev"
+Requires-Dist: pytest==8.4.2; extra == "fabric-dev"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-dev"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-dev"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-dev"
 Provides-Extra: pytorch-all
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-all"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-all"
+Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "pytorch-all"
+Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-all"
+Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-all"
+Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-all"
+Requires-Dist: requests<3.0; extra == "pytorch-all"
+Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-all"
+Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-all"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-all"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-all"
 Provides-Extra: pytorch-dev
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-dev"
+Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-dev"
+Requires-Dist: coverage==7.10.7; extra == "pytorch-dev"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
+Requires-Dist: fastapi; extra == "pytorch-dev"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-dev"
+Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "pytorch-dev"
+Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-dev"
+Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-dev"
+Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-dev"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-dev"
+Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-dev"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-dev"
+Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-dev"
+Requires-Dist: psutil<8.0; extra == "pytorch-dev"
+Requires-Dist: pytest-cov==6.3.0; extra == "pytorch-dev"
+Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-dev"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "pytorch-dev"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "pytorch-dev"
+Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-dev"
+Requires-Dist: pytest==8.4.2; extra == "pytorch-dev"
+Requires-Dist: requests<3.0; extra == "pytorch-dev"
+Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-dev"
+Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-dev"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-dev"
+Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-dev"
+Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "pytorch-dev"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-dev"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-dev"
+Requires-Dist: uvicorn; extra == "pytorch-dev"
+Provides-Extra: data
+Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
 Provides-Extra: extra
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
+Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
+Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
+Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
+Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
+Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
 Provides-Extra: strategies
-Provides-Extra: test
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "strategies"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
 Provides-Extra: examples
-Provides-Extra: data
+Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "examples"
+Requires-Dist: requests<3.0; extra == "examples"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "examples"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "examples"
+Provides-Extra: test
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "test"
+Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "test"
+Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
+Requires-Dist: coverage==7.10.7; extra == "test"
+Requires-Dist: fastapi; extra == "test"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "test"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "test"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "test"
+Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "test"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "test"
+Requires-Dist: pandas<3.0,>2.0; extra == "test"
+Requires-Dist: psutil<8.0; extra == "test"
+Requires-Dist: pytest-cov==6.3.0; extra == "test"
+Requires-Dist: pytest-random-order==1.2.0; extra == "test"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "test"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "test"
+Requires-Dist: pytest-timeout==2.4.0; extra == "test"
+Requires-Dist: pytest==8.4.2; extra == "test"
+Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "test"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "test"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "test"
+Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "test"
+Requires-Dist: uvicorn; extra == "test"
 Provides-Extra: all
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "all"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "all"
+Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "all"
+Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "all"
+Requires-Dist: matplotlib<4.0,>3.1; extra == "all"
+Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "all"
+Requires-Dist: requests<3.0; extra == "all"
+Requires-Dist: rich<15.0,>=12.3.0; extra == "all"
+Requires-Dist: tensorboardX<3.0,>=2.2; extra == "all"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "all"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "all"
 Provides-Extra: dev
-License-File: LICENSE
+Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "dev"
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "dev"
+Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "dev"
+Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
+Requires-Dist: coverage==7.10.7; extra == "dev"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
+Requires-Dist: fastapi; extra == "dev"
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "dev"
+Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "dev"
+Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "dev"
+Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "dev"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "dev"
+Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "dev"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "dev"
+Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "dev"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "dev"
+Requires-Dist: pandas<3.0,>2.0; extra == "dev"
+Requires-Dist: psutil<8.0; extra == "dev"
+Requires-Dist: pytest-cov==6.3.0; extra == "dev"
+Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "dev"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "dev"
+Requires-Dist: pytest-timeout==2.4.0; extra == "dev"
+Requires-Dist: pytest==8.4.2; extra == "dev"
+Requires-Dist: requests<3.0; extra == "dev"
+Requires-Dist: rich<15.0,>=12.3.0; extra == "dev"
+Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "dev"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "dev"
+Requires-Dist: tensorboardX<3.0,>=2.2; extra == "dev"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "dev"
+Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "dev"
+Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "dev"
+Requires-Dist: torchvision<1.0,>=0.16.0; extra == "dev"
+Requires-Dist: uvicorn; extra == "dev"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: download-url
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 <div align="center">
 
@@ -55,7 +273,7 @@ License-File: LICENSE
 <br/>
 <br/>
 
-**The deep learning framework to pretrai and finetune AI models.**
+**The deep learning framework to pretrain and finetune AI models.**
 
 **Deploying models?** Check out [LitServe](https://github.com/Lightning-AI/litserve?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme), the PyTorch Lightning for inference engines
 
@@ -68,7 +286,7 @@ ______________________________________________________________________
   <a href="#lightning-fabric-expert-control">Fabric</a> •
   <a href="https://lightning.ai/?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme">Lightning Cloud</a> •   
   <a href="#community">Community</a> •
-  <a href="https://pytorch-lightning.readthedocs.io/en/2.6.0.dev20251012">Docs</a>
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.6.0.dev20251019">Docs</a>
 </p>
 
 <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -610,7 +828,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
 
 |       System / PyTorch ver.        | 1.13                                                                                                                                                                                                                            | 2.0                                                                                                                                                                                                                             |                                                                                                               2.1                                                                                                               |
 | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
-|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.6.0.dev20251012)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
+|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.6.0.dev20251019)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
 |  Linux (multiple Python versions)  | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |
 |   OSX (multiple Python versions)   | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |
 | Windows (multiple Python versions) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |
@@ -639,5 +857,3 @@ If you have any questions please:
 1. [Read the docs](https://lightning.ai/docs?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme).
 1. [Search through existing Discussions](https://github.com/Lightning-AI/lightning/discussions), or [add a new question](https://github.com/Lightning-AI/lightning/discussions/new)
 1. [Join our discord](https://discord.com/invite/tfXFetEZxv).
-
-

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/README.md

@@ -5,7 +5,7 @@
 <br/>
 <br/>
 
-**The deep learning framework to pretrai and finetune AI models.**
+**The deep learning framework to pretrain and finetune AI models.**
 
 **Deploying models?** Check out [LitServe](https://github.com/Lightning-AI/litserve?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme), the PyTorch Lightning for inference engines
 

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/requirements/fabric/strategies.txt

@@ -5,5 +5,5 @@
 
 # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
 #  shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
-deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin"  # strict
+deepspeed >=0.15.0,<0.17.0; platform_system != "Windows" and platform_system != "Darwin"  # strict
 bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/requirements/fabric/test.txt

@@ -1,9 +1,10 @@
-coverage ==7.10.6
+coverage ==7.10.7
 numpy >=1.21.0, <1.27.0
 pytest ==8.4.2
 pytest-cov ==6.3.0
 pytest-timeout ==2.4.0
-pytest-rerunfailures ==16.0.1
+pytest-rerunfailures ==16.0.1; python_version < "3.10"
+pytest-rerunfailures ==16.1; python_version >= "3.10"
 pytest-random-order ==1.2.0
 click ==8.1.8; python_version < "3.11"
 click ==8.3.0; python_version > "3.10"

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/requirements/pytorch/strategies.txt

@@ -3,4 +3,4 @@
 
 # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
 #  shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
-deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin"  # strict
+deepspeed >=0.15.0,<0.17.0; platform_system != "Windows" and platform_system != "Darwin"  # strict

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/requirements/pytorch/test.txt

@@ -1,8 +1,9 @@
-coverage ==7.10.6
+coverage ==7.10.7
 pytest ==8.4.2
 pytest-cov ==6.3.0
 pytest-timeout ==2.4.0
-pytest-rerunfailures ==16.0.1
+pytest-rerunfailures ==16.0.1; python_version < "3.10"
+pytest-rerunfailures ==16.1; python_version >= "3.10"
 pytest-random-order ==1.2.0
 
 # needed in tests

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/fabric/strategies/deepspeed.py

@@ -37,6 +37,7 @@ from lightning.fabric.strategies.ddp import DDPStrategy
 from lightning.fabric.strategies.registry import _StrategyRegistry
 from lightning.fabric.strategies.strategy import _Sharded
 from lightning.fabric.utilities.distributed import log
+from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_6
 from lightning.fabric.utilities.load import _move_state_into
 from lightning.fabric.utilities.rank_zero import rank_zero_info, rank_zero_warn
 from lightning.fabric.utilities.seed import reset_seed
@@ -47,6 +48,7 @@ if TYPE_CHECKING:
     from torch.optim.lr_scheduler import _LRScheduler
 
 _DEEPSPEED_AVAILABLE = RequirementCache("deepspeed")
+_DEEPSPEED_GREATER_EQUAL_0_16 = RequirementCache("deepspeed>=0.16.0")
 
 
 # TODO(fabric): Links in the docstrings to PL-specific deepspeed user docs need to be replaced.
@@ -239,6 +241,19 @@ class DeepSpeedStrategy(DDPStrategy, _Sharded):
                 " Install it by running `pip install -U deepspeed`."
             )
 
+        if _TORCH_GREATER_EQUAL_2_6 and not _DEEPSPEED_GREATER_EQUAL_0_16:
+            # Starting with PyTorch 2.6, `torch.load` defaults to `weights_only=True` when loading full checkpoints.
+            # DeepSpeed added support for this behavior in version 0.16.0.
+            import deepspeed
+
+            deepspeed_version = deepspeed.__version__
+
+            raise ImportError(
+                f"PyTorch >= 2.6 requires DeepSpeed >= 0.16.0. "
+                f"Detected DeepSpeed version: {deepspeed_version}. "
+                "Please upgrade by running `pip install -U 'deepspeed>=0.16.0'`."
+            )
+
         super().__init__(
             accelerator=accelerator,
             parallel_devices=parallel_devices,

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/fabric/utilities/imports.py

@@ -36,5 +36,6 @@ _TORCH_GREATER_EQUAL_2_4 = compare_version("torch", operator.ge, "2.4.0")
 _TORCH_GREATER_EQUAL_2_4_1 = compare_version("torch", operator.ge, "2.4.1")
 _TORCH_GREATER_EQUAL_2_5 = compare_version("torch", operator.ge, "2.5.0")
 _TORCH_LESS_EQUAL_2_6 = compare_version("torch", operator.le, "2.6.0")
+_TORCH_GREATER_EQUAL_2_6 = compare_version("torch", operator.ge, "2.6.0")
 _TORCHMETRICS_GREATER_EQUAL_1_0_0 = compare_version("torchmetrics", operator.ge, "1.0.0")
 _PYTHON_GREATER_EQUAL_3_10_0 = (sys.version_info.major, sys.version_info.minor) >= (3, 10)

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/pytorch/CHANGELOG.md

@@ -52,6 +52,9 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 - Fixed preventing recursive symlink creation iwhen `save_last='link'` and `save_top_k=-1` ([#21186](https://github.com/Lightning-AI/pytorch-lightning/pull/21186))
 
 
+- Fixed `ModelPruning` sparsity logging bug that caused incorrect sparsity percentages ([#21223](https://github.com/Lightning-AI/pytorch-lightning/pull/21223))
+
+
 - Fixed `LightningCLI` loading of hyperparameters from `ckpt_path` failing for subclass model mode ([#21246](https://github.com/Lightning-AI/pytorch-lightning/pull/21246))
 
 

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/pytorch/callbacks/pruning.py

@@ -349,7 +349,7 @@ class ModelPruning(Callback):
     def _log_sparsity_stats(
         self, prev: list[tuple[int, int]], curr: list[tuple[int, int]], amount: Union[int, float] = 0
     ) -> None:
-        total_params = sum(p.numel() for layer, _ in self._parameters_to_prune for p in layer.parameters())
+        total_params = sum(total for _, total in curr)
         prev_total_zeros = sum(zeros for zeros, _ in prev)
         curr_total_zeros = sum(zeros for zeros, _ in curr)
         log.info(

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/pytorch/strategies/deepspeed.py

@@ -35,10 +35,12 @@ from lightning.fabric.plugins.collectives.torch_collective import default_pg_tim
 from lightning.fabric.strategies import _StrategyRegistry
 from lightning.fabric.strategies.deepspeed import (
     _DEEPSPEED_AVAILABLE,
+    _DEEPSPEED_GREATER_EQUAL_0_16,
     _format_precision_config,
     _validate_checkpoint_directory,
     _validate_device_index_selection,
 )
+from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_6
 from lightning.fabric.utilities.optimizer import _optimizers_to_device
 from lightning.fabric.utilities.seed import reset_seed
 from lightning.fabric.utilities.types import _PATH
@@ -262,6 +264,19 @@ class DeepSpeedStrategy(DDPStrategy):
                 " Install it by running `pip install -U deepspeed`."
             )
 
+        if _TORCH_GREATER_EQUAL_2_6 and not _DEEPSPEED_GREATER_EQUAL_0_16:
+            # Starting with PyTorch 2.6, `torch.load` defaults to `weights_only=True` when loading full checkpoints.
+            # DeepSpeed added support for this behavior in version 0.16.0.
+            import deepspeed
+
+            deepspeed_version = deepspeed.__version__
+
+            raise ImportError(
+                f"PyTorch >= 2.6 requires DeepSpeed >= 0.16.0. "
+                f"Detected DeepSpeed version: {deepspeed_version}. "
+                "Please upgrade by running `pip install -U 'deepspeed>=0.16.0'`."
+            )
+
         super().__init__(
             accelerator=accelerator,
             parallel_devices=parallel_devices,

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py

@@ -91,8 +91,7 @@ class _Sync:
         """Used to compute the syncing function and cache it."""
         fn = self.no_op if self.fn is None or not self.should or self.rank_zero_only else self.fn
         # save the function as `_fn` as the meta are being re-created and the object references need to match.
-        # ignore typing, bad support for `partial`: mypy/issues/1484
-        self._fn: Callable = partial(fn, reduce_op=self.op, group=self.group)  # type: ignore[unused-ignore]
+        self._fn: Callable = partial(fn, reduce_op=self.op, group=self.group)
 
     @property
     def __call__(self) -> Any:

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/pytorch/utilities/deepspeed.py

@@ -93,10 +93,10 @@ def convert_zero_checkpoint_to_fp32_state_dict(
     ]
     checkpoint_dir = ds_checkpoint_dir(checkpoint_dir)
     optim_files = get_optim_files(checkpoint_dir)
-    optim_state = torch.load(optim_files[0], map_location=CPU_DEVICE)
+    optim_state = torch.load(optim_files[0], map_location=CPU_DEVICE, weights_only=False)
     zero_stage = optim_state["optimizer_state_dict"]["zero_stage"]
     model_file = get_model_state_file(checkpoint_dir, zero_stage)
-    client_state = torch.load(model_file, map_location=CPU_DEVICE)
+    client_state = torch.load(model_file, map_location=CPU_DEVICE, weights_only=False)
     client_state = {key: value for key, value in client_state.items() if key not in deepspeed_states}
     # State dict keys will include reference to wrapper _LightningModuleWrapperBase in old checkpoints created in
     # Lightning version < 2.1. Delete the `_forward_module` prefix before saving.

{lightning-2.6.0.dev20251012 → lightning-2.6.0.dev20251019}/src/lightning/pytorch/utilities/model_helpers.py

@@ -19,7 +19,7 @@ from typing import TYPE_CHECKING, Any, Callable, Generic, Optional, TypeVar
 
 from lightning_utilities.core.imports import RequirementCache
 from torch import nn
-from typing_extensions import Concatenate, ParamSpec
+from typing_extensions import Concatenate, ParamSpec, override
 
 import lightning.pytorch as pl
 
@@ -104,26 +104,32 @@ _P = ParamSpec("_P")  # parameters of the decorated method
 _R_co = TypeVar("_R_co", covariant=True)  # return type of the decorated method
 
 
-class _restricted_classmethod_impl(Generic[_T, _R_co, _P]):
+class _restricted_classmethod_impl(classmethod, Generic[_T, _P, _R_co]):
     """Drop-in replacement for @classmethod, but raises an exception when the decorated method is called on an instance
     instead of a class type."""
 
+    method: Callable[Concatenate[type[_T], _P], _R_co]
+
     def __init__(self, method: Callable[Concatenate[type[_T], _P], _R_co]) -> None:
+        super().__init__(method)
         self.method = method
 
-    def __get__(self, instance: Optional[_T], cls: type[_T]) -> Callable[_P, _R_co]:
+    @override
+    def __get__(self, instance: _T, cls: Optional[type[_T]] = None) -> Callable[_P, _R_co]:  # type: ignore[override]
         # The wrapper ensures that the method can be inspected, but not called on an instance
         @functools.wraps(self.method)
         def wrapper(*args: Any, **kwargs: Any) -> _R_co:
             # Workaround for https://github.com/pytorch/pytorch/issues/67146
             is_scripting = any(os.path.join("torch", "jit") in frameinfo.filename for frameinfo in inspect.stack())
+            cls_type = cls if cls is not None else type(instance)
             if instance is not None and not is_scripting:
                 raise TypeError(
-                    f"The classmethod `{cls.__name__}.{self.method.__name__}` cannot be called on an instance."
+                    f"The classmethod `{cls_type.__name__}.{self.method.__name__}` cannot be called on an instance."
                     " Please call it on the class type and make sure the return value is used."
                 )
-            return self.method(cls, *args, **kwargs)
+            return self.method(cls_type, *args, **kwargs)
 
+        wrapper.__func__ = self.method
         return wrapper
 
 

lightning-2.6.0.dev20251019/src/lightning/version.info

@@ -0,0 +1 @@
+2.6.0.dev20251019