PyTorch Lightning is now called Lightning | Follow our new handle @LightningAI | THE deep learning framework for building and training ML models