lightning 2.5.2__tar.gz → 2.5.3__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 (299) hide show
  1. {lightning-2.5.2/src/lightning.egg-info → lightning-2.5.3}/PKG-INFO +60 -59
  2. {lightning-2.5.2 → lightning-2.5.3}/README.md +6 -0
  3. {lightning-2.5.2 → lightning-2.5.3}/pyproject.toml +1 -0
  4. {lightning-2.5.2 → lightning-2.5.3}/requirements/base.txt +3 -3
  5. {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/base.txt +4 -4
  6. {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/examples.txt +2 -3
  7. {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/strategies.txt +1 -1
  8. {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/test.txt +3 -3
  9. {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/base.txt +6 -6
  10. {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/docs.txt +3 -1
  11. {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/examples.txt +2 -3
  12. {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/extra.txt +1 -1
  13. {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/strategies.txt +1 -1
  14. {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/test.txt +5 -4
  15. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__about__.py +1 -1
  16. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/CHANGELOG.md +13 -0
  17. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/registry.py +2 -2
  18. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/cli.py +10 -2
  19. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/xla.py +15 -0
  20. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/bitsandbytes.py +5 -3
  21. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/consolidate_checkpoint.py +6 -5
  22. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/device_parser.py +15 -0
  23. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/imports.py +2 -1
  24. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/seed.py +4 -5
  25. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/spike.py +1 -4
  26. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/testing/_runif.py +7 -5
  27. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/CHANGELOG.md +24 -0
  28. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/lr_finder.py +1 -1
  29. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/model_checkpoint.py +49 -6
  30. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/rich_progress.py +39 -5
  31. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/module.py +62 -11
  32. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/optimizer.py +1 -1
  33. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/transformer.py +13 -4
  34. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/loop.py +8 -0
  35. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/manual.py +2 -1
  36. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/training_epoch_loop.py +5 -1
  37. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/advanced.py +6 -4
  38. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/call.py +2 -1
  39. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +6 -11
  40. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/data_connector.py +8 -0
  41. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +1 -1
  42. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/lr_finder.py +12 -25
  43. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/imports.py +4 -1
  44. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_summary/model_summary.py +17 -1
  45. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/parsing.py +2 -1
  46. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/testing/_runif.py +11 -4
  47. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +3 -2
  48. lightning-2.5.3/src/lightning/version.info +1 -0
  49. {lightning-2.5.2 → lightning-2.5.3/src/lightning.egg-info}/PKG-INFO +60 -59
  50. {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/SOURCES.txt +1 -1
  51. {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/requires.txt +34 -39
  52. lightning-2.5.3/src/version.info +1 -0
  53. lightning-2.5.2/src/lightning/version.info +0 -1
  54. lightning-2.5.2/src/version.info +0 -1
  55. {lightning-2.5.2 → lightning-2.5.3}/.actions/assistant.py +0 -0
  56. {lightning-2.5.2 → lightning-2.5.3}/CITATION.cff +0 -0
  57. {lightning-2.5.2 → lightning-2.5.3}/LICENSE +0 -0
  58. {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/docs.txt +0 -0
  59. {lightning-2.5.2 → lightning-2.5.3}/setup.cfg +0 -0
  60. {lightning-2.5.2 → lightning-2.5.3}/setup.py +0 -0
  61. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__init__.py +0 -0
  62. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__setup__.py +0 -0
  63. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__version__.py +0 -0
  64. /lightning-2.5.2/src/lightning/ffd6261c9058703d8c54093b4f93e2b4008030713613e504 → /lightning-2.5.3/src/lightning/b7f7dd8e64cd613ff6563bfe1965950a2c3243b7be6e14de +0 -0
  65. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/data/__init__.py +0 -0
  66. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/__init__.py +0 -0
  67. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/_graveyard/__init__.py +0 -0
  68. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/_graveyard/tpu.py +0 -0
  69. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/__init__.py +0 -0
  70. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/accelerator.py +0 -0
  71. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/cpu.py +0 -0
  72. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/cuda.py +0 -0
  73. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/mps.py +0 -0
  74. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/xla.py +0 -0
  75. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/connector.py +0 -0
  76. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/fabric.py +0 -0
  77. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/__init__.py +0 -0
  78. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/csv_logs.py +0 -0
  79. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/logger.py +0 -0
  80. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/tensorboard.py +0 -0
  81. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/__init__.py +0 -0
  82. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
  83. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
  84. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/single_device.py +0 -0
  85. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/torch_collective.py +0 -0
  86. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
  87. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
  88. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
  89. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/lightning.py +0 -0
  90. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/lsf.py +0 -0
  91. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
  92. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
  93. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
  94. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/__init__.py +0 -0
  95. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
  96. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/torch_io.py +0 -0
  97. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/xla.py +0 -0
  98. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
  99. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/amp.py +0 -0
  100. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/deepspeed.py +0 -0
  101. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/double.py +0 -0
  102. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/fsdp.py +0 -0
  103. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/half.py +0 -0
  104. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/precision.py +0 -0
  105. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/transformer_engine.py +0 -0
  106. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/utils.py +0 -0
  107. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/xla.py +0 -0
  108. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/__init__.py +0 -0
  109. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/ddp.py +0 -0
  110. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/deepspeed.py +0 -0
  111. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/dp.py +0 -0
  112. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/fsdp.py +0 -0
  113. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
  114. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
  115. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/multiprocessing.py +0 -0
  116. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/subprocess_script.py +0 -0
  117. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
  118. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/model_parallel.py +0 -0
  119. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/parallel.py +0 -0
  120. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/registry.py +0 -0
  121. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/single_device.py +0 -0
  122. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/single_xla.py +0 -0
  123. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/strategy.py +0 -0
  124. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/xla.py +0 -0
  125. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/xla_fsdp.py +0 -0
  126. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/__init__.py +0 -0
  127. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/apply_func.py +0 -0
  128. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/cloud_io.py +0 -0
  129. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/data.py +0 -0
  130. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/device_dtype_mixin.py +0 -0
  131. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/distributed.py +0 -0
  132. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/enums.py +0 -0
  133. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/exceptions.py +0 -0
  134. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/init.py +0 -0
  135. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/load.py +0 -0
  136. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/logger.py +0 -0
  137. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/optimizer.py +0 -0
  138. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/rank_zero.py +0 -0
  139. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/registry.py +0 -0
  140. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
  141. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/throughput.py +0 -0
  142. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/types.py +0 -0
  143. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/warnings.py +0 -0
  144. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/wrappers.py +0 -0
  145. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/py.typed +0 -0
  146. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/__init__.py +0 -0
  147. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
  148. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
  149. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
  150. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/precision.py +0 -0
  151. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
  152. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/__init__.py +0 -0
  153. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
  154. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/cpu.py +0 -0
  155. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/cuda.py +0 -0
  156. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/mps.py +0 -0
  157. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/xla.py +0 -0
  158. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/__init__.py +0 -0
  159. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
  160. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/callback.py +0 -0
  161. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
  162. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/device_stats_monitor.py +0 -0
  163. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/early_stopping.py +0 -0
  164. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/finetuning.py +0 -0
  165. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +0 -0
  166. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
  167. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/lr_monitor.py +0 -0
  168. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/model_summary.py +0 -0
  169. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
  170. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/prediction_writer.py +0 -0
  171. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
  172. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/progress_bar.py +0 -0
  173. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +0 -0
  174. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/pruning.py +0 -0
  175. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/rich_model_summary.py +0 -0
  176. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/spike.py +0 -0
  177. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +0 -0
  178. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/throughput_monitor.py +0 -0
  179. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/timer.py +0 -0
  180. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/cli.py +0 -0
  181. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/__init__.py +0 -0
  182. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/datamodule.py +0 -0
  183. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/hooks.py +0 -0
  184. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
  185. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/mixins/hparams_mixin.py +0 -0
  186. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/saving.py +0 -0
  187. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/__init__.py +0 -0
  188. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/boring_classes.py +0 -0
  189. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/lstm.py +0 -0
  190. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
  191. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/__init__.py +0 -0
  192. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/comet.py +0 -0
  193. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/csv_logs.py +0 -0
  194. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/logger.py +0 -0
  195. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/mlflow.py +0 -0
  196. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/neptune.py +0 -0
  197. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/tensorboard.py +0 -0
  198. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/utilities.py +0 -0
  199. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/wandb.py +0 -0
  200. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/__init__.py +0 -0
  201. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/evaluation_loop.py +0 -0
  202. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/fetchers.py +0 -0
  203. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/fit_loop.py +0 -0
  204. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
  205. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/automatic.py +0 -0
  206. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
  207. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/prediction_loop.py +0 -0
  208. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/progress.py +0 -0
  209. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/utilities.py +0 -0
  210. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/overrides/__init__.py +0 -0
  211. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/overrides/distributed.py +0 -0
  212. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/__init__.py +0 -0
  213. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
  214. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
  215. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/async_plugin.py +0 -0
  216. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
  217. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
  218. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/wrapper.py +0 -0
  219. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
  220. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
  221. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
  222. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/amp.py +0 -0
  223. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +0 -0
  224. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/deepspeed.py +0 -0
  225. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/double.py +0 -0
  226. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/fsdp.py +0 -0
  227. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/half.py +0 -0
  228. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/precision.py +0 -0
  229. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
  230. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
  231. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/__init__.py +0 -0
  232. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/base.py +0 -0
  233. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/profiler.py +0 -0
  234. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/pytorch.py +0 -0
  235. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/simple.py +0 -0
  236. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/xla.py +0 -0
  237. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/serve/__init__.py +0 -0
  238. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/serve/servable_module.py +0 -0
  239. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/serve/servable_module_validator.py +0 -0
  240. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/__init__.py +0 -0
  241. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/ddp.py +0 -0
  242. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/deepspeed.py +0 -0
  243. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/fsdp.py +0 -0
  244. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
  245. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
  246. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +0 -0
  247. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +0 -0
  248. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
  249. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/model_parallel.py +0 -0
  250. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/parallel.py +0 -0
  251. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/single_device.py +0 -0
  252. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/single_xla.py +0 -0
  253. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/strategy.py +0 -0
  254. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/xla.py +0 -0
  255. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/__init__.py +0 -0
  256. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
  257. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
  258. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/callback_connector.py +0 -0
  259. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +0 -0
  260. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
  261. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
  262. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +0 -0
  263. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/signal_connector.py +0 -0
  264. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/setup.py +0 -0
  265. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/states.py +0 -0
  266. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/trainer.py +0 -0
  267. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/__init__.py +0 -0
  268. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/batch_size_scaling.py +0 -0
  269. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/tuning.py +0 -0
  270. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/__init__.py +0 -0
  271. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/_pytree.py +0 -0
  272. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/argparse.py +0 -0
  273. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/combined_loader.py +0 -0
  274. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/compile.py +0 -0
  275. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +0 -0
  276. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/data.py +0 -0
  277. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/deepspeed.py +0 -0
  278. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/enums.py +0 -0
  279. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/exceptions.py +0 -0
  280. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/grads.py +0 -0
  281. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/memory.py +0 -0
  282. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
  283. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
  284. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/migration/utils.py +0 -0
  285. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_helpers.py +0 -0
  286. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_registry.py +0 -0
  287. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
  288. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +0 -0
  289. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
  290. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
  291. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/seed.py +0 -0
  292. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
  293. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
  294. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/types.py +0 -0
  295. {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/warnings.py +0 -0
  296. {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/dependency_links.txt +0 -0
  297. {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/entry_points.txt +0 -0
  298. {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/not-zip-safe +0 -0
  299. {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/top_level.txt +0 -0
@@ -1,11 +1,11 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lightning
3
- Version: 2.5.2
3
+ Version: 2.5.3
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
7
7
  Author: Lightning AI et al.
8
- Author-email: pytorch@lightning.ai
8
+ Author-email: developer@lightning.ai
9
9
  License: Apache-2.0
10
10
  Project-URL: Bug Tracker, https://github.com/Lightning-AI/pytorch-lightning/issues
11
11
  Project-URL: Documentation, https://lightning.ai/lightning-docs
@@ -27,93 +27,89 @@ Classifier: Programming Language :: Python :: 3.12
27
27
  Requires-Python: >=3.9
28
28
  Description-Content-Type: text/markdown
29
29
  License-File: LICENSE
30
- Requires-Dist: PyYAML<8.0,>=5.4
30
+ Requires-Dist: PyYAML<8.0,>5.4
31
31
  Requires-Dist: fsspec[http]<2027.0,>=2022.5.0
32
32
  Requires-Dist: lightning-utilities<2.0,>=0.10.0
33
33
  Requires-Dist: packaging<27.0,>=20.0
34
34
  Requires-Dist: torch<4.0,>=2.1.0
35
- Requires-Dist: torchmetrics<3.0,>=0.7.0
35
+ Requires-Dist: torchmetrics<3.0,>0.7.0
36
36
  Requires-Dist: tqdm<6.0,>=4.57.0
37
- Requires-Dist: typing-extensions<6.0,>=4.4.0
37
+ Requires-Dist: typing-extensions<6.0,>4.5.0
38
38
  Requires-Dist: pytorch-lightning
39
+ Provides-Extra: pytorch-extra
40
+ Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
41
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-extra"
42
+ Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-extra"
43
+ Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-extra"
44
+ Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
45
+ Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
46
+ Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
39
47
  Provides-Extra: pytorch-strategies
40
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
48
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
41
49
  Provides-Extra: pytorch-test
42
50
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-test"
43
- Requires-Dist: coverage==7.9.1; extra == "pytorch-test"
51
+ Requires-Dist: coverage==7.10.3; extra == "pytorch-test"
44
52
  Requires-Dist: fastapi; extra == "pytorch-test"
45
53
  Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-test"
46
54
  Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-test"
47
55
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-test"
56
+ Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "pytorch-test"
48
57
  Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-test"
49
58
  Requires-Dist: psutil<8.0; extra == "pytorch-test"
50
59
  Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-test"
51
- Requires-Dist: pytest-random-order==1.1.1; extra == "pytorch-test"
60
+ Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-test"
52
61
  Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-test"
53
62
  Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-test"
54
- Requires-Dist: pytest==8.4.0; extra == "pytorch-test"
63
+ Requires-Dist: pytest==8.4.1; extra == "pytorch-test"
55
64
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-test"
56
65
  Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "pytorch-test"
57
66
  Requires-Dist: uvicorn; extra == "pytorch-test"
58
- Provides-Extra: pytorch-extra
59
- Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
60
- Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-extra"
61
- Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-extra"
62
- Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-extra"
63
- Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
64
- Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
65
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
66
67
  Provides-Extra: pytorch-examples
67
68
  Requires-Dist: ipython[all]<9.0; extra == "pytorch-examples"
68
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-examples"
69
69
  Requires-Dist: requests<3.0; extra == "pytorch-examples"
70
70
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-examples"
71
71
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-examples"
72
72
  Provides-Extra: fabric-strategies
73
73
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-strategies"
74
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
74
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
75
75
  Provides-Extra: fabric-test
76
76
  Requires-Dist: click==8.1.8; extra == "fabric-test"
77
- Requires-Dist: coverage==7.9.1; extra == "fabric-test"
77
+ Requires-Dist: coverage==7.10.3; extra == "fabric-test"
78
78
  Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-test"
79
79
  Requires-Dist: pytest-cov==6.2.1; extra == "fabric-test"
80
- Requires-Dist: pytest-random-order==1.1.1; extra == "fabric-test"
80
+ Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-test"
81
81
  Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-test"
82
82
  Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-test"
83
- Requires-Dist: pytest==8.4.0; extra == "fabric-test"
83
+ Requires-Dist: pytest==8.4.1; extra == "fabric-test"
84
84
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-test"
85
85
  Provides-Extra: fabric-examples
86
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-examples"
87
86
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-examples"
88
87
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-examples"
89
88
  Provides-Extra: fabric-all
90
89
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-all"
91
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
92
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-all"
90
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
93
91
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-all"
94
92
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-all"
95
93
  Provides-Extra: fabric-dev
96
94
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-dev"
97
95
  Requires-Dist: click==8.1.8; extra == "fabric-dev"
98
- Requires-Dist: coverage==7.9.1; extra == "fabric-dev"
99
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
100
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-dev"
96
+ Requires-Dist: coverage==7.10.3; extra == "fabric-dev"
97
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
101
98
  Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-dev"
102
99
  Requires-Dist: pytest-cov==6.2.1; extra == "fabric-dev"
103
- Requires-Dist: pytest-random-order==1.1.1; extra == "fabric-dev"
100
+ Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-dev"
104
101
  Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-dev"
105
102
  Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-dev"
106
- Requires-Dist: pytest==8.4.0; extra == "fabric-dev"
103
+ Requires-Dist: pytest==8.4.1; extra == "fabric-dev"
107
104
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-dev"
108
105
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-dev"
109
106
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-dev"
110
107
  Provides-Extra: pytorch-all
111
108
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-all"
112
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
109
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
113
110
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-all"
114
111
  Requires-Dist: ipython[all]<9.0; extra == "pytorch-all"
115
112
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-all"
116
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-all"
117
113
  Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-all"
118
114
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-all"
119
115
  Requires-Dist: requests<3.0; extra == "pytorch-all"
@@ -124,25 +120,25 @@ Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-all"
124
120
  Provides-Extra: pytorch-dev
125
121
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-dev"
126
122
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-dev"
127
- Requires-Dist: coverage==7.9.1; extra == "pytorch-dev"
128
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
123
+ Requires-Dist: coverage==7.10.3; extra == "pytorch-dev"
124
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
129
125
  Requires-Dist: fastapi; extra == "pytorch-dev"
130
126
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-dev"
131
127
  Requires-Dist: ipython[all]<9.0; extra == "pytorch-dev"
132
128
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-dev"
133
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-dev"
134
129
  Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
135
130
  Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-dev"
136
131
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-dev"
137
132
  Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-dev"
138
133
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-dev"
134
+ Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "pytorch-dev"
139
135
  Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-dev"
140
136
  Requires-Dist: psutil<8.0; extra == "pytorch-dev"
141
137
  Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-dev"
142
- Requires-Dist: pytest-random-order==1.1.1; extra == "pytorch-dev"
138
+ Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-dev"
143
139
  Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-dev"
144
140
  Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-dev"
145
- Requires-Dist: pytest==8.4.0; extra == "pytorch-dev"
141
+ Requires-Dist: pytest==8.4.1; extra == "pytorch-dev"
146
142
  Requires-Dist: requests<3.0; extra == "pytorch-dev"
147
143
  Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-dev"
148
144
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-dev"
@@ -151,39 +147,39 @@ Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-dev"
151
147
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-dev"
152
148
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-dev"
153
149
  Requires-Dist: uvicorn; extra == "pytorch-dev"
150
+ Provides-Extra: extra
151
+ Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
152
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
153
+ Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
154
+ Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
155
+ Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
156
+ Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
157
+ Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
154
158
  Provides-Extra: strategies
155
159
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "strategies"
156
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
160
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
157
161
  Provides-Extra: test
158
162
  Requires-Dist: click==8.1.8; extra == "test"
159
163
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
160
- Requires-Dist: coverage==7.9.1; extra == "test"
164
+ Requires-Dist: coverage==7.10.3; extra == "test"
161
165
  Requires-Dist: fastapi; extra == "test"
162
166
  Requires-Dist: numpy<2.0,>=1.17.2; extra == "test"
163
167
  Requires-Dist: onnx<2.0,>=1.12.0; extra == "test"
164
168
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "test"
169
+ Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "test"
165
170
  Requires-Dist: pandas<3.0,>2.0; extra == "test"
166
171
  Requires-Dist: psutil<8.0; extra == "test"
167
172
  Requires-Dist: pytest-cov==6.2.1; extra == "test"
168
- Requires-Dist: pytest-random-order==1.1.1; extra == "test"
173
+ Requires-Dist: pytest-random-order==1.2.0; extra == "test"
169
174
  Requires-Dist: pytest-rerunfailures==15.1; extra == "test"
170
175
  Requires-Dist: pytest-timeout==2.4.0; extra == "test"
171
- Requires-Dist: pytest==8.4.0; extra == "test"
176
+ Requires-Dist: pytest==8.4.1; extra == "test"
172
177
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "test"
173
178
  Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "test"
174
179
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "test"
175
180
  Requires-Dist: uvicorn; extra == "test"
176
- Provides-Extra: extra
177
- Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
178
- Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
179
- Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
180
- Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
181
- Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
182
- Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
183
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
184
181
  Provides-Extra: examples
185
182
  Requires-Dist: ipython[all]<9.0; extra == "examples"
186
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "examples"
187
183
  Requires-Dist: requests<3.0; extra == "examples"
188
184
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "examples"
189
185
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "examples"
@@ -191,11 +187,10 @@ Provides-Extra: data
191
187
  Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
192
188
  Provides-Extra: all
193
189
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "all"
194
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
190
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
195
191
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "all"
196
192
  Requires-Dist: ipython[all]<9.0; extra == "all"
197
193
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "all"
198
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "all"
199
194
  Requires-Dist: matplotlib<4.0,>3.1; extra == "all"
200
195
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "all"
201
196
  Requires-Dist: requests<3.0; extra == "all"
@@ -207,25 +202,25 @@ Provides-Extra: dev
207
202
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "dev"
208
203
  Requires-Dist: click==8.1.8; extra == "dev"
209
204
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
210
- Requires-Dist: coverage==7.9.1; extra == "dev"
211
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
205
+ Requires-Dist: coverage==7.10.3; extra == "dev"
206
+ Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
212
207
  Requires-Dist: fastapi; extra == "dev"
213
208
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "dev"
214
209
  Requires-Dist: ipython[all]<9.0; extra == "dev"
215
210
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "dev"
216
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "dev"
217
211
  Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
218
212
  Requires-Dist: numpy<2.0,>=1.17.2; extra == "dev"
219
213
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "dev"
220
214
  Requires-Dist: onnx<2.0,>=1.12.0; extra == "dev"
221
215
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "dev"
216
+ Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "dev"
222
217
  Requires-Dist: pandas<3.0,>2.0; extra == "dev"
223
218
  Requires-Dist: psutil<8.0; extra == "dev"
224
219
  Requires-Dist: pytest-cov==6.2.1; extra == "dev"
225
- Requires-Dist: pytest-random-order==1.1.1; extra == "dev"
220
+ Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
226
221
  Requires-Dist: pytest-rerunfailures==15.1; extra == "dev"
227
222
  Requires-Dist: pytest-timeout==2.4.0; extra == "dev"
228
- Requires-Dist: pytest==8.4.0; extra == "dev"
223
+ Requires-Dist: pytest==8.4.1; extra == "dev"
229
224
  Requires-Dist: requests<3.0; extra == "dev"
230
225
  Requires-Dist: rich<15.0,>=12.3.0; extra == "dev"
231
226
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "dev"
@@ -270,7 +265,7 @@ ______________________________________________________________________
270
265
  <a href="#lightning-fabric-expert-control">Fabric</a> •
271
266
  <a href="https://lightning.ai/">Lightning AI</a> •
272
267
  <a href="#community">Community</a> •
273
- <a href="https://pytorch-lightning.readthedocs.io/en/2.5.2">Docs</a>
268
+ <a href="https://pytorch-lightning.readthedocs.io/en/2.5.3">Docs</a>
274
269
  </p>
275
270
 
276
271
  <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -307,6 +302,12 @@ ______________________________________________________________________
307
302
 
308
303
  &nbsp;
309
304
 
305
+ # Why PyTorch Lightning?
306
+
307
+ 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 more DIY.
308
+
309
+ Fun analogy: If PyTorch is Javascript, PyTorch Lightning is ReactJS or NextJS.
310
+
310
311
  # Lightning has 2 core packages
311
312
 
312
313
  [PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
@@ -795,7 +796,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
795
796
 
796
797
  | System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
797
798
  | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
798
- | 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.2)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
799
+ | 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.3)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
799
800
  | 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) |
800
801
  | 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) |
801
802
  | 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) |
@@ -55,6 +55,12 @@ ______________________________________________________________________
55
55
 
56
56
  &nbsp;
57
57
 
58
+ # Why PyTorch Lightning?
59
+
60
+ 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 more DIY.
61
+
62
+ Fun analogy: If PyTorch is Javascript, PyTorch Lightning is ReactJS or NextJS.
63
+
58
64
  # Lightning has 2 core packages
59
65
 
60
66
  [PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
@@ -180,6 +180,7 @@ markers = [
180
180
  ]
181
181
  filterwarnings = [
182
182
  "error::FutureWarning",
183
+ "ignore::FutureWarning:onnxscript", # Temporary ignore until onnxscript is updated
183
184
  ]
184
185
  xfail_strict = true
185
186
  junit_duration_report = "call"
@@ -1,9 +1,9 @@
1
- PyYAML<7.0,>=5.4
1
+ PyYAML<7.0,>5.4
2
2
  fsspec[http]<2026.0,>=2022.5.0
3
3
  lightning-utilities<1.0,>=0.10.0
4
4
  packaging<26.0,>=20.0
5
5
  torch<3.0,>=2.1.0
6
- torchmetrics<2.0,>=0.7.0
6
+ torchmetrics<2.0,>0.7.0
7
7
  tqdm<5.0,>=4.57.0
8
- typing-extensions<5.0,>=4.4.0
8
+ typing-extensions<5.0,>4.5.0
9
9
 
@@ -1,8 +1,8 @@
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
- torch >=2.1.0, <2.8.0
5
- fsspec[http] >=2022.5.0, <2025.6.0
4
+ torch >=2.1.0, <2.9.0
5
+ fsspec[http] >=2022.5.0, <2025.8.0
6
6
  packaging >=20.0, <=25.0
7
- typing-extensions >=4.4.0, <4.15.0
8
- lightning-utilities >=0.10.0, <0.15.0
7
+ typing-extensions >4.5.0, <4.15.0
8
+ lightning-utilities >=0.10.0, <0.16.0
@@ -1,6 +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.23.0
5
- torchmetrics >=0.10.0, <1.8.0
6
- lightning-utilities >=0.8.0, <0.15.0
4
+ torchvision >=0.16.0, <0.24.0
5
+ torchmetrics >=0.10.0, <1.9.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.8.2, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
8
+ deepspeed >=0.9.3, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
9
9
  bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
@@ -1,9 +1,9 @@
1
- coverage ==7.9.1
1
+ coverage ==7.10.3
2
2
  numpy >=1.17.2, <1.27.0
3
- pytest ==8.4.0
3
+ pytest ==8.4.1
4
4
  pytest-cov ==6.2.1
5
5
  pytest-timeout ==2.4.0
6
6
  pytest-rerunfailures ==15.1
7
- pytest-random-order ==1.1.1
7
+ pytest-random-order ==1.2.0
8
8
  click ==8.1.8
9
9
  tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute
@@ -1,11 +1,11 @@
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
- torch >=2.1.0, <2.8.0
4
+ torch >=2.1.0, <2.9.0
5
5
  tqdm >=4.57.0, <4.68.0
6
- PyYAML >=5.4, <6.1.0
7
- fsspec[http] >=2022.5.0, <2025.6.0
8
- torchmetrics >=0.7.0, <1.8.0
6
+ PyYAML >5.4, <6.1.0
7
+ fsspec[http] >=2022.5.0, <2025.8.0
8
+ torchmetrics >0.7.0, <1.9.0
9
9
  packaging >=20.0, <=25.0
10
- typing-extensions >=4.4.0, <4.15.0
11
- lightning-utilities >=0.10.0, <0.15.0
10
+ typing-extensions >4.5.0, <4.15.0
11
+ lightning-utilities >=0.10.0, <0.16.0
@@ -4,4 +4,6 @@ nbformat # used for generate empty notebook
4
4
  ipython[notebook] <8.19.0
5
5
  setuptools<81.0 # workaround for `error in ipython setup command: use_2to3 is invalid.`
6
6
 
7
- -r ../../_notebooks/.actions/requires.txt
7
+ onnxscript >= 0.2.2, <0.4.0
8
+
9
+ #-r ../../_notebooks/.actions/requires.txt
@@ -2,7 +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.23.0
5
+ torchvision >=0.16.0, <0.24.0
6
6
  ipython[all] <8.19.0
7
- torchmetrics >=0.10.0, <1.8.0
8
- lightning-utilities >=0.8.0, <0.15.0
7
+ torchmetrics >=0.10.0, <1.9.0
@@ -6,6 +6,6 @@ matplotlib>3.1, <3.10.0
6
6
  omegaconf >=2.2.3, <2.4.0
7
7
  hydra-core >=1.2.0, <1.4.0
8
8
  jsonargparse[signatures,jsonnet] >=4.39.0, <4.41.0
9
- rich >=12.3.0, <14.1.0
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.8.2, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
6
+ deepspeed >=0.9.3, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
@@ -1,9 +1,9 @@
1
- coverage ==7.9.1
2
- pytest ==8.4.0
1
+ coverage ==7.10.3
2
+ pytest ==8.4.1
3
3
  pytest-cov ==6.2.1
4
4
  pytest-timeout ==2.4.0
5
5
  pytest-rerunfailures ==15.1
6
- pytest-random-order ==1.1.1
6
+ pytest-random-order ==1.2.0
7
7
 
8
8
  # needed in tests
9
9
  cloudpickle >=1.3, <3.2.0
@@ -11,9 +11,10 @@ scikit-learn >0.22.1, <1.7.0
11
11
  numpy >=1.17.2, <1.27.0
12
12
  onnx >=1.12.0, <1.19.0
13
13
  onnxruntime >=1.12.0, <1.21.0
14
+ onnxscript >= 0.2.2, <0.4.0
14
15
  psutil <7.0.1 # for `DeviceStatsMonitor`
15
16
  pandas >2.0, <2.4.0 # needed in benchmarks
16
17
  fastapi # for `ServableModuleValidator` # not setting version as re-defined in App
17
18
  uvicorn # for `ServableModuleValidator` # not setting version as re-defined in App
18
19
 
19
- tensorboard >=2.9.1, <2.20.0 # for `TensorBoardLogger`
20
+ tensorboard >=2.9.1, <2.21.0 # for `TensorBoardLogger`
@@ -15,7 +15,7 @@
15
15
  import time
16
16
 
17
17
  __author__ = "Lightning AI et al."
18
- __author_email__ = "pytorch@lightning.ai"
18
+ __author_email__ = "developer@lightning.ai"
19
19
  __license__ = "Apache-2.0"
20
20
  __copyright__ = f"Copyright (c) 2018-{time.strftime('%Y')}, {__author__}."
21
21
  __homepage__ = "https://github.com/Lightning-AI/lightning"
@@ -6,6 +6,19 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
6
6
 
7
7
  ---
8
8
 
9
+ ## [2.5.3] - 2025-08-DD
10
+
11
+ ### Changed
12
+
13
+ - Enable "auto" for `devices` and `accelerator` as CLI arguments ([#20913](https://github.com/Lightning-AI/pytorch-lightning/pull/20913))
14
+ - Raise ValueError when seed is `out-of-bounds` or `cannot be cast to int` ([#21029](https://github.com/Lightning-AI/pytorch-lightning/pull/21029))
15
+
16
+ ### Fixed
17
+
18
+ - Fixed XLA strategy to add support for `global_ordinal`, `local_ordinal`, `world_size` which came instead of deprecated methods ([#20852](https://github.com/Lightning-AI/pytorch-lightning/issues/20852))
19
+ - Fixed remove extra `name` parameter in accelerator registry decorator ([#20975](https://github.com/Lightning-AI/pytorch-lightning/pull/20975))
20
+
21
+
9
22
  ## [2.5.2] - 2025-3-20
10
23
 
11
24
  ### Changed
@@ -73,14 +73,14 @@ class _AcceleratorRegistry(dict):
73
73
  data["description"] = description
74
74
  data["init_params"] = init_params
75
75
 
76
- def do_register(name: str, accelerator: Callable) -> Callable:
76
+ def do_register(accelerator: Callable) -> Callable:
77
77
  data["accelerator"] = accelerator
78
78
  data["accelerator_name"] = name
79
79
  self[name] = data
80
80
  return accelerator
81
81
 
82
82
  if accelerator is not None:
83
- return do_register(name, accelerator)
83
+ return do_register(accelerator)
84
84
 
85
85
  return do_register
86
86
 
@@ -25,7 +25,7 @@ from lightning.fabric.accelerators import CPUAccelerator, CUDAAccelerator, MPSAc
25
25
  from lightning.fabric.plugins.precision.precision import _PRECISION_INPUT_STR, _PRECISION_INPUT_STR_ALIAS
26
26
  from lightning.fabric.strategies import STRATEGY_REGISTRY
27
27
  from lightning.fabric.utilities.consolidate_checkpoint import _process_cli_args
28
- from lightning.fabric.utilities.device_parser import _parse_gpu_ids
28
+ from lightning.fabric.utilities.device_parser import _parse_gpu_ids, _select_auto_accelerator
29
29
  from lightning.fabric.utilities.distributed import _suggested_max_num_threads
30
30
  from lightning.fabric.utilities.load import _load_distributed_checkpoint
31
31
 
@@ -34,7 +34,7 @@ _log = logging.getLogger(__name__)
34
34
  _CLICK_AVAILABLE = RequirementCache("click")
35
35
  _LIGHTNING_SDK_AVAILABLE = RequirementCache("lightning_sdk")
36
36
 
37
- _SUPPORTED_ACCELERATORS = ("cpu", "gpu", "cuda", "mps", "tpu")
37
+ _SUPPORTED_ACCELERATORS = ("cpu", "gpu", "cuda", "mps", "tpu", "auto")
38
38
 
39
39
 
40
40
  def _get_supported_strategies() -> list[str]:
@@ -187,6 +187,14 @@ def _set_env_variables(args: Namespace) -> None:
187
187
 
188
188
  def _get_num_processes(accelerator: str, devices: str) -> int:
189
189
  """Parse the `devices` argument to determine how many processes need to be launched on the current machine."""
190
+
191
+ if accelerator == "auto" or accelerator is None:
192
+ accelerator = _select_auto_accelerator()
193
+ if devices == "auto":
194
+ if accelerator == "cuda" or accelerator == "mps" or accelerator == "cpu":
195
+ devices = "1"
196
+ else:
197
+ raise ValueError(f"Cannot default to '1' device for accelerator='{accelerator}'")
190
198
  if accelerator == "gpu":
191
199
  parsed_devices = _parse_gpu_ids(devices, include_cuda=True, include_mps=True)
192
200
  elif accelerator == "cuda":
@@ -66,6 +66,11 @@ class XLAEnvironment(ClusterEnvironment):
66
66
  The output is cached for performance.
67
67
 
68
68
  """
69
+ if _XLA_GREATER_EQUAL_2_1:
70
+ from torch_xla import runtime as xr
71
+
72
+ return xr.world_size()
73
+
69
74
  import torch_xla.core.xla_model as xm
70
75
 
71
76
  return xm.xrt_world_size()
@@ -82,6 +87,11 @@ class XLAEnvironment(ClusterEnvironment):
82
87
  The output is cached for performance.
83
88
 
84
89
  """
90
+ if _XLA_GREATER_EQUAL_2_1:
91
+ from torch_xla import runtime as xr
92
+
93
+ return xr.global_ordinal()
94
+
85
95
  import torch_xla.core.xla_model as xm
86
96
 
87
97
  return xm.get_ordinal()
@@ -98,6 +108,11 @@ class XLAEnvironment(ClusterEnvironment):
98
108
  The output is cached for performance.
99
109
 
100
110
  """
111
+ if _XLA_GREATER_EQUAL_2_1:
112
+ from torch_xla import runtime as xr
113
+
114
+ return xr.local_ordinal()
115
+
101
116
  import torch_xla.core.xla_model as xm
102
117
 
103
118
  return xm.get_local_ordinal()
@@ -256,10 +256,12 @@ def _import_bitsandbytes() -> ModuleType:
256
256
  if int8params.has_fp16_weights:
257
257
  int8params.data = B
258
258
  else:
259
- if hasattr(bnb.functional, "double_quant"):
259
+ # bitsandbytes >= 0.45 supports an improved API
260
+ if hasattr(bnb.functional, "int8_vectorwise_quant"):
261
+ CB, SCB, _ = bnb.functional.int8_vectorwise_quant(B)
262
+ else: # old method is deprecated in 0.45, removed in 0.46+.
260
263
  CB, _, SCB, _, _ = bnb.functional.double_quant(B)
261
- else: # for bitsandbytes versions ≥0.46
262
- CB, SCB = bnb.functional.int8_double_quant(B)
264
+
263
265
  int8params.data = CB
264
266
  setattr(int8params, "CB", CB)
265
267
  setattr(int8params, "SCB", SCB)