lightning/examples/app_multi_node/pl_boring_script.py

import lightning as L
from lightning.pytorch.demos.boring_classes import BoringModel

if __name__ == "__main__":
    model = BoringModel()
    trainer = L.Trainer(max_epochs=1)
    trainer.fit(model)