Add support for PyTorch 2.1

Should we increase the dependency ranges in [`_setup_utils.py`](https://github.com/autogluon/autogluon/blob/master/core/src/autogluon/core/_setup_utils.py) to `torch>=2.0,<2.2` and, potentially, `lightning>=2.0,<2.2` in the AutoGluon v1.0 release?

To do:
- [x] #3708
- [x] #3827

@zhiqiangdon @tonyhoo @Innixma 