Lightning-AI/pytorch-lightning-2.5.0.post0.zip
The previewer is not showing all the files.
Lightning-AI-pytorch-lightning-9177ec0
.actions
assistant.py
21.0 kB
pull_legacy_checkpoints.sh
404 Bytes
requirements.txt
51 Bytes
.azure
README.md
1.8 kB
gpu-benchmarks.yml
3.5 kB
gpu-tests-fabric.yml
7.0 kB
gpu-tests-pytorch.yml
8.6 kB
start.sh
2.7 kB
.codecov.yml
2.2 kB
.git-blame-ignore-revs
582 Bytes
.github
BECOMING_A_CORE_CONTRIBUTOR.md
4.4 kB
CODEOWNERS
950 Bytes
CODE_OF_CONDUCT.md
3.4 kB
CONTRIBUTING.md
16.2 kB
ISSUE_TEMPLATE
1_bug_report.yaml
3.8 kB
2_refactor.yaml
1.5 kB
3_feature_request.yaml
1.8 kB
4_documentation.yaml
1.3 kB
config.yml
617 Bytes
PULL_REQUEST_TEMPLATE.md
2.3 kB
actions
pip-wheels
action.yml
2.5 kB
pkg-check
action.yml
1.8 kB
pkg-install
action.yml
1.8 kB
pkg-publish
action.yml
1.1 kB
advanced-issue-labeler.yml
732 Bytes
checkgroup.yml
10.3 kB
dependabot.yml
1.3 kB
label-change.yml
1.8 kB
lightning-probot.yml
22 Bytes
markdown-links-config.json
492 Bytes
mergify.yml
2.2 kB
stale.yml
2.2 kB
workflows
README.md
7.5 kB
_build-packages.yml
1.9 kB
_legacy-checkpoints.yml
5.6 kB
call-clear-cache.yml
1.3 kB
ci-check-md-links.yml
589 Bytes
ci-checkpoints.yml
607 Bytes
ci-pkg-install.yml
3.6 kB
ci-rtfd.yml
280 Bytes
ci-schema.yml
388 Bytes
ci-tests-fabric.yml
8.8 kB
ci-tests-pytorch.yml
10.2 kB
cleanup-caches.yml
1.0 kB
code-checks.yml
1.2 kB
docker-build.yml
6.4 kB
docs-build.yml
7.2 kB
docs-tutorials.yml
2.6 kB
labeler-issue.yml
793 Bytes
labeler-pr.yml
326 Bytes
probot-auto-cc.yml
572 Bytes
probot-check-group.yml
741 Bytes
release-nightly.yml
2.1 kB
release-pkg.yml
6.8 kB
tpu-tests.yml.disabled
6.0 kB
.gitignore
2.8 kB
.gitmodules
129 Bytes
.pre-commit-config.yaml
3.4 kB
.readthedocs.yml
1.8 kB
CITATION.cff
632 Bytes
LICENSE
11.3 kB
Makefile
1.9 kB
README.md
25.9 kB
SECURITY.md
23 Bytes
_notebooks
dockers
README.md
2.0 kB
base-cuda
Dockerfile
4.2 kB
docs
Dockerfile
2.1 kB
nvidia
Dockerfile
2.6 kB
release
Dockerfile
1.9 kB
docs
README.md
2.5 kB
crossroad.html
280 Bytes
generate_docs_for_tags.sh
1.6 kB
redirect.html
218 Bytes
rtfd-build.sh
571 Bytes
source-fabric
Makefile
589 Bytes
_static
copybutton.js
2.7 kB
images
icon.svg
2.8 kB
logo-large.svg
2.8 kB
logo-small.svg
2.8 kB
logo.png
14.4 kB
logo.svg
36.9 kB
main.css
234 Bytes
_templates
autosummary
module.rst
650 Bytes
classtemplate.rst
255 Bytes
classtemplate_no_index.rst
166 Bytes
layout.html
405 Bytes
theme_variables.jinja
675 Bytes
advanced
compile.rst
16.8 kB
distributed_communication.rst
9.6 kB
gradient_accumulation.rst
2.1 kB
model_init.rst
3.2 kB
model_parallel
fsdp.rst
16.2 kB
index.rst
6.5 kB
tp.rst
11.0 kB
tp_fsdp.rst
11.3 kB
multiple_setup.rst
3.3 kB
api
accelerators.rst
385 Bytes
collectives.rst
494 Bytes
environments.rst
588 Bytes
fabric.rst
267 Bytes
fabric_args.rst
7.1 kB
fabric_methods.rst
13.6 kB
io.rst
468 Bytes
loggers.rst
516 Bytes
precision.rst
492 Bytes
strategies.rst
509 Bytes
utilities.rst
772 Bytes
wrappers.rst
4.5 kB
conf.py
13.8 kB
examples
index.rst
2.4 kB
fundamentals
accelerators.rst
1.8 kB
code_structure.rst
4.4 kB
convert.rst
4.3 kB
installation.rst
1.4 kB
launch.rst
7.3 kB
notebooks.rst
2.6 kB
precision.rst
14.1 kB
glossary
index.rst
5.5 kB
guide
callbacks.rst
3.1 kB
checkpoint
checkpoint.rst
7.0 kB
distributed_checkpoint.rst
7.2 kB
index.rst
684 Bytes
index.rst
5.2 kB
lightning_module.rst
4.4 kB
loggers
wandb.rst
3.3 kB
logging.rst
3.3 kB
multi_node
barebones.rst
4.4 kB
cloud.rst
4.5 kB
other.rst
1.7 kB
slurm.rst
4.1 kB
trainer_template.rst
230 Bytes
index.rst
6.0 kB
levels
advanced.rst
2.1 kB
basic.rst
1.9 kB
intermediate.rst
1.7 kB
links.rst
188 Bytes
make.bat
754 Bytes
source-pytorch
Makefile
583 Bytes
_static
copybutton.js
2.7 kB
images
accelerator
ipus
profiler.png
129.6 kB
benchmarks
figure-parity-times.png
31.5 kB
general
PTL101_youtube_thumbnail.jpg
168.9 kB
fast_2.gif
2.6 MB
pl_overview.gif
1.1 MB
pl_overview_flat.jpg
419.9 kB
pl_quick_start_full_compressed.gif
7.9 MB
tf_loss.jpg
47.0 kB
tf_tags.jpg
58.5 kB
tutorial_cover.jpg
95.9 kB
icon.svg
2.8 kB
lightning_lite
lite.gif
90.5 kB
lightning_module
pt_to_pl.png
400.1 kB
pt_trainer.png
355.8 kB
logo.png
13.1 kB
logo.svg
31.2 kB
logo_light.svg
12.0 kB
mnist_imgs
mnist_cpu_bar.png
45.0 kB
mnist_gpu.png
67.8 kB
mnist_tb.png
195.3 kB
pt_to_pl.jpg
1.8 MB
restart_runtime.png
165.8 kB
runtime_tpu.png
119.3 kB
tpu_fast.png
194.6 kB
tpu_start.png
35.5 kB
trainer
lr_finder.png
17.5 kB
main.css
234 Bytes
_templates
autosummary
module.rst
650 Bytes
classtemplate.rst
256 Bytes
layout.html
405 Bytes
theme_variables.jinja
1.1 kB
accelerators
accelerator_prepare.rst
6.2 kB
gpu.rst
1.5 kB
gpu_advanced.rst
640 Bytes
gpu_basic.rst
4.6 kB
gpu_expert.rst
799 Bytes
gpu_faq.rst
2.4 kB
gpu_intermediate.rst
6.8 kB
mps.rst
643 Bytes
mps_basic.rst
2.4 kB
tpu.rst
1.2 kB
tpu_advanced.rst
2.3 kB
tpu_basic.rst
4.4 kB
tpu_faq.rst
3.5 kB
tpu_intermediate.rst
2.1 kB
advanced
compile.rst
16.9 kB
ddp_optimizations.rst
6.3 kB
finetuning.rst
35 Bytes
model_init.rst
3.0 kB
model_parallel
deepspeed.rst
21.6 kB
fsdp.rst
18.5 kB
index.rst
6.5 kB
tp.rst
11.5 kB
tp_fsdp.rst
11.6 kB
model_parallel.rst
4.2 kB
post_training_quantization.rst
8.7 kB
pretrained.rst
35 Bytes
pruning_quantization.rst
2.8 kB
speed.rst
16.4 kB
strategy_registry.rst
1.7 kB
training_tricks.rst
16.5 kB
transfer_learning.rst
3.8 kB
warnings.rst
1.4 kB
api_references.rst
4.1 kB
benchmarking
benchmarks.rst
1.2 kB
cli
lightning_cli.rst
3.0 kB
lightning_cli_advanced.rst
8.5 kB
lightning_cli_advanced_2.rst
5.8 kB
lightning_cli_advanced_3.rst
15.2 kB
lightning_cli_expert.rst
10.4 kB
lightning_cli_faq.rst
4.2 kB
lightning_cli_intermediate.rst
5.0 kB
lightning_cli_intermediate_2.rst
7.8 kB
clouds
cluster.rst
1.3 kB
cluster_advanced.rst
5.3 kB
cluster_expert.rst
1.5 kB
cluster_intermediate_1.rst
2.6 kB
cluster_intermediate_2.rst
2.9 kB
lightning_ai.rst
6.2 kB
common
checkpointing.rst
1.8 kB
checkpointing_advanced.rst
1.5 kB
checkpointing_basic.rst
7.1 kB
checkpointing_expert.rst
5.5 kB
checkpointing_intermediate.rst
10.4 kB
checkpointing_migration.rst
1.6 kB
child_modules.rst
2.2 kB
console_logs.rst
865 Bytes
early_stopping.rst
4.4 kB
evaluation.rst
704 Bytes
evaluation_basic.rst
3.8 kB
evaluation_intermediate.rst
5.6 kB
gradient_accumulation.rst
2.1 kB
hyperparameters.rst
1.6 kB
index.rst
6.3 kB
lightning_module.rst
40.4 kB
notebooks.rst
3.7 kB
optimization.rst
6.4 kB
precision.rst
933 Bytes
precision_basic.rst
3.7 kB
precision_expert.rst
715 Bytes
precision_intermediate.rst
10.9 kB
progress_bar.rst
6.2 kB
remote_fs.rst
2.9 kB
tbptt.rst
2.2 kB
trainer.rst
42.3 kB
common_usecases.rst
4.3 kB
community
ecosystem-ci.rst
2.0 kB
governance.rst
3.2 kB
index.rst
1.9 kB
conf.py
27.6 kB
data
access.rst
1.8 kB
alternatives.rst
6.9 kB
data.rst
945 Bytes
datamodule.rst
16.4 kB
iterables.rst
4.3 kB
debug
debugging.rst
805 Bytes
debugging_advanced.rst
1.3 kB
debugging_basic.rst
5.3 kB
debugging_intermediate.rst
3.1 kB
deploy
production.rst
1.7 kB
production_advanced.rst
3.1 kB
production_advanced_2.rst
2.3 kB
production_basic.rst
3.7 kB
production_intermediate.rst
2.8 kB
expertise_levels.rst
7.0 kB
extensions
accelerator.rst
3.9 kB
callbacks.rst
8.3 kB
callbacks_state.rst
2.6 kB
datamodules_state.rst
747 Bytes
entry_points.rst
1.8 kB
logging.rst
14.6 kB
plugins.rst
2.9 kB
strategy.rst
6.8 kB
glossary
index.rst
11.8 kB
index.rst
4.3 kB
integrations
hpu
index.rst
857 Bytes
strategies
index.rst
550 Bytes
levels
advanced.rst
1.9 kB
advanced_level_15.rst
853 Bytes
advanced_level_16.rst
790 Bytes
advanced_level_17.rst
850 Bytes
advanced_level_18.rst
856 Bytes
advanced_level_19.rst
803 Bytes
advanced_level_20.rst
991 Bytes
advanced_level_21.rst
946 Bytes
basic_level_2.rst
964 Bytes
basic_level_5.rst
981 Bytes
core_level_3.rst
759 Bytes
core_level_6.rst
982 Bytes
core_skills.rst
1.7 kB
expert.rst
1.2 kB
expert_level_22.rst
788 Bytes
expert_level_23.rst
553 Bytes
expert_level_24.rst
1.4 kB
intermediate.rst
2.1 kB
intermediate_level_10.rst
1.1 kB
intermediate_level_11.rst
903 Bytes
intermediate_level_12.rst
1.1 kB
intermediate_level_13.rst
1.3 kB
intermediate_level_14.rst
1.3 kB
intermediate_level_7.rst
965 Bytes
intermediate_level_9.rst
1.1 kB
links.rst
114 Bytes
make.bat
754 Bytes
model
build_model.rst
1.3 kB
build_model_advanced.rst
635 Bytes
build_model_intermediate.rst
1.3 kB
manual_optimization.rst
11.5 kB
own_your_loop.rst
533 Bytes
train_model_basic.rst
3.6 kB
notebooks.rst
168 Bytes
past_versions.rst
8.2 kB
starter
converting.rst
6.4 kB
installation.rst
2.4 kB
introduction.rst
10.3 kB
style_guide.rst
6.9 kB
tuning
profiler.rst
1.1 kB
profiler_advanced.rst
2.0 kB
profiler_basic.rst
5.4 kB
profiler_expert.rst
3.3 kB
profiler_intermediate.rst
10.8 kB
tutorials.rst
75 Bytes
upgrade
from_1_4.rst
824 Bytes
from_1_5.rst
711 Bytes
from_1_6.rst
598 Bytes
from_1_7.rst
485 Bytes
from_1_8.rst
372 Bytes
from_1_9.rst
202 Bytes
from_2_0.rst
202 Bytes
migration_guide.rst
1.5 kB
sections
1_4_advanced.rst
1.6 kB
1_4_devel.rst
0 Bytes
1_4_regular.rst
2.6 kB
1_5_advanced.rst
2.5 kB
1_5_devel.rst
893 Bytes
1_5_regular.rst
1.6 kB
1_6_advanced.rst
2.6 kB
1_6_devel.rst
1.2 kB
1_6_regular.rst
4.4 kB
1_7_advanced.rst
5.1 kB
1_7_devel.rst
5.6 kB
1_7_regular.rst
1.8 kB
1_8_advanced.rst
1.7 kB
1_8_devel.rst
775 Bytes
1_8_regular.rst
658 Bytes
1_9_advanced.rst
11.8 kB
1_9_devel.rst
11.1 kB
1_9_regular.rst
2.5 kB
2_0_advanced.rst
1.9 kB
2_0_devel.rst
423 Bytes
2_0_regular.rst
2.1 kB
versioning.rst
6.5 kB
visualize
experiment_managers.rst
676 Bytes
loggers.rst
1.3 kB
logging_advanced.rst
10.8 kB
logging_basic.rst
4.3 kB
logging_expert.rst
3.9 kB
logging_intermediate.rst
2.0 kB
supported_exp_managers.rst
6.4 kB
examples
README.md
1.2 kB
fabric
build_your_own_trainer
README.md
808 Bytes
run.py
2.8 kB
trainer.py
23.3 kB
dcgan
README.md
2.0 kB
train_fabric.py
9.1 kB
train_torch.py
9.3 kB
fp8_distributed_transformer
README.md
3.1 kB
requirements.txt
15 Bytes
train.py
3.1 kB
image_classifier
README.md
1.0 kB
train_fabric.py
7.5 kB
train_torch.py
5.6 kB
kfold_cv
README.md
1.5 kB
train_fabric.py
7.5 kB
language_model
README.md
635 Bytes
train.py
2.5 kB
meta_learning
README.md
1.5 kB
train_fabric.py
5.5 kB
train_torch.py
6.0 kB
reinforcement_learning
README.md
6.6 kB
requirements.txt
75 Bytes
rl
__init__.py
0 Bytes
agent.py
9.2 kB
loss.py
849 Bytes
utils.py
7.1 kB
train_fabric.py
8.3 kB
train_fabric_decoupled.py
14.7 kB
train_torch.py
10.9 kB
tensor_parallel
README.md
1.5 kB
data.py
678 Bytes
model.py
17.3 kB
parallelism.py
4.7 kB
train.py
2.8 kB
pytorch
basics
README.md
1.7 kB
autoencoder.py
7.2 kB
backbone_image_classifier.py
4.6 kB
profiler_example.py
4.2 kB
transformer.py
1.9 kB
bug_report
bug_report_model.ipynb
15.7 kB
bug_report_model.py
1.7 kB
domain_templates
computer_vision_fine_tuning.py
10.3 kB
generative_adversarial_net.py
7.4 kB
imagenet.py
7.3 kB
reinforce_learn_Qnet.py
13.3 kB
reinforce_learn_ppo.py
17.1 kB
semantic_segmentation.py
14.3 kB
fp8_distributed_transformer
README.md
3.2 kB
requirements.txt
15 Bytes
train.py
2.7 kB
hpu
mnist_sample.py
2.4 kB
ops_bf16_mnist.txt
12 Bytes
ops_fp32_mnist.txt
14 Bytes
servable_module
production.py
4.2 kB
tensor_parallel
README.md
1.9 kB
data.py
678 Bytes
model.py
17.3 kB
parallelism.py
4.7 kB
train.py
2.8 kB
run_fabric_examples.sh
800 Bytes
run_pl_examples.sh
1.5 kB
pyproject.toml
6.6 kB
requirements
README.md
827 Bytes
_integrations
accelerators.txt
69 Bytes
ci.txt
119 Bytes
collect_env_details.py
2.8 kB
data
data.txt
144 Bytes
docs.txt
632 Bytes
doctests.txt
42 Bytes
fabric
base.txt
404 Bytes
docs.txt
28 Bytes
examples.txt
346 Bytes
strategies.txt
764 Bytes
test.txt
319 Bytes
pytorch
base.txt
519 Bytes
check-avail-extras.py
197 Bytes
docs.txt
218 Bytes
examples.txt
384 Bytes
extra.txt
680 Bytes
loggers.info
178 Bytes
strategies.txt
511 Bytes
test.txt
593 Bytes
typing.txt
274 Bytes
requirements.txt
105 Bytes
setup.py
7.8 kB
src
lightning
MANIFEST.in
168 Bytes
__about__.py
1.3 kB
__init__.py
1.0 kB
__setup__.py
5.9 kB
__version__.py
355 Bytes
data
README.md
16.4 kB
__init__.py
955 Bytes
fabric
CHANGELOG.md
37.0 kB
__init__.py
1.7 kB
_graveyard
__init__.py
634 Bytes
tpu.py
4.4 kB
accelerators
__init__.py
1.3 kB
accelerator.py
1.9 kB
cpu.py
2.9 kB
cuda.py
7.2 kB
mps.py
3.0 kB
registry.py
4.5 kB
xla.py
6.3 kB
cli.py
9.2 kB
connector.py
28.4 kB
fabric.py
49.9 kB
loggers
__init__.py
795 Bytes
csv_logs.py
9.5 kB
logger.py
4.5 kB
tensorboard.py
13.0 kB
plugins
__init__.py
1.8 kB
collectives
__init__.py
325 Bytes
collective.py
3.5 kB
single_device.py
2.6 kB
torch_collective.py
8.6 kB
environments
__init__.py
1.3 kB
cluster_environment.py
2.4 kB
kubeflow.py
2.3 kB
lightning.py
3.9 kB
lsf.py
7.7 kB
mpi.py
4.2 kB
slurm.py
8.9 kB
torchelastic.py
3.4 kB
xla.py
3.6 kB
io
__init__.py
843 Bytes
checkpoint_io.py
2.6 kB
torch_io.py
3.6 kB
xla.py
3.0 kB
precision
__init__.py
1.5 kB
amp.py
5.1 kB
bitsandbytes.py
21.3 kB
deepspeed.py
3.6 kB
double.py
2.0 kB
fsdp.py
6.8 kB
half.py
2.4 kB
precision.py
5.6 kB
transformer_engine.py
8.2 kB
utils.py
2.6 kB
xla.py
2.5 kB
strategies
__init__.py
1.7 kB
ddp.py
10.2 kB
deepspeed.py
41.9 kB
dp.py
4.1 kB
fsdp.py
40.9 kB
launchers
__init__.py
933 Bytes
launcher.py
1.4 kB
multiprocessing.py
10.2 kB
subprocess_script.py
10.0 kB
xla.py
5.0 kB
model_parallel.py
26.3 kB
parallel.py
4.5 kB
registry.py
3.9 kB
single_device.py
2.9 kB
single_xla.py
3.2 kB
strategy.py
18.4 kB
xla.py
11.4 kB
xla_fsdp.py
29.9 kB
utilities
__init__.py
1.5 kB
apply_func.py
4.8 kB
cloud_io.py
5.3 kB
consolidate_checkpoint.py
2.9 kB
data.py
23.2 kB
device_dtype_mixin.py
4.3 kB
device_parser.py
7.4 kB
distributed.py
17.6 kB
enums.py
692 Bytes
exceptions.py
694 Bytes
imports.py
1.7 kB
init.py
4.5 kB
load.py
9.8 kB
logger.py
5.7 kB
optimizer.py
1.8 kB
rank_zero.py
3.1 kB
registry.py
2.3 kB
seed.py
6.2 kB
spike.py
7.6 kB
testing
__init__.py
99 Bytes
_runif.py
5.0 kB
throughput.py
27.5 kB
types.py
2.8 kB
warnings.py
2.3 kB
wrappers.py
17.6 kB
py.typed
0 Bytes
pytorch
CHANGELOG.md
464.9 kB
__init__.py
2.6 kB
_graveyard
__init__.py
771 Bytes
_torchmetrics.py
967 Bytes
hpu.py
3.6 kB
precision.py
3.3 kB
tpu.py
4.5 kB
accelerators
__init__.py
1.3 kB
accelerator.py
1.5 kB
cpu.py
3.2 kB
cuda.py
5.7 kB
mps.py
3.6 kB
xla.py
2.0 kB
callbacks
__init__.py
2.7 kB
batch_size_finder.py
8.2 kB
callback.py
11.0 kB
checkpoint.py
361 Bytes
device_stats_monitor.py
5.6 kB
early_stopping.py
11.7 kB
finetuning.py
19.1 kB
gradient_accumulation_scheduler.py
6.1 kB
lambda_function.py
3.4 kB
lr_finder.py
4.8 kB
lr_monitor.py
15.2 kB
model_checkpoint.py
38.8 kB
model_summary.py
3.8 kB
on_exception_checkpoint.py
2.3 kB
prediction_writer.py
6.1 kB
progress
__init__.py
945 Bytes
progress_bar.py
9.0 kB
rich_progress.py
24.7 kB
tqdm_progress.py
17.1 kB
pruning.py
23.0 kB
rich_model_summary.py
4.3 kB
spike.py
1.2 kB
stochastic_weight_avg.py
17.9 kB
throughput_monitor.py
10.4 kB
timer.py
8.3 kB
cli.py
39.1 kB
core
__init__.py
760 Bytes
datamodule.py
13.9 kB
hooks.py
26.8 kB
mixins
__init__.py
672 Bytes
hparams_mixin.py
7.0 kB
module.py
70.6 kB
optimizer.py
18.4 kB
saving.py
14.6 kB
demos
__init__.py
470 Bytes
boring_classes.py
10.4 kB
lstm.py
3.5 kB
mnist_datamodule.py
9.3 kB
transformer.py
7.0 kB
loggers
__init__.py
1.1 kB
comet.py
15.8 kB
csv_logs.py
5.6 kB
logger.py
5.1 kB
mlflow.py
14.9 kB
neptune.py
23.3 kB
tensorboard.py
10.4 kB
utilities.py
4.3 kB
wandb.py
25.0 kB
loops
__init__.py
1.1 kB
evaluation_loop.py
26.7 kB
fetchers.py
7.3 kB
fit_loop.py
23.0 kB
loop.py
3.8 kB
optimization
__init__.py
768 Bytes
automatic.py
13.1 kB
closure.py
2.6 kB
manual.py
5.5 kB
prediction_loop.py
18.7 kB
progress.py
9.2 kB
training_epoch_loop.py
26.2 kB
utilities.py
8.3 kB
overrides
__init__.py
0 Bytes
distributed.py
11.2 kB
plugins
__init__.py
1.4 kB
environments
__init__.py
881 Bytes
io
__init__.py
847 Bytes
async_plugin.py
2.4 kB
checkpoint_plugin.py
675 Bytes
torch_plugin.py
680 Bytes
wrapper.py
2.9 kB
xla_plugin.py
678 Bytes
layer_sync.py
3.9 kB
precision
__init__.py
1.5 kB
amp.py
5.8 kB
bitsandbytes.py
1.6 kB
deepspeed.py
5.7 kB
double.py
4.8 kB
fsdp.py
8.1 kB
half.py
2.6 kB
precision.py
7.4 kB
transformer_engine.py
1.9 kB
xla.py
3.7 kB
profilers
__init__.py
1.1 kB
advanced.py
5.1 kB
base.py
1.1 kB
profiler.py
5.5 kB
pytorch.py
24.5 kB
simple.py
6.8 kB
xla.py
3.0 kB
serve
__init__.py
210 Bytes
servable_module.py
3.3 kB
servable_module_validator.py
7.3 kB
strategies
__init__.py
1.7 kB
ddp.py
19.0 kB
deepspeed.py
41.1 kB
fsdp.py
30.6 kB
launchers
__init__.py
936 Bytes
launcher.py
931 Bytes
multiprocessing.py
14.4 kB
subprocess_script.py
7.0 kB
xla.py
7.5 kB
model_parallel.py
15.8 kB
parallel.py
5.1 kB
single_device.py
3.3 kB
single_xla.py
4.7 kB
strategy.py
25.6 kB
xla.py
13.2 kB
trainer
__init__.py
745 Bytes
call.py
12.5 kB
configuration_validator.py
8.6 kB
connectors
__init__.py
0 Bytes
accelerator_connector.py
31.9 kB
callback_connector.py
10.7 kB
checkpoint_connector.py
24.4 kB
data_connector.py
22.5 kB
logger_connector
__init__.py
114 Bytes
fx_validator.py
9.3 kB
logger_connector.py
10.3 kB
result.py
20.5 kB
signal_connector.py
6.6 kB
setup.py
8.2 kB
states.py
2.9 kB
trainer.py
72.6 kB
tuner
__init__.py
644 Bytes
batch_size_scaling.py
12.9 kB
lr_finder.py
19.5 kB
tuning.py
11.5 kB
utilities
__init__.py
1.9 kB
_pytree.py
1.5 kB
argparse.py
2.7 kB
combined_loader.py
16.3 kB
compile.py
5.9 kB
consolidate_checkpoint.py
1.2 kB
data.py
16.9 kB
deepspeed.py
4.8 kB
enums.py
1.5 kB
exceptions.py
1.5 kB
grads.py
2.0 kB
imports.py
2.1 kB
memory.py
3.1 kB
migration
__init__.py
701 Bytes
migration.py
14.5 kB
utils.py
9.1 kB
model_helpers.py
5.3 kB
model_summary
__init__.py
910 Bytes
model_summary.py
18.9 kB
model_summary_deepspeed.py
4.6 kB
parameter_tying.py
2.4 kB
parsing.py
12.1 kB
rank_zero.py
1.1 kB
seed.py
1.8 kB
signature_utils.py
1.4 kB
testing
__init__.py
100 Bytes
_runif.py
3.7 kB
types.py
4.1 kB
upgrade_checkpoint.py
3.4 kB
warnings.py
723 Bytes
lightning_fabric
MANIFEST.in
287 Bytes
README.md
13.4 kB
__about__.py
477 Bytes
__setup__.py
4.9 kB
__version__.py
355 Bytes
py.typed
0 Bytes
shell-folder_code-lives-lightning.info
264 Bytes
pytorch_lightning
MANIFEST.in
495 Bytes
README.md
15.6 kB
__about__.py
2.2 kB
__setup__.py
5.0 kB
__version__.py
355 Bytes
py.typed
0 Bytes
shell-folder_code-lives-lightning.info
266 Bytes
version.info
12 Bytes
tests
README.md
3.2 kB
legacy
README.md
893 Bytes
back-compatible-versions.txt
644 Bytes
checkpoints
.gitkeep
0 Bytes
generate_checkpoints.sh
2.0 kB
requirements.txt
0 Bytes
simple_classif_training.py
2.0 kB
parity_fabric
__init__.py
0 Bytes
conftest.py
1.1 kB
models.py
2.5 kB
run_standalone_tasks.sh
1.4 kB
test_parity_ddp.py
5.4 kB
test_parity_simple.py
4.7 kB
utils.py
2.2 kB
parity_pytorch
__init__.py
274 Bytes
generate_comparison.py
2.2 kB
measure.py
1.1 kB
models.py
2.4 kB
test_basic_parity.py
5.3 kB
test_sync_batchnorm_parity.py
4.4 kB
run_standalone_tests.sh
4.4 kB
tests_fabric
__init__.py
251 Bytes
accelerators
__init__.py
0 Bytes
test_cpu.py
1.6 kB
test_cuda.py
5.7 kB
test_mps.py
1.9 kB
test_registry.py
2.4 kB
test_xla.py
2.0 kB
conftest.py
10.2 kB
graveyard
test_tpu.py
1.5 kB
helpers
__init__.py
0 Bytes
dataloaders.py
1.6 kB
datasets.py
704 Bytes
runif.py
854 Bytes
loggers
__init__.py
0 Bytes
test_csv.py
8.5 kB
test_tensorboard.py
8.6 kB
plugins
__init__.py
0 Bytes
collectives
__init__.py
0 Bytes
test_single_device.py
828 Bytes
test_torch_collective.py
13.0 kB
environments
__init__.py
0 Bytes
test_kubeflow.py
2.7 kB
test_lightning.py
3.2 kB
test_lsf.py
4.1 kB
test_mpi.py
4.9 kB
test_slurm.py
6.9 kB
test_torchelastic.py
3.2 kB
test_xla.py
3.8 kB
precision
__init__.py
0 Bytes
test_all.py
926 Bytes
test_amp.py
4.2 kB
test_amp_integration.py
3.9 kB
test_bitsandbytes.py
11.3 kB
test_deepspeed.py
3.5 kB
test_deepspeed_integration.py
1.5 kB
test_double.py
1.2 kB
test_double_integration.py
2.0 kB
test_fsdp.py
5.7 kB
test_half.py
2.4 kB
test_transformer_engine.py
4.8 kB
test_utils.py
1.7 kB
test_xla.py
2.2 kB
test_xla_integration.py
2.4 kB
run_tpu_tests.sh
1.5 kB
strategies
__init__.py
0 Bytes
launchers
__init__.py
0 Bytes
test_multiprocessing.py
4.8 kB
test_multiprocessing_integration.py
2.1 kB
test_subprocess_script.py
8.4 kB
test_ddp.py
7.5 kB
test_ddp_integration.py
5.3 kB
test_deepspeed.py
17.8 kB
test_deepspeed_integration.py
16.3 kB
test_dp.py
3.6 kB
test_fsdp.py
17.7 kB
test_fsdp_integration.py
27.8 kB
test_model_parallel.py
16.3 kB
test_model_parallel_integration.py
29.3 kB
test_registry.py
2.3 kB
test_single_device.py
3.9 kB
test_strategy.py
12.5 kB
test_xla.py
8.6 kB
test_xla_fsdp.py
5.6 kB
test_xla_fsdp_integration.py
8.2 kB
test_cli.py
8.4 kB
test_connector.py
46.7 kB
test_fabric.py
50.3 kB
test_wrappers.py
27.5 kB
utilities
__init__.py
0 Bytes
test_apply_func.py
2.3 kB
test_cloud_io.py
3.7 kB
test_consolidate_checkpoint.py
4.0 kB
test_data.py
24.3 kB
test_device_dtype_mixin.py
4.7 kB
test_device_parser.py
3.5 kB
test_distributed.py
9.6 kB
test_imports.py
3.0 kB
test_init.py
4.2 kB
test_load.py
4.8 kB
test_logger.py
6.3 kB
test_optimizer.py
2.8 kB
test_rank_zero.py
1.8 kB
test_registry.py
2.1 kB
test_runif.py
250 Bytes
test_seed.py
5.0 kB
test_spike.py
6.4 kB
test_throughput.py
11.8 kB
test_warnings.py
5.6 kB
tests_pytorch
__init__.py
1.3 kB
accelerators
__init__.py
0 Bytes
test_common.py
2.0 kB
test_cpu.py
2.8 kB
test_gpu.py
2.3 kB
test_mps.py
5.1 kB