pytorch-lightning>=2.3.0,<2.3.3
# the below is uncommented when master is targeting a specific pl dev master commit
#git+https://github.com/Lightning-AI/pytorch-lightning.git@90d04b5b86f37994cdceccc6de32f0e93b1cc7f0#egg=pytorch-lightning
torch>=2.0.0
mpmath<1.4.0  # temporary requirement to avoid installation of alpha version of mpmath
