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.
- {lightning-2.5.6/src/lightning.egg-info → lightning-2.6.0.dev0}/PKG-INFO +78 -57
- {lightning-2.5.6 → lightning-2.6.0.dev0}/README.md +4 -3
- {lightning-2.5.6 → lightning-2.6.0.dev0}/pyproject.toml +0 -3
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/base.txt +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/examples.txt +1 -1
- lightning-2.6.0.dev0/requirements/fabric/extra.txt +4 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/strategies.txt +1 -1
- lightning-2.6.0.dev0/requirements/fabric/test.txt +12 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/base.txt +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/docs.txt +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/examples.txt +2 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/extra.txt +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/strategies.txt +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/pytorch/test.txt +10 -6
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__setup__.py +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/CHANGELOG.md +27 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/fabric.py +199 -49
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/lightning.py +28 -6
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/deepspeed.py +30 -16
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/fsdp.py +4 -3
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/strategy.py +7 -4
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/xla_fsdp.py +3 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/device_dtype_mixin.py +5 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/imports.py +1 -0
- lightning-2.6.0.dev0/src/lightning/fabric/utilities/port_manager.py +233 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/testing/_runif.py +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/CHANGELOG.md +56 -4
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/__init__.py +2 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/early_stopping.py +32 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/lr_monitor.py +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/model_checkpoint.py +12 -6
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/model_summary.py +6 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/rich_progress.py +28 -20
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +4 -4
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/pruning.py +3 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/rich_model_summary.py +4 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/throughput_monitor.py +25 -7
- lightning-2.6.0.dev0/src/lightning/pytorch/callbacks/weight_averaging.py +363 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/cli.py +12 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/module.py +115 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/mlflow.py +2 -4
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/wandb.py +4 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/evaluation_loop.py +5 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/fit_loop.py +9 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/training_epoch_loop.py +8 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/profiler.py +4 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/ddp.py +22 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/deepspeed.py +15 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/xla.py +4 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +1 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/callback_connector.py +4 -9
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/data_connector.py +4 -3
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +1 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/setup.py +73 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/trainer.py +22 -5
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/batch_size_scaling.py +16 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/deepspeed.py +2 -2
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/imports.py +1 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_helpers.py +11 -5
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_summary/model_summary.py +63 -13
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/testing/_runif.py +7 -3
- lightning-2.6.0.dev0/src/lightning/version.info +1 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0/src/lightning.egg-info}/PKG-INFO +78 -57
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/SOURCES.txt +4 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/requires.txt +96 -43
- lightning-2.6.0.dev0/src/version.info +1 -0
- lightning-2.5.6/requirements/fabric/test.txt +0 -10
- lightning-2.5.6/src/lightning/version.info +0 -1
- lightning-2.5.6/src/version.info +0 -1
- {lightning-2.5.6 → lightning-2.6.0.dev0}/.actions/assistant.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/CITATION.cff +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/LICENSE +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/base.txt +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/requirements/fabric/docs.txt +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/setup.cfg +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/setup.py +0 -0
- /lightning-2.5.6/src/lightning/d42c98d5f7b8f5e086a66f872d1e2133132dc322fc7ce9dd → /lightning-2.6.0.dev0/src/lightning/36e4fb47a659f3f6d57e62c858555053c62817fcd39048c0 +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__about__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/__version__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/data/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/_graveyard/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/_graveyard/tpu.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/accelerator.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/cpu.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/cuda.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/mps.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/registry.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/accelerators/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/cli.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/connector.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/csv_logs.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/logger.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/loggers/tensorboard.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/single_device.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/collectives/torch_collective.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/lsf.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/environments/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/torch_io.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/io/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/amp.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/bitsandbytes.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/deepspeed.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/double.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/fsdp.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/half.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/precision.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/transformer_engine.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/utils.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/plugins/precision/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/ddp.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/dp.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/multiprocessing.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/subprocess_script.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/model_parallel.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/parallel.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/registry.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/single_device.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/single_xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/strategies/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/apply_func.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/cloud_io.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/consolidate_checkpoint.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/data.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/device_parser.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/distributed.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/enums.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/exceptions.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/init.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/load.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/logger.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/optimizer.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/rank_zero.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/registry.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/seed.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/spike.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/throughput.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/types.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/utilities/warnings.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/fabric/wrappers.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/py.typed +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/precision.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/cpu.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/cuda.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/mps.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/accelerators/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/callback.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/device_stats_monitor.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/finetuning.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/lr_finder.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/prediction_writer.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/progress/progress_bar.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/spike.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/callbacks/timer.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/datamodule.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/hooks.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/mixins/hparams_mixin.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/optimizer.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/core/saving.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/boring_classes.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/lstm.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/demos/transformer.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/comet.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/csv_logs.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/logger.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/neptune.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/tensorboard.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loggers/utilities.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/fetchers.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/loop.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/automatic.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/optimization/manual.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/prediction_loop.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/progress.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/loops/utilities.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/overrides/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/overrides/distributed.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/async_plugin.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/wrapper.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/amp.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/deepspeed.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/double.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/fsdp.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/half.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/precision.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/advanced.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/base.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/pytorch.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/simple.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/profilers/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/serve/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/serve/servable_module.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/serve/servable_module_validator.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/fsdp.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/model_parallel.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/parallel.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/single_device.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/single_xla.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/strategies/strategy.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/call.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/connectors/signal_connector.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/trainer/states.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/lr_finder.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/tuner/tuning.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/_pytree.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/argparse.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/combined_loader.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/compile.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/data.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/enums.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/exceptions.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/grads.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/memory.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/migration/utils.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_registry.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/parsing.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/seed.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/types.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning/pytorch/utilities/warnings.py +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/dependency_links.txt +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/entry_points.txt +0 -0
- {lightning-2.5.6 → lightning-2.6.0.dev0}/src/lightning.egg-info/not-zip-safe +0 -0
- {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.
|
|
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
|
|
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.
|
|
48
|
-
Requires-Dist: coverage==7.10.
|
|
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.
|
|
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.
|
|
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
|
|
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]<
|
|
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.
|
|
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.
|
|
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.
|
|
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
|
|
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.
|
|
98
|
-
Requires-Dist: coverage==7.10.
|
|
99
|
-
Requires-Dist: deepspeed
|
|
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.
|
|
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.
|
|
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
|
|
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]<
|
|
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.
|
|
126
|
-
Requires-Dist: deepspeed
|
|
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]<
|
|
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.
|
|
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.
|
|
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
|
|
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]<
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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
|
|
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]<
|
|
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.
|
|
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.
|
|
211
|
-
Requires-Dist: deepspeed
|
|
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]<
|
|
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.
|
|
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.
|
|
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
|
|
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.
|
|
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
|
-
|
|
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](
|
|
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\] | | | [](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
|
|
818
839
|
| Linux (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
819
840
|
| OSX (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
820
841
|
| Windows (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](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
|
|
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
|
-
|
|
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](
|
|
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.
|
|
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.
|
|
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.
|
|
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
|
|
@@ -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.
|
|
6
|
-
ipython[all] >=8.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.
|
|
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.
|
|
6
|
+
deepspeed >=0.15.0,<0.17.0; platform_system != "Windows" and platform_system != "Darwin" # strict
|
|
@@ -1,8 +1,9 @@
|
|
|
1
|
-
coverage ==7.10.
|
|
2
|
-
pytest ==8.4.
|
|
3
|
-
pytest-cov ==6.
|
|
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.
|
|
14
|
+
onnxruntime >=1.12.0, <1.24.0
|
|
14
15
|
onnxscript >= 0.1.0, < 0.5.0
|
|
15
|
-
psutil <7.
|
|
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
|