PyTorch Lightning: Multiple scalars (e.g. train and valid loss) in same Tensorboard graph

Question

With PyTorch Tensorboard I can log my train and valid loss in a single Tensorboard graph like this:

writer = torch.utils.tensorboard.SummaryWriter()

for i in ran