python-socketio[client]>=5.0.0
pynvml>=11.0.0
flask>=2.0.0

[all]
flexium[debug,dev,test-frameworks,torch]

[debug]
debugpy>=1.6.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
mypy>=1.0.0
ruff>=0.1.0

[test-frameworks]
pytorch-lightning>=2.0.0
transformers>=4.20.0
accelerate>=0.20.0
timm>=0.9.0
datasets>=2.0.0

[torch]
torch>=1.9.0
torchvision>=0.10.0
