lightning 2.5.6__tar.gz → 2.6.0.dev0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (303) hide show
  1. {lightning-2.5.6/src/lightning.egg-info → lightning-2.6.0.dev0}/PKG-INFO +78 -57
  2. {lightning-2.5.6 → lightning-2.6.0.dev0}/README.md +4 -3
  3. {lightning-2.5.6 → lightning-2.6.0.dev0}/pyproject.toml +0 -3
  4. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/base.txt +1 -1
  5. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/examples.txt +1 -1
  6. lightning-2.6.0.dev0/requirements/fabric/extra.txt +4 -0
  7. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/strategies.txt +1 -1
  8. lightning-2.6.0.dev0/requirements/fabric/test.txt +12 -0
  9. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/base.txt +1 -1
  10. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/docs.txt +1 -1
  11. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/examples.txt +2 -2
  12. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/extra.txt +1 -1
  13. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/strategies.txt +1 -1
  14. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/test.txt +10 -6
  15. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__setup__.py +1 -1
  16. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/CHANGELOG.md +27 -0
  17. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/fabric.py +199 -49
  18. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/lightning.py +28 -6
  19. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/deepspeed.py +30 -16
  20. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/fsdp.py +4 -3
  21. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/strategy.py +7 -4
  22. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/xla_fsdp.py +3 -2
  23. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/device_dtype_mixin.py +5 -1
  24. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/imports.py +1 -0
  25. lightning-2.6.0.dev0/src/lightning/fabric/utilities/port_manager.py +233 -0
  26. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/testing/_runif.py +1 -1
  27. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/CHANGELOG.md +56 -4
  28. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/__init__.py +2 -0
  29. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/early_stopping.py +32 -0
  30. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/lr_monitor.py +1 -1
  31. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/model_checkpoint.py +12 -6
  32. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/model_summary.py +6 -0
  33. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/rich_progress.py +28 -20
  34. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +4 -4
  35. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/pruning.py +3 -2
  36. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/rich_model_summary.py +4 -1
  37. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +1 -1
  38. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/throughput_monitor.py +25 -7
  39. lightning-2.6.0.dev0/src/lightning/pytorch/callbacks/weight_averaging.py +363 -0
  40. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/cli.py +12 -0
  41. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/module.py +115 -2
  42. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/mlflow.py +2 -4
  43. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/wandb.py +4 -1
  44. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/evaluation_loop.py +5 -1
  45. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/fit_loop.py +9 -2
  46. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/training_epoch_loop.py +8 -0
  47. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/profiler.py +4 -1
  48. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/ddp.py +22 -0
  49. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/deepspeed.py +15 -0
  50. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/xla.py +4 -1
  51. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +1 -1
  52. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/callback_connector.py +4 -9
  53. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/data_connector.py +4 -3
  54. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +1 -2
  55. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/setup.py +73 -2
  56. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/trainer.py +22 -5
  57. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/batch_size_scaling.py +16 -2
  58. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/deepspeed.py +2 -2
  59. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/imports.py +1 -0
  60. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_helpers.py +11 -5
  61. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_summary/model_summary.py +63 -13
  62. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/testing/_runif.py +7 -3
  63. lightning-2.6.0.dev0/src/lightning/version.info +1 -0
  64. {lightning-2.5.6 → lightning-2.6.0.dev0/src/lightning.egg-info}/PKG-INFO +78 -57
  65. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/SOURCES.txt +4 -1
  66. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/requires.txt +96 -43
  67. lightning-2.6.0.dev0/src/version.info +1 -0
  68. lightning-2.5.6/requirements/fabric/test.txt +0 -10
  69. lightning-2.5.6/src/lightning/version.info +0 -1
  70. lightning-2.5.6/src/version.info +0 -1
  71. {lightning-2.5.6 → lightning-2.6.0.dev0}/.actions/assistant.py +0 -0
  72. {lightning-2.5.6 → lightning-2.6.0.dev0}/CITATION.cff +0 -0
  73. {lightning-2.5.6 → lightning-2.6.0.dev0}/LICENSE +0 -0
  74. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/base.txt +0 -0
  75. {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/docs.txt +0 -0
  76. {lightning-2.5.6 → lightning-2.6.0.dev0}/setup.cfg +0 -0
  77. {lightning-2.5.6 → lightning-2.6.0.dev0}/setup.py +0 -0
  78. /lightning-2.5.6/src/lightning/d42c98d5f7b8f5e086a66f872d1e2133132dc322fc7ce9dd → /lightning-2.6.0.dev0/src/lightning/36e4fb47a659f3f6d57e62c858555053c62817fcd39048c0 +0 -0
  79. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__about__.py +0 -0
  80. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__init__.py +0 -0
  81. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__version__.py +0 -0
  82. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/data/__init__.py +0 -0
  83. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/__init__.py +0 -0
  84. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/_graveyard/__init__.py +0 -0
  85. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/_graveyard/tpu.py +0 -0
  86. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/__init__.py +0 -0
  87. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/accelerator.py +0 -0
  88. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/cpu.py +0 -0
  89. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/cuda.py +0 -0
  90. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/mps.py +0 -0
  91. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/registry.py +0 -0
  92. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/xla.py +0 -0
  93. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/cli.py +0 -0
  94. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/connector.py +0 -0
  95. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/__init__.py +0 -0
  96. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/csv_logs.py +0 -0
  97. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/logger.py +0 -0
  98. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/tensorboard.py +0 -0
  99. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/__init__.py +0 -0
  100. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
  101. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
  102. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/single_device.py +0 -0
  103. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/torch_collective.py +0 -0
  104. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
  105. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
  106. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
  107. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/lsf.py +0 -0
  108. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
  109. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
  110. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
  111. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/xla.py +0 -0
  112. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/__init__.py +0 -0
  113. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
  114. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/torch_io.py +0 -0
  115. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/xla.py +0 -0
  116. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
  117. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/amp.py +0 -0
  118. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/bitsandbytes.py +0 -0
  119. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/deepspeed.py +0 -0
  120. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/double.py +0 -0
  121. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/fsdp.py +0 -0
  122. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/half.py +0 -0
  123. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/precision.py +0 -0
  124. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/transformer_engine.py +0 -0
  125. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/utils.py +0 -0
  126. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/xla.py +0 -0
  127. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/__init__.py +0 -0
  128. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/ddp.py +0 -0
  129. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/dp.py +0 -0
  130. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
  131. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
  132. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/multiprocessing.py +0 -0
  133. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/subprocess_script.py +0 -0
  134. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
  135. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/model_parallel.py +0 -0
  136. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/parallel.py +0 -0
  137. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/registry.py +0 -0
  138. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/single_device.py +0 -0
  139. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/single_xla.py +0 -0
  140. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/xla.py +0 -0
  141. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/__init__.py +0 -0
  142. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/apply_func.py +0 -0
  143. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/cloud_io.py +0 -0
  144. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/consolidate_checkpoint.py +0 -0
  145. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/data.py +0 -0
  146. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/device_parser.py +0 -0
  147. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/distributed.py +0 -0
  148. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/enums.py +0 -0
  149. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/exceptions.py +0 -0
  150. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/init.py +0 -0
  151. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/load.py +0 -0
  152. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/logger.py +0 -0
  153. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/optimizer.py +0 -0
  154. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/rank_zero.py +0 -0
  155. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/registry.py +0 -0
  156. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/seed.py +0 -0
  157. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/spike.py +0 -0
  158. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
  159. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/throughput.py +0 -0
  160. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/types.py +0 -0
  161. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/warnings.py +0 -0
  162. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/wrappers.py +0 -0
  163. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/py.typed +0 -0
  164. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/__init__.py +0 -0
  165. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
  166. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
  167. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
  168. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/precision.py +0 -0
  169. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
  170. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/__init__.py +0 -0
  171. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
  172. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/cpu.py +0 -0
  173. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/cuda.py +0 -0
  174. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/mps.py +0 -0
  175. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/xla.py +0 -0
  176. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
  177. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/callback.py +0 -0
  178. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
  179. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/device_stats_monitor.py +0 -0
  180. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/finetuning.py +0 -0
  181. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +0 -0
  182. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
  183. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/lr_finder.py +0 -0
  184. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
  185. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/prediction_writer.py +0 -0
  186. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
  187. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/progress_bar.py +0 -0
  188. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/spike.py +0 -0
  189. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/timer.py +0 -0
  190. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/__init__.py +0 -0
  191. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/datamodule.py +0 -0
  192. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/hooks.py +0 -0
  193. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
  194. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/mixins/hparams_mixin.py +0 -0
  195. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/optimizer.py +0 -0
  196. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/saving.py +0 -0
  197. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/__init__.py +0 -0
  198. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/boring_classes.py +0 -0
  199. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/lstm.py +0 -0
  200. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
  201. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/transformer.py +0 -0
  202. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/__init__.py +0 -0
  203. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/comet.py +0 -0
  204. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/csv_logs.py +0 -0
  205. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/logger.py +0 -0
  206. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/neptune.py +0 -0
  207. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/tensorboard.py +0 -0
  208. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/utilities.py +0 -0
  209. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/__init__.py +0 -0
  210. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/fetchers.py +0 -0
  211. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/loop.py +0 -0
  212. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
  213. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/automatic.py +0 -0
  214. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
  215. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/manual.py +0 -0
  216. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/prediction_loop.py +0 -0
  217. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/progress.py +0 -0
  218. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/utilities.py +0 -0
  219. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/overrides/__init__.py +0 -0
  220. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/overrides/distributed.py +0 -0
  221. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/__init__.py +0 -0
  222. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
  223. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
  224. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/async_plugin.py +0 -0
  225. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
  226. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
  227. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/wrapper.py +0 -0
  228. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
  229. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
  230. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
  231. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/amp.py +0 -0
  232. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +0 -0
  233. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/deepspeed.py +0 -0
  234. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/double.py +0 -0
  235. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/fsdp.py +0 -0
  236. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/half.py +0 -0
  237. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/precision.py +0 -0
  238. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
  239. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
  240. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/__init__.py +0 -0
  241. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/advanced.py +0 -0
  242. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/base.py +0 -0
  243. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/pytorch.py +0 -0
  244. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/simple.py +0 -0
  245. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/xla.py +0 -0
  246. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/serve/__init__.py +0 -0
  247. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/serve/servable_module.py +0 -0
  248. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/serve/servable_module_validator.py +0 -0
  249. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/__init__.py +0 -0
  250. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/fsdp.py +0 -0
  251. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
  252. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
  253. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +0 -0
  254. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +0 -0
  255. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
  256. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/model_parallel.py +0 -0
  257. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/parallel.py +0 -0
  258. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/single_device.py +0 -0
  259. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/single_xla.py +0 -0
  260. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/strategy.py +0 -0
  261. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/__init__.py +0 -0
  262. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/call.py +0 -0
  263. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
  264. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
  265. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +0 -0
  266. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
  267. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
  268. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +0 -0
  269. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/signal_connector.py +0 -0
  270. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/states.py +0 -0
  271. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/__init__.py +0 -0
  272. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/lr_finder.py +0 -0
  273. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/tuning.py +0 -0
  274. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/__init__.py +0 -0
  275. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/_pytree.py +0 -0
  276. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/argparse.py +0 -0
  277. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/combined_loader.py +0 -0
  278. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/compile.py +0 -0
  279. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +0 -0
  280. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/data.py +0 -0
  281. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/enums.py +0 -0
  282. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/exceptions.py +0 -0
  283. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/grads.py +0 -0
  284. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/memory.py +0 -0
  285. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
  286. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
  287. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/migration/utils.py +0 -0
  288. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_registry.py +0 -0
  289. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
  290. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +0 -0
  291. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
  292. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/parsing.py +0 -0
  293. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
  294. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/seed.py +0 -0
  295. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
  296. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
  297. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/types.py +0 -0
  298. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +0 -0
  299. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/warnings.py +0 -0
  300. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/dependency_links.txt +0 -0
  301. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/entry_points.txt +0 -0
  302. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/not-zip-safe +0 -0
  303. {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lightning
3
- Version: 2.5.6
3
+ Version: 2.6.0.dev0
4
4
  Summary: The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
5
5
  Home-page: https://github.com/Lightning-AI/lightning
6
6
  Download-URL: https://github.com/Lightning-AI/lightning
@@ -36,22 +36,26 @@ Requires-Dist: torchmetrics<3.0,>0.7.0
36
36
  Requires-Dist: tqdm<6.0,>=4.57.0
37
37
  Requires-Dist: typing-extensions<6.0,>4.5.0
38
38
  Requires-Dist: pytorch-lightning
39
+ Provides-Extra: fabric-extra
40
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-extra"
39
41
  Provides-Extra: fabric-strategies
40
42
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-strategies"
41
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
43
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
42
44
  Provides-Extra: fabric-examples
43
45
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-examples"
44
46
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-examples"
45
47
  Provides-Extra: fabric-test
46
48
  Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-test"
47
- Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-test"
48
- Requires-Dist: coverage==7.10.6; extra == "fabric-test"
49
+ Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "fabric-test"
50
+ Requires-Dist: coverage==7.10.7; extra == "fabric-test"
51
+ Requires-Dist: huggingface-hub; extra == "fabric-test"
49
52
  Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-test"
50
- Requires-Dist: pytest-cov==6.2.1; extra == "fabric-test"
53
+ Requires-Dist: pytest-cov==6.3.0; extra == "fabric-test"
51
54
  Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-test"
52
- Requires-Dist: pytest-rerunfailures==16.0; extra == "fabric-test"
55
+ Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "fabric-test"
56
+ Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "fabric-test"
53
57
  Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-test"
54
- Requires-Dist: pytest==8.4.1; extra == "fabric-test"
58
+ Requires-Dist: pytest==8.4.2; extra == "fabric-test"
55
59
  Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-test"
56
60
  Provides-Extra: pytorch-extra
57
61
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
@@ -62,55 +66,62 @@ Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
62
66
  Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
63
67
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
64
68
  Provides-Extra: pytorch-strategies
65
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
69
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
66
70
  Provides-Extra: pytorch-examples
67
- Requires-Dist: ipython[all]<10.0,>=8.0.0; extra == "pytorch-examples"
71
+ Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "pytorch-examples"
68
72
  Requires-Dist: requests<3.0; extra == "pytorch-examples"
69
73
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-examples"
70
74
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-examples"
71
75
  Provides-Extra: pytorch-test
72
76
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-test"
73
- Requires-Dist: coverage==7.10.6; extra == "pytorch-test"
77
+ Requires-Dist: coverage==7.10.7; extra == "pytorch-test"
74
78
  Requires-Dist: fastapi; extra == "pytorch-test"
79
+ Requires-Dist: huggingface-hub; extra == "pytorch-test"
75
80
  Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-test"
76
81
  Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-test"
77
82
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-test"
78
83
  Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-test"
79
84
  Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-test"
80
85
  Requires-Dist: psutil<8.0; extra == "pytorch-test"
81
- Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-test"
86
+ Requires-Dist: pytest-cov==6.3.0; extra == "pytorch-test"
82
87
  Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-test"
83
- Requires-Dist: pytest-rerunfailures==16.0; extra == "pytorch-test"
88
+ Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "pytorch-test"
89
+ Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "pytorch-test"
84
90
  Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-test"
85
- Requires-Dist: pytest==8.4.1; extra == "pytorch-test"
91
+ Requires-Dist: pytest==8.4.2; extra == "pytorch-test"
86
92
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-test"
87
93
  Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-test"
94
+ Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "pytorch-test"
88
95
  Requires-Dist: uvicorn; extra == "pytorch-test"
89
96
  Provides-Extra: fabric-all
90
97
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-all"
91
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
98
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
99
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-all"
92
100
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-all"
93
101
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-all"
94
102
  Provides-Extra: fabric-dev
95
103
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-dev"
96
104
  Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-dev"
97
- Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-dev"
98
- Requires-Dist: coverage==7.10.6; extra == "fabric-dev"
99
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
105
+ Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "fabric-dev"
106
+ Requires-Dist: coverage==7.10.7; extra == "fabric-dev"
107
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
108
+ Requires-Dist: huggingface-hub; extra == "fabric-dev"
109
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "fabric-dev"
100
110
  Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-dev"
101
- Requires-Dist: pytest-cov==6.2.1; extra == "fabric-dev"
111
+ Requires-Dist: pytest-cov==6.3.0; extra == "fabric-dev"
102
112
  Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-dev"
103
- Requires-Dist: pytest-rerunfailures==16.0; extra == "fabric-dev"
113
+ Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "fabric-dev"
114
+ Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "fabric-dev"
104
115
  Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-dev"
105
- Requires-Dist: pytest==8.4.1; extra == "fabric-dev"
116
+ Requires-Dist: pytest==8.4.2; extra == "fabric-dev"
106
117
  Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-dev"
107
118
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-dev"
108
119
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-dev"
109
120
  Provides-Extra: pytorch-all
110
121
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-all"
111
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
122
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
112
123
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-all"
113
- Requires-Dist: ipython[all]<10.0,>=8.0.0; extra == "pytorch-all"
124
+ Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "pytorch-all"
114
125
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-all"
115
126
  Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-all"
116
127
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-all"
@@ -122,11 +133,12 @@ Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-all"
122
133
  Provides-Extra: pytorch-dev
123
134
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-dev"
124
135
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-dev"
125
- Requires-Dist: coverage==7.10.6; extra == "pytorch-dev"
126
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
136
+ Requires-Dist: coverage==7.10.7; extra == "pytorch-dev"
137
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
127
138
  Requires-Dist: fastapi; extra == "pytorch-dev"
139
+ Requires-Dist: huggingface-hub; extra == "pytorch-dev"
128
140
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-dev"
129
- Requires-Dist: ipython[all]<10.0,>=8.0.0; extra == "pytorch-dev"
141
+ Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "pytorch-dev"
130
142
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-dev"
131
143
  Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
132
144
  Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-dev"
@@ -136,35 +148,46 @@ Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-dev"
136
148
  Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-dev"
137
149
  Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-dev"
138
150
  Requires-Dist: psutil<8.0; extra == "pytorch-dev"
139
- Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-dev"
151
+ Requires-Dist: pytest-cov==6.3.0; extra == "pytorch-dev"
140
152
  Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-dev"
141
- Requires-Dist: pytest-rerunfailures==16.0; extra == "pytorch-dev"
153
+ Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "pytorch-dev"
154
+ Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "pytorch-dev"
142
155
  Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-dev"
143
- Requires-Dist: pytest==8.4.1; extra == "pytorch-dev"
156
+ Requires-Dist: pytest==8.4.2; extra == "pytorch-dev"
144
157
  Requires-Dist: requests<3.0; extra == "pytorch-dev"
145
158
  Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-dev"
146
159
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-dev"
147
160
  Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-dev"
148
161
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-dev"
162
+ Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "pytorch-dev"
149
163
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-dev"
150
164
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-dev"
151
165
  Requires-Dist: uvicorn; extra == "pytorch-dev"
152
166
  Provides-Extra: data
153
167
  Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
168
+ Provides-Extra: extra
169
+ Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
170
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
171
+ Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
172
+ Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
173
+ Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
174
+ Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
175
+ Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
154
176
  Provides-Extra: strategies
155
177
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "strategies"
156
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
178
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
157
179
  Provides-Extra: examples
158
- Requires-Dist: ipython[all]<10.0,>=8.0.0; extra == "examples"
180
+ Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "examples"
159
181
  Requires-Dist: requests<3.0; extra == "examples"
160
182
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "examples"
161
183
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "examples"
162
184
  Provides-Extra: test
163
185
  Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "test"
164
- Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "test"
186
+ Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "test"
165
187
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
166
- Requires-Dist: coverage==7.10.6; extra == "test"
188
+ Requires-Dist: coverage==7.10.7; extra == "test"
167
189
  Requires-Dist: fastapi; extra == "test"
190
+ Requires-Dist: huggingface-hub; extra == "test"
168
191
  Requires-Dist: numpy<2.0,>1.20.0; extra == "test"
169
192
  Requires-Dist: numpy<2.0,>=1.21.0; extra == "test"
170
193
  Requires-Dist: onnx<2.0,>1.12.0; extra == "test"
@@ -172,28 +195,22 @@ Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "test"
172
195
  Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "test"
173
196
  Requires-Dist: pandas<3.0,>2.0; extra == "test"
174
197
  Requires-Dist: psutil<8.0; extra == "test"
175
- Requires-Dist: pytest-cov==6.2.1; extra == "test"
198
+ Requires-Dist: pytest-cov==6.3.0; extra == "test"
176
199
  Requires-Dist: pytest-random-order==1.2.0; extra == "test"
177
- Requires-Dist: pytest-rerunfailures==16.0; extra == "test"
200
+ Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "test"
201
+ Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "test"
178
202
  Requires-Dist: pytest-timeout==2.4.0; extra == "test"
179
- Requires-Dist: pytest==8.4.1; extra == "test"
203
+ Requires-Dist: pytest==8.4.2; extra == "test"
180
204
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "test"
181
205
  Requires-Dist: tensorboard<3.0,>=2.11; extra == "test"
182
206
  Requires-Dist: tensorboardX<3.0,>=2.6; extra == "test"
207
+ Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "test"
183
208
  Requires-Dist: uvicorn; extra == "test"
184
- Provides-Extra: extra
185
- Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
186
- Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
187
- Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
188
- Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
189
- Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
190
- Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
191
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
192
209
  Provides-Extra: all
193
210
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "all"
194
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
211
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
195
212
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "all"
196
- Requires-Dist: ipython[all]<10.0,>=8.0.0; extra == "all"
213
+ Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "all"
197
214
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "all"
198
215
  Requires-Dist: matplotlib<4.0,>3.1; extra == "all"
199
216
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "all"
@@ -205,13 +222,14 @@ Requires-Dist: torchvision<1.0,>=0.16.0; extra == "all"
205
222
  Provides-Extra: dev
206
223
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "dev"
207
224
  Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "dev"
208
- Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "dev"
225
+ Requires-Dist: click==8.3.0; python_version > "3.10" and extra == "dev"
209
226
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
210
- Requires-Dist: coverage==7.10.6; extra == "dev"
211
- Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
227
+ Requires-Dist: coverage==7.10.7; extra == "dev"
228
+ Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
212
229
  Requires-Dist: fastapi; extra == "dev"
230
+ Requires-Dist: huggingface-hub; extra == "dev"
213
231
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "dev"
214
- Requires-Dist: ipython[all]<10.0,>=8.0.0; extra == "dev"
232
+ Requires-Dist: ipython[all]<11.0,>=8.0.0; extra == "dev"
215
233
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "dev"
216
234
  Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
217
235
  Requires-Dist: numpy<2.0,>1.20.0; extra == "dev"
@@ -222,17 +240,19 @@ Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "dev"
222
240
  Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "dev"
223
241
  Requires-Dist: pandas<3.0,>2.0; extra == "dev"
224
242
  Requires-Dist: psutil<8.0; extra == "dev"
225
- Requires-Dist: pytest-cov==6.2.1; extra == "dev"
243
+ Requires-Dist: pytest-cov==6.3.0; extra == "dev"
226
244
  Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
227
- Requires-Dist: pytest-rerunfailures==16.0; extra == "dev"
245
+ Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "dev"
246
+ Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "dev"
228
247
  Requires-Dist: pytest-timeout==2.4.0; extra == "dev"
229
- Requires-Dist: pytest==8.4.1; extra == "dev"
248
+ Requires-Dist: pytest==8.4.2; extra == "dev"
230
249
  Requires-Dist: requests<3.0; extra == "dev"
231
250
  Requires-Dist: rich<15.0,>=12.3.0; extra == "dev"
232
251
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "dev"
233
252
  Requires-Dist: tensorboard<3.0,>=2.11; extra == "dev"
234
253
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "dev"
235
254
  Requires-Dist: tensorboardX<3.0,>=2.6; extra == "dev"
255
+ Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "dev"
236
256
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "dev"
237
257
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "dev"
238
258
  Requires-Dist: uvicorn; extra == "dev"
@@ -259,7 +279,7 @@ Dynamic: summary
259
279
  <br/>
260
280
  <br/>
261
281
 
262
- **The deep learning framework to pretrai and finetune AI models.**
282
+ **The deep learning framework to pretrain and finetune AI models.**
263
283
 
264
284
  **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
265
285
 
@@ -272,7 +292,7 @@ ______________________________________________________________________
272
292
  <a href="#lightning-fabric-expert-control">Fabric</a> •
273
293
  <a href="https://lightning.ai/?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme">Lightning Cloud</a> •
274
294
  <a href="#community">Community</a> •
275
- <a href="https://pytorch-lightning.readthedocs.io/en/2.5.6">Docs</a>
295
+ <a href="https://pytorch-lightning.readthedocs.io/en/2.6.0dev0">Docs</a>
276
296
  </p>
277
297
 
278
298
  <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -318,7 +338,8 @@ Over 340,000 developers use [Lightning Cloud](https://lightning.ai/?utm_source=p
318
338
  - [Notebooks](https://lightning.ai/notebooks?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme): Persistent GPU workspaces where AI helps you code and analyze.
319
339
  - [Inference](https://lightning.ai/deploy?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme): Deploy models as inference APIs.
320
340
 
321
- # Why PyTorch Lightning
341
+ <a id="why-pytorch-lightning"></a>
342
+ # Why PyTorch Lightning?
322
343
 
323
344
  Training models in plain PyTorch is tedious and error-prone - you have to manually handle things like backprop, mixed precision, multi-GPU, and distributed training, often rewriting code for every new project. PyTorch Lightning organizes PyTorch code to automate those complexities so you can focus on your model and data, while keeping full control and scaling from CPU to multi-node without changing your core code. But if you want control of those things, you can still opt into [expert-level control](#lightning-fabric-expert-control).
324
345
 
@@ -428,7 +449,7 @@ Explore various types of training possible with PyTorch Lightning. Pretrain and
428
449
 
429
450
  | Task | Description | Run |
430
451
  |------|--------------|-----|
431
- | [Hello world](#pytorch-lightning-example) | Pretrain - Hello world example | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
452
+ | [Hello world](https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Pretrain - Hello world example | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
432
453
  | [Image classification](https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - ResNet-34 model to classify images of cars | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
433
454
  | [Image segmentation](https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - ResNet-50 model to segment images | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
434
455
  | [Object detection](https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - Faster R-CNN model to detect objects | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
@@ -814,7 +835,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
814
835
 
815
836
  | System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
816
837
  | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
817
- | 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.5.6)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
838
+ | 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.0dev0)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
818
839
  | 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) |
819
840
  | 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) |
820
841
  | 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) |
@@ -5,7 +5,7 @@
5
5
  <br/>
6
6
  <br/>
7
7
 
8
- **The deep learning framework to pretrai and finetune AI models.**
8
+ **The deep learning framework to pretrain and finetune AI models.**
9
9
 
10
10
  **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
11
11
 
@@ -64,7 +64,8 @@ Over 340,000 developers use [Lightning Cloud](https://lightning.ai/?utm_source=p
64
64
  - [Notebooks](https://lightning.ai/notebooks?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme): Persistent GPU workspaces where AI helps you code and analyze.
65
65
  - [Inference](https://lightning.ai/deploy?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme): Deploy models as inference APIs.
66
66
 
67
- # Why PyTorch Lightning
67
+ <a id="why-pytorch-lightning"></a>
68
+ # Why PyTorch Lightning?
68
69
 
69
70
  Training models in plain PyTorch is tedious and error-prone - you have to manually handle things like backprop, mixed precision, multi-GPU, and distributed training, often rewriting code for every new project. PyTorch Lightning organizes PyTorch code to automate those complexities so you can focus on your model and data, while keeping full control and scaling from CPU to multi-node without changing your core code. But if you want control of those things, you can still opt into [expert-level control](#lightning-fabric-expert-control).
70
71
 
@@ -215,7 +216,7 @@ Explore various types of training possible with PyTorch Lightning. Pretrain and
215
216
 
216
217
  | Task | Description | Run |
217
218
  |------|--------------|-----|
218
- | [Hello world](#pytorch-lightning-example) | Pretrain - Hello world example | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
219
+ | [Hello world](https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Pretrain - Hello world example | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
219
220
  | [Image classification](https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - ResNet-34 model to classify images of cars | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
220
221
  | [Image segmentation](https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - ResNet-50 model to segment images | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
221
222
  | [Object detection](https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - Faster R-CNN model to detect objects | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
@@ -91,7 +91,6 @@ ignore = [
91
91
  ]
92
92
  "src/**" = [
93
93
  "S101", # todo: Use of `assert` detected
94
- "S105", "S106", "S107", # todo: Possible hardcoded password: ...
95
94
  "S113", # todo: Probable use of requests call without timeout
96
95
  "S301", # todo: `pickle` and modules that wrap it can be unsafe when used to deserialize untrusted data, possible security issue
97
96
  "S324", # todo: Probable use of insecure hash functions in `hashlib`
@@ -106,7 +105,6 @@ ignore = [
106
105
  ]
107
106
  "tests/**" = [
108
107
  "S101", # Use of `assert` detected
109
- "S105", "S106", # todo: Possible hardcoded password: ...
110
108
  "S301", # `pickle` and modules that wrap it can be unsafe when used to deserialize untrusted data, possible security issue
111
109
  "S113", # todo: Probable use of requests call without timeout
112
110
  "S311", # todo: Standard pseudo-random generators are not suitable for cryptographic purposes
@@ -120,7 +118,6 @@ ignore = [
120
118
  "S607", # todo: Starting a process with a partial executable path
121
119
  "RET504", # todo:Unnecessary variable assignment before `return` statement
122
120
  "PT004", # todo: Fixture `tmpdir_unittest_fixture` does not return anything, add leading underscore
123
- "PT011", # todo: `pytest.raises(ValueError)` is too broad, set the `match` parameter or use a more specific exception
124
121
  "PT012", # todo: `pytest.raises()` block should contain a single simple statement
125
122
  "PT019", # todo: Fixture `_` without value is injected as parameter, use `@pytest.mark.usefixtures` instead
126
123
  ]
@@ -2,7 +2,7 @@
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
4
  torch >=2.1.0, <2.9.0
5
- fsspec[http] >=2022.5.0, <2025.8.0
5
+ fsspec[http] >=2022.5.0, <2025.11.0
6
6
  packaging >=20.0, <=25.0
7
7
  typing-extensions >4.5.0, <4.16.0
8
8
  lightning-utilities >=0.10.0, <0.16.0
@@ -1,5 +1,5 @@
1
1
  # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
- torchvision >=0.16.0, <0.24.0
4
+ torchvision >=0.16.0, <0.25.0
5
5
  torchmetrics >=0.10.0, <1.9.0
@@ -0,0 +1,4 @@
1
+ # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
2
+ # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
+
4
+ hydra-core >=1.2.0, <1.4.0
@@ -5,5 +5,5 @@
5
5
 
6
6
  # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
7
7
  # shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
8
- deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin" # strict
8
+ deepspeed >=0.15.0,<0.17.0; platform_system != "Windows" and platform_system != "Darwin" # strict
9
9
  bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
@@ -0,0 +1,12 @@
1
+ coverage ==7.10.7
2
+ numpy >=1.21.0, <1.27.0
3
+ pytest ==8.4.2
4
+ pytest-cov ==6.3.0
5
+ pytest-timeout ==2.4.0
6
+ pytest-rerunfailures ==16.0.1; python_version < "3.10"
7
+ pytest-rerunfailures ==16.1; python_version >= "3.10"
8
+ pytest-random-order ==1.2.0
9
+ click ==8.1.8; python_version < "3.11"
10
+ click ==8.3.0; python_version > "3.10"
11
+ tensorboardX >=2.6, <2.7.0 # todo: relax it back to `>=2.2` after fixing tests
12
+ huggingface-hub
@@ -4,7 +4,7 @@
4
4
  torch >=2.1.0, <2.9.0
5
5
  tqdm >=4.57.0, <4.68.0
6
6
  PyYAML >5.4, <6.1.0
7
- fsspec[http] >=2022.5.0, <2025.8.0
7
+ fsspec[http] >=2022.5.0, <2025.11.0
8
8
  torchmetrics >0.7.0, <1.9.0
9
9
  packaging >=20.0, <=25.0
10
10
  typing-extensions >4.5.0, <4.16.0
@@ -1,7 +1,7 @@
1
1
  -r ../docs.txt
2
2
 
3
3
  nbformat # used for generate empty notebook
4
- ipython[notebook] <9.6.0
4
+ ipython[notebook] <9.7.0
5
5
  setuptools<81.0 # workaround for `error in ipython setup command: use_2to3 is invalid.`
6
6
 
7
7
  onnxscript >= 0.2.2, < 0.5.0
@@ -2,6 +2,6 @@
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
4
  requests <2.33.0
5
- torchvision >=0.16.0, <0.24.0
6
- ipython[all] >=8.0.0, <9.0.0
5
+ torchvision >=0.16.0, <0.25.0
6
+ ipython[all] >=8.0.0, <10.0.0
7
7
  torchmetrics >=0.10.0, <1.9.0
@@ -5,7 +5,7 @@
5
5
  matplotlib>3.1, <3.11.0
6
6
  omegaconf >=2.2.3, <2.4.0
7
7
  hydra-core >=1.2.0, <1.4.0
8
- jsonargparse[signatures,jsonnet] >=4.39.0, <4.41.0
8
+ jsonargparse[signatures,jsonnet] >=4.39.0, <4.43.0
9
9
  rich >=12.3.0, <14.2.0
10
10
  tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute
11
11
  bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
@@ -3,4 +3,4 @@
3
3
 
4
4
  # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
5
5
  # shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
6
- deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin" # strict
6
+ deepspeed >=0.15.0,<0.17.0; platform_system != "Windows" and platform_system != "Darwin" # strict
@@ -1,8 +1,9 @@
1
- coverage ==7.10.6
2
- pytest ==8.4.1
3
- pytest-cov ==6.2.1
1
+ coverage ==7.10.7
2
+ pytest ==8.4.2
3
+ pytest-cov ==6.3.0
4
4
  pytest-timeout ==2.4.0
5
- pytest-rerunfailures ==16.0
5
+ pytest-rerunfailures ==16.0.1; python_version < "3.10"
6
+ pytest-rerunfailures ==16.1; python_version >= "3.10"
6
7
  pytest-random-order ==1.2.0
7
8
 
8
9
  # needed in tests
@@ -10,11 +11,14 @@ cloudpickle >=1.3, <3.2.0
10
11
  scikit-learn >0.22.1, <1.8.0
11
12
  numpy >1.20.0, <1.27.0
12
13
  onnx >1.12.0, <1.20.0
13
- onnxruntime >=1.12.0, <1.23.0
14
+ onnxruntime >=1.12.0, <1.24.0
14
15
  onnxscript >= 0.1.0, < 0.5.0
15
- psutil <7.0.1 # for `DeviceStatsMonitor`
16
+ psutil <7.1.2 # for `DeviceStatsMonitor`
16
17
  pandas >2.0, <2.4.0 # needed in benchmarks
17
18
  fastapi # for `ServableModuleValidator` # not setting version as re-defined in App
18
19
  uvicorn # for `ServableModuleValidator` # not setting version as re-defined in App
19
20
 
20
21
  tensorboard >=2.11, <2.21.0 # for `TensorBoardLogger`
22
+
23
+ torch-tensorrt; platform_system == "Linux" and python_version >= "3.12"
24
+ huggingface-hub
@@ -41,7 +41,7 @@ def _prepare_extras() -> dict[str, Any]:
41
41
  }
42
42
 
43
43
  # project specific extras groups
44
- extras["fabric-all"] = extras["fabric-strategies"] + extras["fabric-examples"]
44
+ extras["fabric-all"] = extras["fabric-extra"] + extras["fabric-strategies"] + extras["fabric-examples"]
45
45
  extras["fabric-dev"] = extras["fabric-all"] + extras["fabric-test"]
46
46
  extras["pytorch-all"] = extras["pytorch-extra"] + extras["pytorch-strategies"] + extras["pytorch-examples"]
47
47
  extras["pytorch-dev"] = extras["pytorch-all"] + extras["pytorch-test"]
@@ -4,6 +4,32 @@ All notable changes to this project will be documented in this file.
4
4
 
5
5
  The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
6
6
 
7
+
8
+ ## [unreleased] - YYYY-MM-DD
9
+
10
+ ### Added
11
+
12
+ -
13
+
14
+
15
+ ### Removed
16
+
17
+ -
18
+
19
+
20
+ ### Changed
21
+
22
+ -
23
+
24
+
25
+ - Set `_DeviceDtypeModuleMixin._device` from torch's default device function ([#21164](https://github.com/Lightning-AI/pytorch-lightning/pull/21164))
26
+
27
+
28
+ ### Fixed
29
+
30
+ - Fixed `EADDRINUSE` errors in distributed tests with port manager and retry logic ([#21309](https://github.com/Lightning-AI/pytorch-lightning/pull/21309))
31
+
32
+
7
33
  ---
8
34
 
9
35
  ## [2.5.5] - 2025-09-05
@@ -27,6 +53,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
27
53
  - Added support for NVIDIA H200 GPUs in `get_available_flops` ([#21119](https://github.com/Lightning-AI/pytorch-lightning/pull/21119))
28
54
 
29
55
 
56
+
30
57
  ## [2.5.3] - 2025-08-13
31
58
 
32
59
  ### Changed