Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
COMMITS
/ tests/tests_pytorch/callbacks/test_pruning.py October 16, 2025
N
Fix logged sparsity calculation in `ModelPruning` (#21223)
Nicki Skafte Detlefsen committed
September 10, 2025
N
Fix missing reset when pruning with lottery ticket hypothesis (#21191)
Nicki Skafte Detlefsen committed
June 18, 2025
G
check param is of `nn.Parameter` type for pruning sanitization (#20783)
Garibaldi Pineda-Garcia committed
January 7, 2025
P
[pre-commit.ci] pre-commit suggestions (#20532)
pre-commit-ci[bot] committed
July 8, 2024
A
Set `weights_only` in tests to avoid warnings in PyTorch 2.4 (#20057)
awaelchli committed
April 28, 2024
A
Avoid interactions through test artifacts (#19821)
Adrian Wälchli committed
March 17, 2024
F
test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/callbacks` (#19655)
Felix Hirwa Nshuti committed
February 15, 2024
J
precommit: ruff-format (#19434)
Jirka Borovec committed
September 26, 2023
J
ruff: replace isort with ruff +TPU (#17684)
Jirka Borovec committed
September 20, 2023
T
August 9, 2023
J
docformatter: config with black (#18064)
Jirka Borovec committed
August 3, 2023
A
Address test flakiness (#18223)
Adrian Wälchli committed
May 5, 2023
J
ruff: PT some more fixes (#17569)
Jirka Borovec committed
February 23, 2023
C
Trainer: auto default (#16847)
Carlos Mocholí committed
February 6, 2023
J
copyright Lightning AI team (#16647)
Jirka Borovec committed
February 2, 2023
J
tests: switch imports for pytorch (#16595)
Jirka Borovec committed
June 15, 2022
J
Future 4/n: test & legacy in test/ folder (#13295)
Jirka Borovec committed