TF 2.x and PyTorch Lightning Callbacks for GPU monitoring - View it on GitHub