feat: update lightning example to lightning 2.0#603

dwahdany wants to merge 3 commits into

meta-pytorch:mainmeta-pytorch/opacus:mainfrom

dwahdany:maindwahdany/opacus:mainCopy head branch name to clipboard