lightning>=2.1.0
torch>=2.0.0
torchvision>=0.15.0
Pillow>=9.0.0
pandas>=2.0.0
numpy>=1.24.0
scikit-learn>=1.3.0
joblib>=1.3.0
onnx>=1.14.0
onnxruntime<1.23.0,>=1.16.0
onnxscript>=0.1.0
torchmetrics>=1.0.0
pyyaml>=6.0
jsonargparse[signatures]>=4.27.7
tensorboard>=2.13.0

[api]
fastapi>=0.104.0
uvicorn[standard]>=0.24.0
jinja2>=3.1.0
python-multipart>=0.0.6
pydantic>=2.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
flake8>=6.0.0
mypy>=1.0.0

[jupyter]
jupyter>=1.0.0
ipykernel>=6.25.0
