numpy ~= 1.26.4
pytorch-lightning ~= 2.2.5
