torch>=2.0
pydantic

[distributed]
pytorch-lightning>=2.0
