torch>=2.0
numpy
pillow

[dev]
pytest
ruff
build
twine

[hub]
huggingface_hub

[light]
bitsandbytes

[monitor]
nvidia-ml-py
