torch>=2.0.0
lightning>=2.0.0

[test]
coverage>=5.0
mypy==1.5.1
pytest-cov
pytest>=6.0
