Is PyTorch Lightning supported

As of June 2021, the development and implementation of PyTorch Lightning is in progress and this will be included in a future release.