Carlos Mocholí

Add `torch.compile` guide to docs

1

### 📚 Documentation This applies to - [x] Fabric (https://github.com/Lightning-AI/pytorch-lightning/pull/19330) - [ ] PyTorch Lightning Things to mention: - Compiling `nn.Module` vs `LightningModule` or `fabric.setup(module)` (#17250) - Known limitations. -...

Label tracking meta-issue (edit me to get automatically CC'ed on issues!)

9

This issue is used by [lightning-probot](https://github.com/PyTorchLightning/probot) to manage subscriptions to labels. To subscribe yourself to a label, add a line `* label @yourusername`, or add your username to an existing...

## 🐛 Bug Remove or replace (with Fabric) the following: https://github.com/Lightning-AI/lightning-flash/blob/master/flash_examples/integrations/baal/image_classification_active_learning.py https://github.com/Lightning-AI/lightning-flash/blob/master/flash/image/classification/integrations/baal/loop.py As it will no longer work after https://github.com/Lightning-AI/lightning/pull/16384 cc @awaelchli

## What does this PR do? Supports ```python import os import thunder import torch import torch.distributed as torch_dist from thunder import ThunderModule world_size = int(os.environ.get("WORLD_SIZE", 1)) local_rank = int(os.environ.get("LOCAL_RANK", 0))...

adding DDP/FSDP transform after JITting does not work

2

## 🐛 Bug The snippet below looks hacky, but it's how I'm approaching support for having the user control the `thunder.jit` call outside of Fabric: https://github.com/Lightning-AI/litgpt/pull/1204 The objective is that...

## 🐛 Bug ```python extensions/thunder/pretrain.py:146: in setup main( extensions/thunder/pretrain.py:233: in main fit(fabric, devices, state, train_dataloader, val_dataloader, out_dir, tokenizer_dir, train, eval) extensions/thunder/pretrain.py:253: in fit validate(fabric, model, val_dataloader, max_iters=2) # sanity check...

Label tracking meta-issue (edit me to get automatically CC'ed on issues!)

1

This issue is used by [lightning-probot](https://github.com/Lightning-AI/probot) to manage subscriptions to labels. To subscribe yourself to a label, add a line `* label @yourusername`, or add your username to an existing...

Follow-up to https://github.com/Lightning-AI/lightning-thunder/pull/140

Follow-up to https://github.com/Lightning-AI/litgpt/pull/1177 cc @rasbt

This reverts commit d8dc97e4160307a674a6f4bd3ccdacbaa90b4c90.