Allow pytorch 2.0 & pytorch-lightning 2.0

[alexandervaneck]

Hi, I'm just getting started with pycave and would like to use pytorch 2.0. It's torch.compile mechanism could be really powerful.

Would you please allow your package to also allow 2.0 for both pytorch and pytorch-lightning?

Also, would you expected any speedups from using torch.compile?

Thank you! 🙇