#dl

https://github.com/Lightning-AI/lightning/releases/tag/2.0.0

You can compile (torch 2.0) LightningModule now.

import torch
import lightning as L
model = LitModel()
# This will compile forward and {training,validation,test,predict}_step
compiled_model = torch.compile(model)
trainer = L.Trainer()
trainer.fit(compiled_model)
GitHub
Release Lightning 2.0: Fast, Flexible, Stable · Lightning-AI/lightning
Release Lightning 2.0: Fast, Flexible, Stable · Lightning-AI/lightning
Lightning AI is excited to announce the release of Lightning 2.0 ⚡

Highlights
Backward Incompatible Changes

PyTorch
Fabric


Full Changelog

PyTorch
Fabric
App


Contributors

Over the last coupl...
 
 
Back to Top