lightning 2.6.1.dev20260201__tar.gz → 2.6.4__tar.gz

{lightning-2.6.1.dev20260201/src/lightning.egg-info → lightning-2.6.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lightning
-Version: 2.6.1.dev20260201
+Version: 2.6.4
 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
@@ -36,16 +36,11 @@ 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
-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-test
 Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-test"
 Requires-Dist: click==8.3.1; python_version > "3.10" and extra == "fabric-test"
 Requires-Dist: coverage==7.10.7; python_version < "3.10" and extra == "fabric-test"
-Requires-Dist: coverage==7.13.1; python_version >= "3.10" and extra == "fabric-test"
+Requires-Dist: coverage==7.13.4; python_version >= "3.10" and extra == "fabric-test"
 Requires-Dist: huggingface-hub; extra == "fabric-test"
 Requires-Dist: numpy<2.0,>1.21.0; python_version < "3.12" and extra == "fabric-test"
 Requires-Dist: numpy<3.0,>2.1.0; python_version >= "3.12" and extra == "fabric-test"
@@ -59,28 +54,24 @@ Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-test"
 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: 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: fabric-extra
+Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-extra"
+Provides-Extra: fabric-strategies
+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: pytorch-test
 Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-test"
 Requires-Dist: coverage==7.10.7; python_version < "3.10" and extra == "pytorch-test"
-Requires-Dist: coverage==7.13.1; python_version >= "3.10" and extra == "pytorch-test"
+Requires-Dist: coverage==7.13.4; python_version >= "3.10" and extra == "pytorch-test"
 Requires-Dist: fastapi; extra == "pytorch-test"
 Requires-Dist: huggingface-hub; extra == "pytorch-test"
 Requires-Dist: numpy<2.0,>1.21.0; python_version < "3.12" and extra == "pytorch-test"
 Requires-Dist: numpy<3.0,>2.1.0; python_version >= "3.12" and extra == "pytorch-test"
+Requires-Dist: onnx-ir<1.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: pandas<4.0,>2.0; extra == "pytorch-test"
 Requires-Dist: psutil<8.0; extra == "pytorch-test"
 Requires-Dist: pytest-cov==7.0.0; extra == "pytorch-test"
 Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-test"
@@ -90,13 +81,24 @@ Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-test"
 Requires-Dist: pytest==9.0.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 != "Darwin" and python_version >= "3.12") and extra == "pytorch-test"
 Requires-Dist: uvicorn; extra == "pytorch-test"
 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-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-test-gpu
+Requires-Dist: torch-tensorrt; (platform_system != "Darwin" and python_version >= "3.12") and extra == "pytorch-test-gpu"
 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"
@@ -108,7 +110,7 @@ Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra
 Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-dev"
 Requires-Dist: click==8.3.1; python_version > "3.10" and extra == "fabric-dev"
 Requires-Dist: coverage==7.10.7; python_version < "3.10" and extra == "fabric-dev"
-Requires-Dist: coverage==7.13.1; python_version >= "3.10" and extra == "fabric-dev"
+Requires-Dist: coverage==7.13.4; python_version >= "3.10" and 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: huggingface-hub; extra == "fabric-dev"
 Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-dev"
@@ -140,7 +142,7 @@ 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; python_version < "3.10" and extra == "pytorch-dev"
-Requires-Dist: coverage==7.13.1; python_version >= "3.10" and extra == "pytorch-dev"
+Requires-Dist: coverage==7.13.4; python_version >= "3.10" and 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: huggingface-hub; extra == "pytorch-dev"
@@ -151,10 +153,11 @@ Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
 Requires-Dist: numpy<2.0,>1.21.0; python_version < "3.12" and extra == "pytorch-dev"
 Requires-Dist: numpy<3.0,>2.1.0; python_version >= "3.12" and extra == "pytorch-dev"
 Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-dev"
+Requires-Dist: onnx-ir<1.0; 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: pandas<4.0,>2.0; extra == "pytorch-dev"
 Requires-Dist: psutil<8.0; extra == "pytorch-dev"
 Requires-Dist: pytest-cov==7.0.0; extra == "pytorch-dev"
 Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-dev"
@@ -167,35 +170,24 @@ 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 != "Darwin" 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: 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
-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: test
 Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "test"
 Requires-Dist: click==8.3.1; python_version > "3.10" and extra == "test"
 Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
 Requires-Dist: coverage==7.10.7; python_version < "3.10" and extra == "test"
-Requires-Dist: coverage==7.13.1; python_version >= "3.10" and extra == "test"
+Requires-Dist: coverage==7.13.4; python_version >= "3.10" and extra == "test"
 Requires-Dist: fastapi; extra == "test"
 Requires-Dist: huggingface-hub; extra == "test"
 Requires-Dist: numpy<2.0,>1.21.0; python_version < "3.12" and extra == "test"
 Requires-Dist: numpy<3.0,>2.1.0; python_version >= "3.12" and extra == "test"
+Requires-Dist: onnx-ir<1.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: pandas<4.0,>2.0; extra == "test"
 Requires-Dist: psutil<8.0; extra == "test"
 Requires-Dist: pytest-cov==7.0.0; extra == "test"
 Requires-Dist: pytest-random-order==1.2.0; extra == "test"
@@ -206,13 +198,25 @@ Requires-Dist: pytest==9.0.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 != "Darwin" and python_version >= "3.12") and extra == "test"
 Requires-Dist: uvicorn; extra == "test"
 Provides-Extra: examples
 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: 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
+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: test-gpu
+Requires-Dist: torch-tensorrt; (platform_system != "Darwin" and python_version >= "3.12") and extra == "test-gpu"
 Provides-Extra: data
 Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
 Provides-Extra: all
@@ -234,7 +238,7 @@ Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "dev"
 Requires-Dist: click==8.3.1; python_version > "3.10" and extra == "dev"
 Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
 Requires-Dist: coverage==7.10.7; python_version < "3.10" and extra == "dev"
-Requires-Dist: coverage==7.13.1; python_version >= "3.10" and extra == "dev"
+Requires-Dist: coverage==7.13.4; python_version >= "3.10" and 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: huggingface-hub; extra == "dev"
@@ -245,10 +249,11 @@ Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
 Requires-Dist: numpy<2.0,>1.21.0; python_version < "3.12" and extra == "dev"
 Requires-Dist: numpy<3.0,>2.1.0; python_version >= "3.12" and extra == "dev"
 Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "dev"
+Requires-Dist: onnx-ir<1.0; 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: pandas<4.0,>2.0; extra == "dev"
 Requires-Dist: psutil<8.0; extra == "dev"
 Requires-Dist: pytest-cov==7.0.0; extra == "dev"
 Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
@@ -262,7 +267,6 @@ 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 != "Darwin" 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"
@@ -302,7 +306,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.1.dev20260201">Docs</a>
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.6.4">Docs</a>
 </p>
 
 <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -535,9 +539,6 @@ trainer = Trainer(logger=loggers.CometLogger())
 # mlflow
 trainer = Trainer(logger=loggers.MLFlowLogger())
 
-# neptune
-trainer = Trainer(logger=loggers.NeptuneLogger())
-
 # ... and dozens more
 ```
 
@@ -843,7 +844,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.1.dev20260201)](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.4)](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)                 |

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/README.md

@@ -292,9 +292,6 @@ trainer = Trainer(logger=loggers.CometLogger())
 # mlflow
 trainer = Trainer(logger=loggers.MLFlowLogger())
 
-# neptune
-trainer = Trainer(logger=loggers.NeptuneLogger())
-
 # ... and dozens more
 ```
 

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/requirements/fabric/test.txt

@@ -1,4 +1,4 @@
-coverage ==7.13.1; python_version >= "3.10"
+coverage ==7.13.4; python_version >= "3.10"
 coverage ==7.10.7; python_version < "3.10"
 numpy >1.21.0, <1.27.0; python_version < "3.12"
 numpy >2.1.0, <2.4.2; python_version >= "3.12"

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/requirements/pytorch/extra.txt

@@ -6,6 +6,6 @@ matplotlib>3.1, <3.11.0
 omegaconf >=2.2.3, <2.4.0
 hydra-core >=1.2.0, <1.4.0
 jsonargparse[signatures,jsonnet] >=4.39.0, <4.46.0
-rich >=12.3.0, <14.3.0
+rich >=12.3.0, <14.4.0
 tensorboardX >=2.2, <2.7.0  # min version is set by torch.onnx missing attribute
 bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/requirements/pytorch/test.txt

@@ -1,4 +1,4 @@
-coverage ==7.13.1; python_version >= "3.10"
+coverage ==7.13.4; python_version >= "3.10"
 coverage ==7.10.7; python_version < "3.10"
 pytest ==9.0.2
 pytest-cov ==7.0.0
@@ -15,12 +15,12 @@ numpy >2.1.0, <2.4.2; python_version >= "3.12"
 onnx >1.12.0, <1.21.0
 onnxruntime >=1.12.0, <1.24.0
 onnxscript >= 0.1.0, < 0.6.0
+onnx-ir < 0.1.16
 psutil <7.3.0 # for `DeviceStatsMonitor`
-pandas >2.0, <2.4.0  # needed in benchmarks
+pandas >2.0, <3.1.0  # needed in benchmarks
 fastapi  # for `ServableModuleValidator`  # not setting version as re-defined in App
 uvicorn  # for `ServableModuleValidator`  # not setting version as re-defined in App
 
 tensorboard >=2.11, <2.21.0  # for `TensorBoardLogger`
 
-torch-tensorrt; platform_system != "Darwin" and python_version >= "3.12"
 huggingface-hub

lightning-2.6.4/requirements/pytorch/test_gpu.txt

@@ -0,0 +1 @@
+torch-tensorrt; platform_system != "Darwin" and python_version >= "3.12"

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/fabric/CHANGELOG.md

@@ -4,27 +4,33 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
-
 ## [unreleased] - YYYY-MM-DD
 
 ### Added
 
 -
 
+### Changed
+
+-
 
 ### Removed
 
 -
 
-
-### Changed
+### Fixed
 
 -
 
+--
+
+## [2.6.2] - 2026-03-19
 
 ### Fixed
 
--
+- Fixed FSDP mixed precision (`bf16-mixed`, `16-mixed`) initializing model parameters in half precision instead of fp32 ([#21586](https://github.com/Lightning-AI/pytorch-lightning/pull/21586))
+
+- Fixed `device_mesh` type hint in `FSDPStrategy` to accept a 2-element tuple via the CLI ([#21581](https://github.com/Lightning-AI/pytorch-lightning/pull/21581))
 
 ---
 

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/fabric/plugins/precision/fsdp.py

@@ -112,7 +112,7 @@ class FSDPPrecision(Precision):
 
     @override
     def module_init_context(self) -> AbstractContextManager:
-        return _DtypeContextManager(self.mixed_precision_config.param_dtype or torch.float32)
+        return _DtypeContextManager(self._desired_input_dtype)
 
     @override
     def forward_context(self) -> AbstractContextManager:

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/fabric/strategies/fsdp.py

@@ -150,7 +150,7 @@ class FSDPStrategy(ParallelStrategy, _Sharded):
         activation_checkpointing_policy: Optional["_POLICY"] = None,
         sharding_strategy: "_SHARDING_STRATEGY" = "FULL_SHARD",
         state_dict_type: Literal["full", "sharded"] = "sharded",
-        device_mesh: Optional[Union[tuple[int], "DeviceMesh"]] = None,
+        device_mesh: Optional[Union[tuple[int, int], "DeviceMesh"]] = None,
         **kwargs: Any,
     ) -> None:
         super().__init__(

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/fabric/wrappers.py

@@ -117,7 +117,7 @@ class _FabricModule(_DeviceDtypeModuleMixin):
         """
         super().__init__()
         self._forward_module = forward_module
-        self._original_module = original_module or forward_module
+        self._original_module = forward_module if original_module is None else original_module
         self._strategy = strategy
         self._forward_methods = set(_LIGHTNING_MODULE_STEP_METHODS)
         self._fabric_module_initialized = True

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/pytorch/CHANGELOG.md

@@ -12,21 +12,44 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 -
 
+### Changed
+
+-
 
 ### Removed
 
 -
 
+### Fixed
+
+-
+
+---
+
+## [2.6.2] - 2026-03-19
+
+
+### Removed
+
+- Removed support for Neptune logger ([#21572](https://github.com/Lightning-AI/pytorch-lightning/pull/21572))
+
 
 ### Changed
 
--
+- Raise minimum `LitLogger` version to `2026-03-17` ([#21591](https://github.com/Lightning-AI/pytorch-lightning/pull/21591))
 
 
 ### Fixed
 
--
+- Fixed `val_check_interval` raising `ValueError` when `limit_val_batches=0` and interval exceeds training batches ([#21560](https://github.com/Lightning-AI/pytorch-lightning/pull/21560))
+
+- Fixed pkg-resources deprecation issue ([#21538](https://github.com/Lightning-AI/pytorch-lightning/pull/21538))
+
+- Fixed FSDP mixed precision (`bf16-mixed`, `16-mixed`) initializing model parameters in half precision instead of fp32 ([#21586](https://github.com/Lightning-AI/pytorch-lightning/pull/21586))
+
+- Fixed `device_mesh` type hint in `FSDPStrategy` to accept a 2-element tuple via the CLI ([#21581](https://github.com/Lightning-AI/pytorch-lightning/pull/21581))
 
+- Fixed ``RichModelSummary`` model size display formatting ([#21467](https://github.com/Lightning-AI/pytorch-lightning/pull/21467))
 
 ---
 

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/pytorch/callbacks/model_checkpoint.py

@@ -229,7 +229,7 @@ class ModelCheckpoint(Checkpoint):
         ... )
 
         # save epoch and val_loss in name, but specify the formatting yourself (e.g. to avoid problems with Tensorboard
-        # or Neptune, due to the presence of characters like '=' or '/')
+        # or other loggers, due to the presence of characters like '=' or '/')
         # saves a file like: my/path/sample-mnist-epoch02-val_loss0.32.ckpt
         >>> checkpoint_callback = ModelCheckpoint(
         ...     monitor='val/loss',

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/pytorch/callbacks/rich_model_summary.py

@@ -17,7 +17,7 @@ from typing_extensions import override
 
 from lightning.pytorch.callbacks import ModelSummary
 from lightning.pytorch.utilities.imports import _RICH_AVAILABLE
-from lightning.pytorch.utilities.model_summary import get_human_readable_count
+from lightning.pytorch.utilities.model_summary import get_formatted_model_size, get_human_readable_count
 
 
 class RichModelSummary(ModelSummary):
@@ -105,8 +105,9 @@ class RichModelSummary(ModelSummary):
         console.print(table)
 
         parameters = []
-        for param in [trainable_parameters, total_parameters - trainable_parameters, total_parameters, model_size]:
+        for param in [trainable_parameters, total_parameters - trainable_parameters, total_parameters]:
             parameters.append("{:<{}}".format(get_human_readable_count(int(param)), 10))
+        parameters.append("{:<{}}".format(get_formatted_model_size(model_size), 10))
 
         grid = Table.grid(expand=True)
         grid.add_column()

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/pytorch/core/module.py

@@ -1498,11 +1498,10 @@ class LightningModule(
         example_inputs: Optional[Any] = None,
         **kwargs: Any,
     ) -> Union[ScriptModule, dict[str, ScriptModule]]:
-        """By default compiles the whole model to a :class:`~torch.jit.ScriptModule`. If you want to use tracing,
-        please provided the argument ``method='trace'`` and make sure that either the `example_inputs` argument is
-        provided, or the model has :attr:`example_input_array` set. If you would like to customize the modules that are
-        scripted you should override this method. In case you want to return multiple modules, we recommend using a
-        dictionary.
+        """By default compiles the whole model to a ``torch.jit.ScriptModule``. If you want to use tracing, please
+        provided the argument ``method='trace'`` and make sure that either the `example_inputs` argument is provided,
+        or the model has :attr:`example_input_array` set. If you would like to customize the modules that are scripted
+        you should override this method. In case you want to return multiple modules, we recommend using a dictionary.
 
         .. deprecated::
             ``LightningModule.to_torchscript`` has been deprecated in v2.7 and will be removed in v2.8.
@@ -1514,15 +1513,15 @@ class LightningModule(
             method: Whether to use TorchScript's script or trace method. Default: 'script'
             example_inputs: An input to be used to do tracing when method is set to 'trace'.
               Default: None (uses :attr:`example_input_array`)
-            **kwargs: Additional arguments that will be passed to the :func:`torch.jit.script` or
-              :func:`torch.jit.trace` function.
+            **kwargs: Additional arguments that will be passed to the ``torch.jit.script`` or
+              ``torch.jit.trace`` function.
 
         Note:
             - Requires the implementation of the
               :meth:`~lightning.pytorch.core.LightningModule.forward` method.
             - The exported script will be set to evaluation mode.
             - It is recommended that you install the latest supported version of PyTorch
-              to use this feature without limitations. See also the :mod:`torch.jit`
+              to use this feature without limitations. See also the ``torch.jit``
               documentation for supported features.
 
         Example::

{lightning-2.6.1.dev20260201 → lightning-2.6.4}/src/lightning/pytorch/loggers/__init__.py

@@ -16,7 +16,6 @@ from lightning.pytorch.loggers.csv_logs import CSVLogger
 from lightning.pytorch.loggers.litlogger import LitLogger
 from lightning.pytorch.loggers.logger import Logger
 from lightning.pytorch.loggers.mlflow import MLFlowLogger
-from lightning.pytorch.loggers.neptune import NeptuneLogger
 from lightning.pytorch.loggers.tensorboard import TensorBoardLogger
 from lightning.pytorch.loggers.wandb import WandbLogger
 
@@ -28,5 +27,4 @@ __all__ = [
     "MLFlowLogger",
     "TensorBoardLogger",
     "WandbLogger",
-    "NeptuneLogger",
 ]