torch>=2.0.1
numpy
google-cloud-storage
absl-py
lightning>=2.0

[test]
pytest
