Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
COMMITS
/ tests/tests_fabric/test_wrappers.py January 7, 2025
P
[pre-commit.ci] pre-commit suggestions (#20532)
pre-commit-ci[bot] committed
July 13, 2024
A
Add Python 3.12 to the CPU test matrix (#20078)
awaelchli committed
July 2, 2024
A
Add testing for PyTorch 2.4 (Fabric) (#20028)
awaelchli committed
June 30, 2024
A
Drop PyTorch 2.0 from the test matrix (#20009)
awaelchli committed
May 8, 2024
A
Add function to explicitly mark forward methods in Fabric (#19690)
awaelchli committed
April 27, 2024
A
Remove support for PyTorch 1.13 (#19706)
Adrian Wälchli committed
March 27, 2024
C
Fix monkeypatching of `_FabricModule` methods (#19705)
Carlos Mocholí committed
March 15, 2024
A
Fix `torch.compile` patching when applied as decorator (#19627)
awaelchli committed
February 27, 2024
A
Alternative mechanism to detect missing `Fabric.backward()` call (#19493)
awaelchli committed
January 24, 2024
A
Reapply `torch.compile` in Fabric.setup() (#19280)
awaelchli committed
October 23, 2023
A
Avoid false-positive warnings about method calls on the Fabric-wrapped module (#18819)
Adrian Wälchli committed
October 4, 2023
A
Drop support for PyTorch 1.11 (#18691)
Adrian Wälchli committed
October 3, 2023
P
[pre-commit.ci] pre-commit suggestions (#18697)
pre-commit-ci[bot] committed
A
Enable passing `load_state_dict(..., assign=True|False)` in FabricModule (#18690)
Adrian Wälchli committed
September 26, 2023
J
ruff: replace isort with ruff +TPU (#17684)
Jirka Borovec committed
September 12, 2023
A
More robust FabricOptimizer/LightningOptimizer wrapping logic (#18516)
Adrian Wälchli committed
September 6, 2023
A
Refresh `_FabricOptimizer.__dict__` when loading a state dict (#18488)
Adrian Wälchli committed
August 9, 2023
J
docformatter: config with black (#18064)
Jirka Borovec committed
June 26, 2023
A
Reduce false positive warnings when calling module methods in Fabric (#17875)
Adrian Wälchli committed
June 12, 2023
B
Pass-through setattr for FabricModule (#17731)
Boon committed
May 11, 2023
A
Add support for saving with full state-dict in Fabric's FSDP (#17526)
Adrian Wälchli committed
May 5, 2023
J
ruff: PT some more fixes (#17569)
Jirka Borovec committed
May 4, 2023
J
ruff: autofix PT (#17541)
Jirka Borovec committed
May 3, 2023
A
Support compiling a module after it was set up by Fabric (#17529)
Adrian Wälchli committed
April 26, 2023
P
[pre-commit.ci] pre-commit suggestions (#17271)
pre-commit-ci[bot] committed
April 25, 2023
D
Update deepspeed requirement support window (#16813)
dependabot[bot] committed
April 24, 2023
A
Enable precision autocast for `LightningModule` step methods in Fabric (#17439)
Adrian Wälchli committed
April 21, 2023
A
Fix LightningModule step methods bypassing DDP wrapper in Fabric (#17424)
Adrian Wälchli committed
March 14, 2023
A
Add `is_wrapped` utility function for Fabric (#16953)
Atharva Phatak committed
March 7, 2023
A
Add test for `torch.compile()` with `Fabric.setup()` (#16977)
Adrian Wälchli committed
March 3, 2023
J
update list of fist party packages (#16859)
Jirka Borovec committed
March 2, 2023
A
Optimize precision conversion in forward of Fabric module wrapper (#16903)
Adrian Wälchli committed
February 23, 2023
C
Fabric: do `set_epoch` for `batch_sampler.sampler` (#16841)
Carlos Mocholí committed
February 6, 2023
J
copyright Lightning AI team (#16647)
Jirka Borovec committed
February 1, 2023
J
tests: switch imports for fabric (#16592)
Jirka Borovec committed
January 10, 2023
A
Rename leftover definitions in Lite tests (#16309)
Adrian Wälchli committed
January 9, 2023
A
Handle `set_to_none` when using DeepSpeed optimizer in Lite (#16275)
Adrian Wälchli committed
January 4, 2023
C
Rename LightningLite to Fabric (#16244)
Carlos Mocholí committed