torch>=2.0.1
s3torchconnectorclient>=1.2.0

[e2e]
torchdata
torchvision
Pillow
boto3
numpy
pytest-xdist

[lightning]
lightning>=2.0

[lightning-tests]
s3torchconnector[lightning]
s3fs

[test]
pytest
pytest-timeout
hypothesis
flake8
black
mypy
