torch>=2.0.1
s3torchconnectorclient>=1.2.4

[e2e]
torchdata
torchvision
Pillow>=10.3.0
boto3
numpy<2
pytest-xdist

[lightning]
lightning>=2.0

[lightning-tests]
s3torchconnector[lightning]
s3fs

[test]
pytest
pytest-timeout
hypothesis
flake8
black
mypy
