lightning>=2.0.0
colossalai>=0.2.0
torch<2.0.0

[test]
codecov>=2.1
coverage>=5.0
mypy==1.1.1
pytest-cov
pytest>=6.0
scikit-learn<1.2.3,>=1.0.0
torchmetrics<0.12.0,>=0.8.0
